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

Keeping it safe - AI Based COVID-19 Tracker using Deep Learning and facial recognition

Training, generation, and analysis code for Learning Particle Physics by Example: Location-Aware Generative Adversarial Networks for Physics

Predict the latency time of the deep learning models

Python package to add text to images, textures and different backgrounds

🔥 Cogitare - A Modern, Fast, and Modular Deep Learning and Machine Learning framework for Python

Implement object segmentation on images using HOG algorithm proposed in CVPR 2005

Code Release for ICCV 2021 (oral), "AdaFit: Rethinking Learning-based Normal Estimation on Point Clouds"

Efficient Training of Audio Transformers with Patchout

PyTorch Implementation of Vector Quantized Variational AutoEncoders.

StyleMapGAN - Official PyTorch Implementation

Joint Gaussian Graphical Model Estimation: A Survey

Unofficial implementation of Alias-Free Generative Adversarial Networks. (https://arxiv.org/abs/2106.12423) in PyTorch

Do Neural Networks for Segmentation Understand Insideness?

Dataset used in "PlantDoc: A Dataset for Visual Plant Disease Detection" accepted in CODS-COMAD 2020

Open-CyKG: An Open Cyber Threat Intelligence Knowledge Graph

Code for "Neural Parts: Learning Expressive 3D Shape Abstractions with Invertible Neural Networks", CVPR 2021

[ACMMM 2021 Oral] Enhanced Invertible Encoding for Learned Image Compression

PromptDet: Expand Your Detector Vocabulary with Uncurated Images

Efficient semidefinite bounds for multi-label discrete graphical models.

Retrieve and analysis data from SDSS (Sloan Digital Sky Survey)