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 our paper Aspect Sentiment Quad Prediction as Paraphrase Generation in EMNLP 2021.

D-NeRF: Neural Radiance Fields for Dynamic Scenes

Cascaded Pyramid Network (CPN) based on Keras (Tensorflow backend)

Transformer based SAR image despeckling

This repo contains the implementation of YOLOv2 in Keras with Tensorflow backend.

Steer OpenAI's Jukebox with Music Taggers

Code for Reciprocal Adversarial Learning for Brain Tumor Segmentation: A Solution to BraTS Challenge 2021 Segmentation Task

Kaggle Ultrasound Nerve Segmentation competition [Keras]

Anomaly Detection Based on Hierarchical Clustering of Mobile Robot Data

Json2Xml tool will help you convert from json COCO format to VOC xml format in Object Detection Problem.

Pytorch implementation of paper "Learning Co-segmentation by Segment Swapping for Retrieval and Discovery"

PConv-Keras - Unofficial implementation of "Image Inpainting for Irregular Holes Using Partial Convolutions". Try at: www.fixmyphoto.ai

Hypersearch weight debugging and losses tutorial

An end-to-end machine learning library to directly optimize AUC loss

PyTorch-Multi-Style-Transfer - Neural Style and MSG-Net

Gin provides a lightweight configuration framework for Python

CTF challenges and write-ups for MicroCTF 2021.

Using this codebase as a tool for my own research. Making some modifications to the original repo for my own purposes.

Kernel Point Convolutions

DLL: Direct Lidar Localization