Active Long Term Memory Networks

TL;DR

This paper introduces Active Long Term Memory Networks (A-LTM), a continual learning model that preserves prior knowledge while acquiring new information by actively maintaining inactive tasks using distillation, addressing catastrophic inference.

Contribution

A-LTM leverages the non-convex nature of neural networks and distillation loss to prevent forgetting, offering a novel approach to multi-task continual learning and domain adaptation.

Findings

A-LTM reduces catastrophic inference in deep linear networks.

A-LTM maintains viewpoint recognition across diverse datasets.

Empirical results show improved knowledge retention during domain shifts.

Abstract

Continual Learning in artificial neural networks suffers from interference and forgetting when different tasks are learned sequentially. This paper introduces the Active Long Term Memory Networks (A-LTM), a model of sequential multi-task deep learning that is able to maintain previously learned association between sensory input and behavioral output while acquiring knew knowledge. A-LTM exploits the non-convex nature of deep neural networks and actively maintains knowledge of previously learned, inactive tasks using a distillation loss. Distortions of the learned input-output map are penalized but hidden layers are free to transverse towards new local optima that are more favorable for the multi-task objective. We re-frame the McClelland's seminal Hippocampal theory with respect to Catastrophic Inference (CI) behavior exhibited by modern deep architectures trained with back-propagation and inhomogeneous sampling of latent factors across epochs. We present empirical results of non-trivial CI during continual learning in Deep Linear Networks trained on the same task, in Convolutional Neural Networks when the task shifts from predicting semantic to graphical factors and during domain adaptation from simple to complex environments. We present results of the A-LTM model's ability to maintain viewpoint recognition learned in the highly controlled iLab-20M dataset with 10 object categories and 88 camera viewpoints, while adapting to the unstructured domain of Imagenet with 1,000 object categories.