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

SIR model parameter estimation using a novel algorithm for differentiated uniformization.

CFC-Net: A Critical Feature Capturing Network for Arbitrary-Oriented Object Detection in Remote Sensing Images

Tutorial page of the Climate Hack, the greatest hackathon ever

Implementation of SwinTransformerV2 in TensorFlow.

Unified Instance and Knowledge Alignment Pretraining for Aspect-based Sentiment Analysis

Abstractive opinion summarization system (SelSum) and the largest dataset of Amazon product summaries (AmaSum). EMNLP 2021 conference paper.

(CVPR2021) Kaleido-BERT: Vision-Language Pre-training on Fashion Domain

Measure WWjj polarization fraction

PyTorch implementation of Anomaly Transformer: Time Series Anomaly Detection with Association Discrepancy

Realtime_Multi-Person_Pose_Estimation

noisy labels; missing labels; semi-supervised learning; entropy; uncertainty; robustness and generalisation.

Implementation for "Conditional entropy minimization principle for learning domain invariant representation features"

The VeriNet toolkit for verification of neural networks

A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation.

Exe-to-xlsm - Simple script to create VBscript of exe and inject to xlsm

Awesome AI Learning with +100 AI Cheat-Sheets, Free online Books, Top Courses, Best Videos and Lectures, Papers, Tutorials, +99 Researchers, Premium Websites, +121 Datasets, Conferences, Frameworks, Tools

A dual benchmarking study of visual forgery and visual forensics techniques

TRACER: Extreme Attention Guided Salient Object Tracing Network implementation in PyTorch

Implicit Graph Neural Networks

PyTorch implementation of convolutional neural networks-based text-to-speech synthesis models