arrow
Return

Joint Memory Optimization for Continual Learning

delete2025-07-01
delete0
PRE
AI
Z
Zhiheng Ma
Y
Yaohui Ma
X
Xiaopeng Hong
H
Huiyun Li
S
Shizhou Zhang
DOI:10.1109/TCSVT.2025.3539661delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Continual learning, focusing on sequential knowledge acquisition and retention, necessitates efficient memory management. This paper introduces a holistic approach, diverging from traditional methods that separately optimize neural network and replay buffer memory. We aim to enhance overall memory efficiency, addressing neural network parameters and replay buffer concurrently within strict memory constraints. This is achieved by harnessing neural network parameter redundancies and employing compression techniques like pruning and quantization, allowing data replay storage without extra memory overhead. Balancing memory use across components is challenging due to the complex search space of combined tasks. We tackle this by conceptualizing it as a bi-level optimization problem, integrating all tasks under a single objective, thus optimizing memory use and managing the interplay between different components. We employ a synergy of optimization techniques to solve this challenging bi-level optimization problem. Our experimental findings affirm the superior performance of our proposed method, outperforming existing techniques such as prompt-based, feature-replay, exemplar-replay, and regularization-based methods under stringent memory constraints, consistently across various datasets and neural network architectures.
Keywords:
Continual learning
model compression

Journal

IEEE Transactions on Circuits and Systems for Video Technology cover
IEEE Transactions on Circuits and Systems for Video Technology
IF:
11.1
Papers:
873
Citations:
3.1W

Organization

H
harbin institute of technology
Scholars:
8.0W
Papers: 6.6W
Citations: 66
S
Shenzhen University of Advanced Technology
Scholars:
339
Papers: 330
Citations: 1
N
Northwestern Polytechnical University
Scholars:
4.6W
Papers: 3.7W
Citations: 5.3W
researcher View more organizations
Cited Papers

Cited Papers

Learning to Prompt for Continual Learning
err2022-06-01
err0
errOAAI
errZifeng Wang; Zizhao Zhang; Chen-Yu Lee; Han Zhang; Ruoxi Sun; Xiaoqi Ren; Guolong Su; Vincent Perot; Jennifer Dy; Tomas Pfister
errShare
errSave
err
IF0
err
err0
errOAAI
err
errShare
errSave
CODA-Prompt: COntinual Decomposed Attention-Based Prompting for Rehearsal-Free Continual Learning
err2023-06-01
err0
errOAAI
errJames Seale Smith; Leonid Karlinsky; Vyshnavi Gutta; Paola Cascante-Bonilla; Donghyun Kim; Assaf Arbelle; Rameswar Panda; Rogerio Feris; Zsolt Kira
errShare
errSave
The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalization
err2021-10-01
err0
errOAAI
errDan Hendrycks; Steven Basart; Norman Mu; Saurav Kadavath; Frank Wang; Evan Dorundo; Rahul Desai; Tyler Zhu; Samyak Parajuli; Mike Guo; Dawn Song; Jacob Steinhardt; Justin Gilmer
errShare
errSave
Adversarial Continual Learning
err2020-11-27
err0
PREAI
errSayna Ebrahimi; Franziska Meier; Roberto Calandra; Trevor Darrell; Marcus Rohrbach
errShare
errSave
Exemplar-Based Continual Learning via Contrastive Learning
err2024-07-01
err0
PREAI
errShuai Chen; Mingyi Zhang; Junge Zhang; Kaiqi Huang
errShare
errSave
researcher View more