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

Code for NeurIPS 2021 paper 'Spatio-Temporal Variational Gaussian Processes'

AutoDeeplab / auto-deeplab / AutoML for semantic segmentation, implemented in Pytorch

Contrastive Feature Loss for Image Prediction

SingleVC performs any-to-one VC, which is an important component of MediumVC project.

Image restoration with neural networks but without learning.

Intent parsing and slot filling in PyTorch with seq2seq + attention

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

FFCV: Fast Forward Computer Vision (and other ML workloads!)

Real-time Neural Representation Fusion for Robust Volumetric Mapping

Home repository for the Regularized Greedy Forest (RGF) library. It includes original implementation from the paper and multithreaded one written in C++, along with various language-specific wrappers.

FcaNet: Frequency Channel Attention Networks

DeepLab-ResNet rebuilt in TensorFlow

QilingLab challenge writeup

Realtime segmentation with ENet, the fast and accurate segmentation net.

[NeurIPS-2021] Slow Learning and Fast Inference: Efficient Graph Similarity Computation via Knowledge Distillation

Local Similarity Pattern and Cost Self-Reassembling for Deep Stereo Matching Networks

Using Clinical Drug Representations for Improving Mortality and Length of Stay Predictions

ALBERT: A Lite BERT for Self-supervised Learning of Language Representations

A Dataset for Direct Quotation Extraction and Attribution in News Articles.

Forecasting directional movements of stock prices for intraday trading using LSTM and random forest