TEAL: New Selection Strategy for Small Buffers in Experience Replay Class Incremental Learning

Shahar Shaul-Ariel, Daphna Weinshall
6/30/2024
cs.LG

Official code: shahariel/TEAL (18 stars, last commit 2025-01-21).

Abstract

Continual Learning is an unresolved challenge, whose relevance increases when considering modern applications. Unlike the human brain, trained deep neural networks suffer from a phenomenon called catastrophic forgetting, wherein they progressively lose previously acquired knowledge upon learning new tasks. To mitigate this problem, numerous methods have been developed, many relying on the replay of past exemplars during new task training. However, as the memory allocated for replay decreases, the effectiveness of these approaches diminishes. On the other hand, maintaining a large memory for the purpose of replay is inefficient and often impractical. Here we introduce TEAL, a novel approach to populate the memory with exemplars, that can be integrated with various experience-replay methods and significantly enhance their performance with small memory buffers. We show that TEAL enhances the average accuracy of existing class-incremental methods and outperforms other selection strategies, achieving state-of-the-art performance even with small memory buffers of 1-3 exemplars per class in the final task. This confirms our initial hypothesis that when memory is scarce, it is best to prioritize the most typical data. Code is available at this https URL: https://github.com/shahariel/TEAL.

View on arXivView PDF

Code Implementations(1)

shahariel/TEALOfficial100%

TEAL: New Selection Strategy for Small Buffers in Experience Replay Class Incremental Learning

180PythonJun 24, 20241 years ago
computer-visioncontinual-learningdeep-learningdeep-neural-networksmachine-learning

Similar Papers

Cite this paper

@article{shaulariel2024teal,
  title  = {TEAL: New Selection Strategy for Small Buffers in Experience Replay Class Incremental Learning},
  author = {Shahar Shaul-Ariel and Daphna Weinshall},
  year   = {2024},
  eprint = {2407.00673},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2407.00673v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2407.00673.svg)](https://codeofpaper.com/paper/2407.00673)

Discussion