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

Pixel-level Crack Detection From Images Of Levee Systems : A Comparative Study

Code and datasets for the paper "Combining Events and Frames using Recurrent Asynchronous Multimodal Networks for Monocular Depth Prediction" (RA-L, 2021)

ivadomed is an integrated framework for medical image analysis with deep learning.

torchlm is aims to build a high level pipeline for face landmarks detection, it supports training, evaluating, exporting, inference(Python/C++) and 100+ data augmentations

EdiBERT, a generative model for image editing

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening

OrienMask: Real-time Instance Segmentation with Discriminative Orientation Maps

MetaDrive: Composing Diverse Scenarios for Generalizable Reinforcement Learning

StackRec: Efficient Training of Very Deep Sequential Recommender Models by Iterative Stacking

Nonnegative spatial factorization for multivariate count data

Official git repo for the CHIRP project

This repository is a basic Machine Learning train & validation Template (Using PyTorch)

基于AlphaPose的TensorRT加速

Contextual Attention Localization for Offline Handwritten Text Recognition

Data Engineering ZoomCamp

Group R-CNN for Point-based Weakly Semi-supervised Object Detection (CVPR2022)

Reliable probability face embeddings

Hide screen when boss is approaching.

Code and real data for the paper "Counterfactual Temporal Point Processes", available at arXiv.

Learning to Estimate Hidden Motions with Global Motion Aggregation