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

Reproducing code of hair style replacement method from Barbershorp.

[Machine Learning Engineer Basic Guide] 부스트캠프 AI Tech - Product Serving 자료

This app is a simple example of using Strealit to create a financial data web app.

This repository is based on Ultralytics/yolov5, with adjustments to enable polygon prediction boxes.

Label-Free Model Evaluation with Semi-Structured Dataset Representations

PHOTONAI is a high level python API for designing and optimizing machine learning pipelines.

Wav2Vec for speech recognition, classification, and audio classification

Source Code For Template-Based Named Entity Recognition Using BART

YOLOv2 in PyTorch

The code for our paper CrossFormer: A Versatile Vision Transformer Based on Cross-scale Attention.

Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)

Neural Factorization of Shape and Reflectance Under An Unknown Illumination

A collection of awesome resources image-to-image translation.

Weakly Supervised Segmentation by Tensorflow.

A small demonstration of using WebDataset with ImageNet and PyTorch Lightning

Code accompanying "Evolving spiking neuron cellular automata and networks to emulate in vitro neuronal activity," accepted to IEEE SSCI ICES 2021

[RSS 2021] An End-to-End Differentiable Framework for Contact-Aware Robot Design

Pytorch library for end-to-end transformer models training and serving

The code is an implementation of Feedback Convolutional Neural Network for Visual Localization and Segmentation.

What can linearized neural networks actually say about generalization?