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

Decision Transformer: A brand new Offline RL Pattern

Camera Distortion-aware 3D Human Pose Estimation in Video with Optimization-based Meta-Learning

PyTorch implementation of some learning rate schedulers for deep learning researcher.

CROSS-LINGUAL ABILITY OF MULTILINGUAL BERT: AN EMPIRICAL STUDY

Code for classifying international patents based on the text of their titles/abstracts

Experiments with differentiable stacks and queues in PyTorch

Adds timm pretrained backbone to pytorch's FasterRcnn model

Awesome-AI-books - Some awesome AI related books and pdfs for learning and downloading

The toolkit to generate auto labeled datasets

Fast image augmentation library and easy to use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about library: https://www.mdpi.com/2078-2489/11/2/125

Hierarchical Attentive Recurrent Tracking

Independent and minimal implementations of some reinforcement learning algorithms using PyTorch (including PPO, A3C, A2C, ...).

harmonic-percussive-residual separation algorithm wrapped as a VST3 plugin (iPlug2)

A more easy-to-use implementation of KPConv

Fast sparse deep learning on CPUs

AI-UPV at IberLEF-2021 EXIST task: Sexism Prediction in Spanish and English Tweets Using Monolingual and Multilingual BERT and Ensemble Models

Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision

COVINS -- A Framework for Collaborative Visual-Inertial SLAM and Multi-Agent 3D Mapping

A Flexible Generative Framework for Graph-based Semi-supervised Learning (NeurIPS 2019)

Shape Matching of Real 3D Object Data to Synthetic 3D CADs (3DV project @ ETHZ)