Skip to content
Snippets Groups Projects
user avatar
hl00586 authored
c7b1ac98
History

Continual Learning Using Task ConditionalNeural Networks

Introduction

Task Conditional Neural Networks (TCNN) leverage the probabilistic neural networks to estimate the probability density of the training samples. Then produce the task likelihood during the test state to fire the task-specific neurons correspondin to the test sampels. TCNN can detect and learn the new tasks fully-automatically without informing the changes to the model.

The schematic diagram of TCNN learning process. TCNN_Process

Dependencies

The code is tested on

  • Python 3.6.8
  • Tensorflow 1.13.1
  • Keras 2.2.4

Reproduce

To run the experiment, run the split_mnist.py by:

python split_mnist.py

The result should be similar to the figure below:

TCNN_Process