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

Unsupervised phone and word segmentation using dynamic programming on self-supervised VQ features.

Animatable Neural Radiance Fields for Modeling Dynamic Human Bodies

FedScale: Benchmarking Model and System Performance of Federated Learning

Functional TensorFlow Implementation of Singular Value Decomposition for paper Fast Graph Learning

Implement of "Training deep neural networks via direct loss minimization" in PyTorch for 0-1 loss

PyTorch Implementation of Daft-Exprt: Robust Prosody Transfer Across Speakers for Expressive Speech Synthesis

Sequence lineage information extracted from RKI sequence data repo

3D Avatar Lip Syncronization from speech (JALI based face-rigging)

CVPR2021 Content-Aware GAN Compression

Implementation of ConvMixer-Patches Are All You Need? in TensorFlow and Keras

PyTorch implementation of Asymmetric Siamese (https://arxiv.org/abs/2204.00613)

Geometric Vector Perceptron --- a rotation-equivariant GNN for learning from biomolecular structure

A FAIR dataset of TCV experimental results for validating edge/divertor turbulence models.

Consumer Fairness in Recommender Systems: Contextualizing Definitions and Mitigations

The repository offers the official implementation of our paper in PyTorch.

Unofficial Pytorch Lightning implementation of Contrastive Syn-to-Real Generalization (ICLR, 2021)

Acute ischemic stroke dataset

Code for 1st place solution in Sleep AI Challenge SNU Hospital

The repo of Feedback Networks, CVPR17

The repository for freeCodeCamp's YouTube course, Algorithmic Trading in Python