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

Post-training Quantization for Neural Networks with Provable Guarantees

Implements VQGAN+CLIP for image and video generation, and style transfers, based on text and image prompts. Emphasis on ease-of-use, documentation, and smooth video creation.

Code accompanying the paper Say As You Wish: Fine-grained Control of Image Caption Generation with Abstract Scene Graphs (Chen et al., CVPR 2020, Oral).

Adversarial-autoencoders - Tensorflow implementation of Adversarial Autoencoders

VID-Fusion: Robust Visual-Inertial-Dynamics Odometry for Accurate External Force Estimation

Help you understand Manual and w/ Clutch point while driving.

Deduplicating Training Data Makes Language Models Better

Gym for multi-agent reinforcement learning

Official implementation of the paper WAV2CLIP: LEARNING ROBUST AUDIO REPRESENTATIONS FROM CLIP

Creating a custom CNN hypertunned architeture for the Fashion MNIST dataset with Python, Keras and Tensorflow.

Python implementation of MULTIseq barcode alignment using fuzzy string matching and GMM barcode assignment

Deep RGB-D Saliency Detection with Depth-Sensitive Attention and Automatic Multi-Modal Fusion (CVPR'2021, Oral)

Implementation of "A MLP-like Architecture for Dense Prediction"

This project uses Template Matching technique for object detecting by detection of template image over base image.

Code for "Layered Neural Rendering for Retiming People in Video."

Tensorflow Implementation of SMU: SMOOTH ACTIVATION FUNCTION FOR DEEP NETWORKS USING SMOOTHING MAXIMUM TECHNIQUE

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Deep Markov Factor Analysis (NeurIPS2021)

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

Caffe-like explicit model constructor. C(onfig)Model