PyTorch implementation of "Continual Learning with Deep Generative Replay", NIPS 2017

Last update: Dec 14, 2022

Overview

pytorch-deep-generative-replay

PyTorch implementation of Continual Learning with Deep Generative Replay, NIPS 2017

Results

Continual Learning on Permutated MNISTs

Test precisions without replay (left), with exact replay (middle), and with Deep Generative Replay (right).

Continual Learning on MNIST-SVHN

Test precisions without replay (left), with exact replay (middle), and with Deep Generative Replay (right).

Generated samples from the scholar trained without any replay (left) and with Deep Generative Replay (right).

Training scholar's generator without replay (left) and with Deep Generative Replay (right).

Continual Learning on SVHN-MNIST

Test precisions without replay (left), with exact replay (middle), and with Deep Generative Replay (right).

Generated samples from the scholar trained without replay (left) and with Deep Generative Replay (right).

Training scholar's generator without replay (left) and with Deep Generative Replay (right).

Installation

$ git clone https://github.com/kuc2477/pytorch-deep-generative-replay
$ pip install -r pytorch-deep-generative-replay/requirements.txt

Commands

Usage

$ ./main.py --help
$ usage: PyTorch implementation of Deep Generative Replay [-h]
                                                          [--experiment {permutated-mnist,svhn-mnist,mnist-svhn}]
                                                          [--mnist-permutation-number MNIST_PERMUTATION_NUMBER]
                                                          [--mnist-permutation-seed MNIST_PERMUTATION_SEED]
                                                          --replay-mode
                                                          {exact-replay,generative-replay,none}
                                                          [--generator-z-size GENERATOR_Z_SIZE]
                                                          [--generator-c-channel-size GENERATOR_C_CHANNEL_SIZE]
                                                          [--generator-g-channel-size GENERATOR_G_CHANNEL_SIZE]
                                                          [--solver-depth SOLVER_DEPTH]
                                                          [--solver-reducing-layers SOLVER_REDUCING_LAYERS]
                                                          [--solver-channel-size SOLVER_CHANNEL_SIZE]
                                                          [--generator-c-updates-per-g-update GENERATOR_C_UPDATES_PER_G_UPDATE]
                                                          [--generator-iterations GENERATOR_ITERATIONS]
                                                          [--solver-iterations SOLVER_ITERATIONS]
                                                          [--importance-of-new-task IMPORTANCE_OF_NEW_TASK]
                                                          [--lr LR]
                                                          [--weight-decay WEIGHT_DECAY]
                                                          [--batch-size BATCH_SIZE]
                                                          [--test-size TEST_SIZE]
                                                          [--sample-size SAMPLE_SIZE]
                                                          [--image-log-interval IMAGE_LOG_INTERVAL]
                                                          [--eval-log-interval EVAL_LOG_INTERVAL]
                                                          [--loss-log-interval LOSS_LOG_INTERVAL]
                                                          [--checkpoint-dir CHECKPOINT_DIR]
                                                          [--sample-dir SAMPLE_DIR]
                                                          [--no-gpus]
                                                          (--train | --test)

To Run Full Experiments

# Run a visdom server and conduct full experiments
$ python -m visdom.server &
$ ./run_full_experiments

To Run a Single Experiment

# Run a visdom server and conduct a desired experiment
$ python -m visdom.server &
$ ./main.py --train --experiment=[permutated-mnist|svhn-mnist|mnist-svhn] --replay-mode=[exact-replay|generative-replay|none]

To Generate Images from the learned Scholar

$ # Run the command below and visit the "samples" directory
$ ./main.py --test --experiment=[permutated-mnist|svhn-mnist|mnist-svhn] --replay-mode=[exact-replay|generative-replay|none]

Note

I failed to find the supplementary materials that the authors mentioned in the paper to contain the experimental details. Thus, I arbitrarily chose a 4-convolutional-layer CNN as a solver in this project. If you know where I can find the additional materials, please let me know through the project's Github issue.

Reference

Continual Learning with Deep Generative Replay, arxiv:1705.08690

Author

Ha Junsoo / @kuc2477 / MIT License

PyTorch implementation of "Continual Learning with Deep Generative Replay", NIPS 2017

Related tags

Overview

pytorch-deep-generative-replay

Results

Continual Learning on Permutated MNISTs

Continual Learning on MNIST-SVHN

Continual Learning on SVHN-MNIST

Installation

Commands

Usage

To Run Full Experiments

To Run a Single Experiment

To Generate Images from the learned Scholar

Note

Reference

Author

Owner

Junsoo Ha

A general-purpose programming language, focused on simplicity, safety and stability.

Angular & Electron desktop UI framework. Angular components for native looking and behaving macOS desktop UI (Electron/Web)

Activating More Pixels in Image Super-Resolution Transformer

Fully convolutional deep neural network to remove transparent overlays from images

A collection of pre-trained StyleGAN2 models trained on different datasets at different resolution.

Codes for NAACL 2021 Paper "Unsupervised Multi-hop Question Answering by Question Generation"

The sixth place winning solution (6/220) in 2021 Gaofen Challenge.

Semantic Bottleneck Scene Generation

A state of the art of new lightweight YOLO model implemented by TensorFlow 2.

Official implementation of NeurIPS 2021 paper "Contextual Similarity Aggregation with Self-attention for Visual Re-ranking"

Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners

Face recognize and crop them

Stacked Hourglass Network with a Multi-level Attention Mechanism: Where to Look for Intervertebral Disc Labeling

Official implementation of "Learning Proposals for Practical Energy-Based Regression", 2021.

Implementations of polygamma, lgamma, and beta functions for PyTorch

Incremental Cross-Domain Adaptation for Robust Retinopathy Screening via Bayesian Deep Learning

Bounding Wasserstein distance with couplings

pyspark🍒🥭 is delicious，just eat it!😋😋

Colab notebook and additional materials for Python-driven analysis of redlining data in Philadelphia

Explaining Hyperparameter Optimization via PDPs