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

Builds a LoRa radio frequency fingerprint identification (RFFI) system based on deep learning techiniques

ivadomed is an integrated framework for medical image analysis with deep learning.

Automatic learning-rate scheduler

This is the official code release for the paper Shape and Material Capture at Home

Autonomous Movement from Simultaneous Localization and Mapping

The devkit of the nuPlan dataset.

Faster RCNN pytorch windows

Neighborhood Reconstructing Autoencoders

Autoencoder - Reducing the Dimensionality of Data with Neural Network

Keras community contributions

Code for the Population-Based Bandits Algorithm, presented at NeurIPS 2020.

UMPNet: Universal Manipulation Policy Network for Articulated Objects

Instance Semantic Segmentation List

Keras Realtime Multi-Person Pose Estimation - Keras version of Realtime Multi-Person Pose Estimation project

Permute Me Softly: Learning Soft Permutations for Graph Representations

Calibrated Hyperspectral Image Reconstruction via Graph-based Self-Tuning Network.

Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track (SIGIR 2021 Full Paper).

This repository includes the official project for the paper: TransMix: Attend to Mix for Vision Transformers.

An educational resource to help anyone learn deep reinforcement learning.

Contrastive Loss Gradient Attack (CLGA)