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

This tutorial aims to learn the basics of deep learning by hands, and master the basics through combination of lectures and exercises

Lbl2Vec learns jointly embedded label, document and word vectors to retrieve documents with predefined topics from an unlabeled document corpus.

Python tools for 3D face: 3DMM, Mesh processing(transform, camera, light, render), 3D face representations.

Official code repository for A Simple Long-Tailed Rocognition Baseline via Vision-Language Model.

Implementation for Paper "Inverting Generative Adversarial Renderer for Face Reconstruction"

StyleGAN - Official TensorFlow Implementation

Pytorch library for fast transformer implementations

A curated list of the top 10 computer vision papers in 2021 with video demos, articles, code and paper reference.

Only works with the dashboard version / branch of jesse

Like a cowsay but without cows!

Incremental Transformer Structure Enhanced Image Inpainting with Masking Positional Encoding (CVPR2022)

[CVPR 2022] Back To Reality: Weak-supervised 3D Object Detection with Shape-guided Label Enhancement

Detectorch - detectron for PyTorch

A deep learning network built with TensorFlow and Keras to classify gender and estimate age.

A python library to build Model Trees with Linear Models at the leaves.

This is the repository for Learning to Generate Piano Music With Sustain Pedals

HIVE: Evaluating the Human Interpretability of Visual Explanations

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

FS2KToolbox FS2K Dataset Towards the translation between Face

FaRL for Facial Representation Learning