implementation of the paper "MarginGAN: Adversarial Training in Semi-Supervised Learning"

Last update: Dec 23, 2022

Related tags

Overview

MarginGAN

This repository is the implementation of the paper "MarginGAN: Adversarial Training in Semi-Supervised Learning".

1."preliminary" is the implementation of Preliminary Experiment on MNIST of the paper. Thank the authors of pytorch-generative-model-collections and examples of pytorch, our code is widely adapted from their repositories.

To train the network, an example is as follows:

python main.py \
  --gan_type MarginGAN \
  --num_labels 600 \
  --lrC 0.1 \
  --epoch 50

2."ablation" is the implementation of Ablation Study on MNIST of the paper. To train the network, an example is as follows:

python main.py \
--gan_type MarginGAN_UG \
--num_labels 600 \
--lrC 0.01 \
--epoch 50

3."further" is the implementation of Experiment on SVHN and CIFAR-10 of the paper. Thank the authors of mean teacher, our code is widely adapted from their repositories.

To train the network, an example is as follows:

python MarginGAN_main.py \
    --dataset cifar10 \
    --train-subdir train+val \
    --eval-subdir test \
    --batch-size 128 \
    --labeled-batch-size 31 \
    --arch cifar_shakeshake26 \
    --consistency-type mse \
    --consistency-rampup 5 \
    --consistency 100.0 \
    --logit-distance-cost 0.01 \
    --weight-decay 2e-4 \
    --lr-rampup 0 \
    --lr 0.05 \
    --nesterov True \
    --labels data-local/labels/cifar10/1000_balanced_labels/00.txt  \
    --epochs 180 \
    --lr-rampdown-epochs 210 \
    --ema-decay 0.97 \
    --generated-batch-size 32

implementation of the paper "MarginGAN: Adversarial Training in Semi-Supervised Learning"

Related tags

Overview

MarginGAN

Owner

Van

Code for TIP 2017 paper --- Illumination Decomposition for Photograph with Multiple Light Sources.

Solution of Kaggle competition: Sartorius - Cell Instance Segmentation

A self-supervised learning framework for audio-visual speech

MXNet implementation for: Drop an Octave: Reducing Spatial Redundancy in Convolutional Neural Networks with Octave Convolution

sktime companion package for deep learning based on TensorFlow

My usage of Real-ESRGAN to upscale anime, some test and results in the test_img folder

Implementation of Hierarchical Transformer Memory (HTM) for Pytorch

Code and data for paper "Deep Photo Style Transfer"

Using deep actor-critic model to learn best strategies in pair trading

Practical and Real-world applications of ML based on the homework of Hung-yi Lee Machine Learning Course 2021

MoveNetを用いたPythonでの姿勢推定のデモ

Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"

Code, Data and Demo for Paper: Controllable Generation from Pre-trained Language Models via Inverse Prompting

The Instructed Glacier Model (IGM)

LeetCode Solutions https://t.me/tenvlad

Official implement of Paper：A deeply supervised image fusion network for change detection in high resolution bi-temporal remote sening images

Code for HodgeNet: Learning Spectral Geometry on Triangle Meshes, in SIGGRAPH 2021.

Disentangled Cycle Consistency for Highly-realistic Virtual Try-On, CVPR 2021

PyTorch implementation for our NeurIPS 2021 Spotlight paper "Long Short-Term Transformer for Online Action Detection".

My solution for the 7th place / 245 in the Umoja Hack 2022 challenge