/Tree-CNN

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Tree-CNN

In this repository I implement 'Tree-CNN: A Hierarchical Deep Convolutional Neural Network for Incremental Learning' method to permit continuous learning without forgeting old skills (catastrophic forgetting problem).

Referece:

Roy, D., Panda, P., & Roy, K. (2020). Tree-CNN: A hierarchical Deep Convolutional Neural Network for incremental learning. Neural Networks, 121, 148-160.

Dependencies

  • python 3.5
  • tensorflow 2.0.0
  • numpy
  • pandas
  • jupyter