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

QAHOI: Query-Based Anchors for Human-Object Interaction Detection (paper)

Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)

PyTorch CZSL framework containing GQA, the open-world setting, and the CGE and CompCos methods.

Bayesian-Torch is a library of neural network layers and utilities extending the core of PyTorch to enable the user to perform stochastic variational inference in Bayesian deep neural networks

Adversarial Reweighting for Partial Domain Adaptation

A PyTorch Implementation of Single Shot MultiBox Detector

Classification models 1D Zoo - Keras and TF.Keras

Keyword-BERT: Keyword-Attentive Deep Semantic Matching

A Collection of Papers and Codes for ICCV2021 Low Level Vision and Image Generation

Per-Pixel Classification is Not All You Need for Semantic Segmentation

Making Structure-from-Motion (COLMAP) more robust to symmetries and duplicated structures

A project to build an AI voice assistant using Python . The Voice assistant interacts with the humans to perform basic tasks.

Fast SHAP value computation for interpreting tree-based models

Segmentation Training Pipeline

A Re-implementation of the paper "A Deep Learning Framework for Character Motion Synthesis and Editing"

Implementation for paper "Towards the Generalization of Contrastive Self-Supervised Learning"

Multi-Agent Reinforcement Learning (MARL) method to learn scalable control polices for multi-agent target tracking.

FlexConv: Continuous Kernel Convolutions with Differentiable Kernel Sizes

Time Dependent DFT in Tamm-Dancoff Approximation

A flexible submap-based framework towards spatio-temporally consistent volumetric mapping and scene understanding.