Abstract
Catastrophic forgetting remains
a core challenge in continual learning (CL), where the models struggle
to retain previous knowledge when learning new tasks. While existing
replay-based CL methods have been proposed to tackle this challenge by
utilizing a memory buffer to store data from previous tasks, they
generally overlook the interdependence between previously learned tasks
and fail to encapsulate the optimally integrated knowledge in previous
tasks, leading to sub-optimal performance of the previous tasks. Against
this issue, we first reformulate replay-based CL methods as a unified
hierarchical gradient aggregation framework. We then incorporate the
Pareto optimization to capture the interrelationship among previously
learned tasks and design a Pareto-Optimized CL algorithm (POCL), which
effectively enhances the overall performance of past tasks while
ensuring the performance of the current task. To further stabilize the
gradients of different tasks, we carefully devise a hyper-gradient-based
implementation manner for POCL. Comprehensive empirical results
demonstrate that the proposed POCL outperforms current state-of-the-art
CL methods across multiple datasets and different settings.
| Original language | English |
|---|---|
| Title of host publication | Proceedings of 41st International Conference on Machine Learning, ICML 2024 |
| Publisher | ML Research Press |
| Pages | 53892-53908 |
| Number of pages | 17 |
| Publication status | Published - Jul 2024 |
| Event | 41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria Duration: 21 Jul 2024 → 27 Jul 2024 https://icml.cc/ https://openreview.net/group?id=ICML.cc/2024/Conference#tab-accept-oral https://proceedings.mlr.press/v235/ |
Publication series
| Name | Proceedings of Machine Learning Research |
|---|---|
| Volume | 235 |
| ISSN (Print) | 2640-3498 |
| Name | Proceedings of the International Conference on Machine Learning |
|---|
Conference
| Conference | 41st International Conference on Machine Learning, ICML 2024 |
|---|---|
| Country/Territory | Austria |
| City | Vienna |
| Period | 21/07/24 → 27/07/24 |
| Internet address |