Learning to Grow Artificial Hippocampi in Vision Transformers for Resilient Lifelong Learning

Savadikar, Chinmay; Dai, Michelle; Wu, Tianfu

doi:10.48550/arxiv.2303.08250

Cited by 1 publication

(1 citation statement)

References 50 publications

Supporting

Mentioning

Contrasting

Order By: Relevance

“…The authors in [95] introduced a universal sleep decoder to align neural representations between wakefulness and sleep. The authors in [96] drew inspiration from the hippocampus's role in the human brain, notably its function in transitioning short-term memory to long-term memory, to implement Artificial Hippocampi for lifelong learning. The authors in [97] emphasized replay-based echoing biological memory mechanisms, while the biological brain engages in selective memory retention of salient experiences, and introduced a framework that combines associative memory with replay-based strategies, thereby bringing CL closer to human memory processes.…”

Section: Related Workmentioning

confidence: 99%

Brain-Inspired Continual Learning: Robust Feature Distillation and Re-Consolidation for Class Incremental Learning

Khan,

Bouaynaya,

Rasool

2024

IEEE Access

View full text Add to dashboard Cite

Artificial intelligence and neuroscience have a long and intertwined history. Advancements in neuroscience research have significantly influenced the development of artificial intelligence systems that have the potential to retain knowledge akin to humans. Building upon foundational insights from neuroscience and existing research in adversarial and continual learning fields, we introduce a novel framework that comprises two key concepts: feature distillation and re-consolidation. The framework distills continual learning (CL) robust features and rehearses them while learning the next task, aiming to replicate the mammalian brain's process of consolidating memories through rehearsing the distilled version of the waking experiences. Furthermore, the proposed framework emulates the mammalian brain's mechanism of memory re-consolidation, where novel experiences influence the assimilation of previous experiences via feature re-consolidation. This process incorporates the new understanding of the CL model after learning the current task into the CL-robust samples of the previous task(s) to mitigate catastrophic forgetting. The proposed framework, called Robust Rehearsal, circumvents the limitations of existing CL frameworks that rely on the availability of pre-trained Oracle CL models to pre-distill CL-robustified datasets for training subsequent CL models. We conducted extensive experiments on three datasets, CIFAR10, CIFAR100, and real-world helicopter attitude datasets, demonstrating that CL models trained using Robust Rehearsal outperform their counterparts' baseline methods. In addition, we conducted a series of experiments to assess the impact of changing memory sizes and the number of tasks, demonstrating that the baseline methods employing robust rehearsal outperform other methods trained without robust rehearsal. Lastly, to shed light on the existence of diverse features, we explore the effects of various optimization training objectives within the realms of joint, continual, and adversarial learning on feature learning in deep neural networks. Our findings indicate that the optimization objective dictates feature learning, which plays a vital role in model performance. Such observation further emphasizes the importance of rehearsing the CL-robust samples in alleviating catastrophic forgetting. In light of our experiments, closely following neuroscience insights can contribute to developing CL approaches to mitigate the long-standing challenge of catastrophic forgetting. INDEX TERMSContinual learning, neuroscience-inspired, brain-inspired, catastrophic forgetting, feature distillation, feature re-consolidation, class-incremental learning, rehearsal-based learning strategies. I. INTRODUCTION Continual Learning (CL), also referred to as incremental, lifelong, or sequential learning, equips deep learning models with the ability to accumulate and expand knowledge over time, similar to humans [1]-[3]. Despite advancements in CL methodologies, current approaches still suffer from a phenomenon known as catastrophic ...

show abstract

Section: Related Workmentioning

confidence: 99%