Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

HPCR: Holistic Proxy-based Contrastive Replay for Online Continual Learning

About

Online continual learning, aimed at developing a neural network that continuously learns new data from a single pass over an online data stream, generally suffers from catastrophic forgetting. Existing replay-based methods alleviate forgetting by replaying partial old data in a proxy-based or contrastive-based replay manner, each with its own shortcomings. Our previous work proposes a novel replay-based method called proxy-based contrastive replay (PCR), which handles the shortcomings by achieving complementary advantages of both replay manners. In this work, we further conduct gradient and limitation analysis of PCR. The analysis results show that PCR still can be further improved in feature extraction, generalization, and anti-forgetting capabilities of the model. Hence, we develop a more advanced method named holistic proxy-based contrastive replay (HPCR). HPCR consists of three components, each tackling one of the limitations of PCR. The contrastive component conditionally incorporates anchor-to-sample pairs to PCR, improving the feature extraction ability. The second is a temperature component that decouples the temperature coefficient into two parts based on their gradient impacts and sets different values for them to enhance the generalization ability. The third is a distillation component that constrains the learning process with additional loss terms to improve the anti-forgetting ability. Experiments on four datasets consistently demonstrate the superiority of HPCR over various state-of-the-art methods.

Huiwei Lin, Shanshan Feng, Baoquan Zhang, Xutao Li, Yunming Ye• 2023

Related benchmarks

TaskDatasetResultRank
Online Class-Incremental LearningCIFAR-100 10 tasks (test)
FAA (%)36.4
115
Online Class-Incremental LearningImageNet-100 10 tasks (test)
FAA (%)28
21
Showing 2 of 2 rows

Other info

Follow for update