arrow
Return

Tuned Compositional Feature Replays for Efficient Stream Learning

delete2025-02-01
delete0
delete
OA
AI
M
Morgan B. Talbot
R
Rushikesh Zawar
R
Rohil Badkundri
M
Mengmi Zhang
G
Gabriel Kreiman *
DOI:10.1109/TNNLS.2023.3344085delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Our brains extract durable, generalizable knowledge from transient experiences of the world. Artificial neural networks come nowhere close to this ability. When tasked with learning to classify objects by training on nonrepeating video frames in temporal order (online stream learning), models that learn well from shuffled datasets catastrophically forget old knowledge upon learning new stimuli. We propose a new continual learning algorithm, compositional replay using memory blocks (CRUMB), which mitigates forgetting by replaying feature maps reconstructed by combining generic parts. CRUMB concatenates trainable and reusable memory block vectors to compositionally reconstruct feature map tensors in convolutional neural networks (CNNs). Storing the indices of memory blocks used to reconstruct new stimuli enables memories of the stimuli to be replayed during later tasks. This reconstruction mechanism also primes the neural network to minimize catastrophic forgetting by biasing it toward attending to information about object shapes more than information about image textures and stabilizes the network during stream learning by providing a shared feature-level basis for all training examples. These properties allow CRUMB to outperform an otherwise identical algorithm that stores and replays raw images while occupying only 3.6% as much memory. We stress-tested CRUMB alongside 13 competing methods on seven challenging datasets. To address the limited number of existing online stream learning datasets, we introduce two new benchmarks by adapting existing datasets for stream learning. With only 3.7%-4.1% as much memory and 15%-43% as much runtime, CRUMB mitigates catastrophic forgetting more effectively than the state-of-the-art. Our code is available at https://github.com/MorganBDT/crumb.git
Keywords:
Brain-inspired replay
catastrophic forgetting
deep learning
stream learning

Journal

IEEE Transactions on Neural Networks and Learning Systems cover
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
Papers:
7.6K
Citations:
7.2W

Organization

H
Harvard University
Scholars:
26.5W
Papers: 22.0W
Citations: 28.7W
Cited Papers

Cited Papers

err
IF0
err
err0
PREAI
err
errShare
errSave
Excessive game playing is associated with musculoskeletal pain among youth athletes: a cross-sectional study in Miyagi prefecture
err2017-12-22
err0
PREAI
errTakuya Sekiguchi; Yoshihiro Hagiwara; Haruki Momma; Masahiro Tsuchiya; Kaoru Kuroki; Kenji Kanazawa; Yutaka Yabe; Shinichiro Yoshida; Masashi Koide; Nobuyuki Itaya; Eiji Itoi; Ryoichi Nagatomi
errShare
errSave
Comparison of test methods for the determination of delamination in glued laminated timber
err2019-08-15
err0
errOAAI
errRodney Vella; Michael T. Heitzmann; Adam Redman; Henri Bailleres
errShare
errSave
Memory Recall: A Simple Neural Network Training Framework Against Catastrophic Forgetting
err2022-05-01
err16
PREAI
errZhang, Baosheng; Guo, Yuchen; Li, Yipeng; He, Yuwei; Wang, Haoqian; Dai, Qionghai
errShare
errSave
researcher View more