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

Build Graph Nets in Tensorflow

This repository contains the needed resources to build the HIRID-ICU-Benchmark dataset

Code for ACL'2021 paper WARP 🌀 Word-level Adversarial ReProgramming

3rd place solution for the Weather4cast 2021 Stage 1 Challenge

Fast (simple) spectral synthesis and emission-line fitting of DESI spectra.

An easier way to build neural search on the cloud

Code for paper: Towards Tokenized Human Dynamics Representation

Implementation of SSMF: Shifting Seasonal Matrix Factorization

Code for our paper 'Generalized Category Discovery'

Pytorch implementation of NeurIPS 2021 paper: Geometry Processing with Neural Fields.

Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning.

StyleSwin: Transformer-based GAN for High-resolution Image Generation

Bridging the Gap between Label- and Reference based Synthesis(ICCV 2021)

A very impractical 3D rendering engine that runs in the python terminal.

Randstad Artificial Intelligence Challenge (powered by VGEN). Soluzione proposta da Stefano Fiorucci (anakin87) - primo classificato

Code for Deterministic Neural Networks with Appropriate Inductive Biases Capture Epistemic and Aleatoric Uncertainty

CARLA: A Python Library to Benchmark Algorithmic Recourse and Counterfactual Explanation Algorithms

Non-Homogeneous Poisson Process Intensity Modeling and Estimation using Measure Transport

Deep motion generator collections

Training PSPNet in Tensorflow. Reproduce the performance from the paper.