Code for the paper "M2m: Imbalanced Classification via Major-to-minor Translation" (CVPR 2020)

Last update: Oct 13, 2022

Related tags

Overview

M2m: Imbalanced Classification via Major-to-minor Translation

This repository contains code for the paper "M2m: Imbalanced Classification via Major-to-minor Translation" by Jaehyung Kim*, Jongheon Jeong* and Jinwoo Shin.

Dependencies

python3
pytorch >= 1.1.0
torchvision
tqdm

Scripts

Please check out run.sh for all the scripts to reproduce the CIFAR-10-LT results reported.

Training procedure of M2m

Train a baseline network g for generating minority samples

python train.py --no_over --ratio 100 --decay 2e-4 --model resnet32 --dataset cifar10 \
--lr 0.1 --batch-size 128 --name 'ERM' --warm 200 --epoch 200

Train another network f using M2m with the pre-trained g

python train.py -gen -r --ratio 100 --decay 2e-4 --model resnet32 --dataset cifar10 \
--lr 0.1 --batch-size 128 --name 'M2m' --beta 0.999 --lam 0.5 --gamma 0.9 \
--step_size 0.1 --attack_iter 10 --warm 160 --epoch 200 --net_g ./checkpoint/pre_trained_g.t7

We also provide a pre-trained ResNet-32 model of g at checkpoint/erm_r100_c10_trial1.t7, so one can directly use M2m without pre-training as follows:

python train.py -gen -r --ratio 100 --decay 2e-4 --model resnet32 --dataset cifar10 \
--lr 0.1 --batch-size 128 --name 'M2m' --beta 0.999 --lam 0.5 --gamma 0.9 \
--step_size 0.1 --attack_iter 10 --warm 160 --epoch 200 --net_g ./checkpoint/erm_r100_c10_trial1.t7

Code for the paper "M2m: Imbalanced Classification via Major-to-minor Translation" (CVPR 2020)

Related tags

Overview

M2m: Imbalanced Classification via Major-to-minor Translation

Dependencies

Scripts

Training procedure of M2m

Owner

diablo2 resurrected loot filter

Camera ready code repo for the NeuRIPS 2021 paper: "Impression learning: Online representation learning with synaptic plasticity".

《Single Image Reflection Removal Beyond Linearity》(CVPR 2019)

FTIR-Deep Learning - FTIR Deep Learning With Python

This repository contains the code to replicate the analysis from the paper "Moving On - Investigating Inventors' Ethnic Origins Using Supervised Learning"

"Structure-Augmented Text Representation Learning for Efficient Knowledge Graph Completion"(WWW 2021)

imbalanced-DL: Deep Imbalanced Learning in Python

LSTM and QRNN Language Model Toolkit for PyTorch

Identify the emotion of multiple speakers in an Audio Segment

Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch

MADT: Offline Pre-trained Multi-Agent Decision Transformer

Training Cifar-10 Classifier Using VGG16

[TIP 2020] Multi-Temporal Scene Classification and Scene Change Detection with Correlation based Fusion

Reproducing code of hair style replacement method from Barbershorp.

The source code for Adaptive Kernel Graph Neural Network at AAAI2022

Code for the Lovász-Softmax loss (CVPR 2018)

Simulating an AI playing 2048 using the Expectimax algorithm

A computational optimization project towards the goal of gerrymandering the results of a hypothetical election in the UK.

CDTrans: Cross-domain Transformer for Unsupervised Domain Adaptation

This repository is for DSA and CP scripts for reference.