May the Forgetting Be with You: Alternate Replay for Learning with Noisy Labels

Monica Millunzi, Lorenzo Bonicelli, Angelo Porrello, Jacopo Credi, Petter N. Kolm +1 more
8/26/2024
cs.LGcs.AIcs.CV

Official code: aimagelab/mammoth (841 stars, MIT, PyTorch, last commit 2026-04-20).

Abstract

Forgetting presents a significant challenge during incremental training, making it particularly demanding for contemporary AI systems to assimilate new knowledge in streaming data environments. To address this issue, most approaches in Continual Learning (CL) rely on the replay of a restricted buffer of past data. However, the presence of noise in real-world scenarios, where human annotation is constrained by time limitations or where data is automatically gathered from the web, frequently renders these strategies vulnerable. In this study, we address the problem of CL under Noisy Labels (CLN) by introducing Alternate Experience Replay (AER), which takes advantage of forgetting to maintain a clear distinction between clean, complex, and noisy samples in the memory buffer. The idea is that complex or mislabeled examples, which hardly fit the previously learned data distribution, are most likely to be forgotten. To grasp the benefits of such a separation, we equip AER with Asymmetric Balanced Sampling (ABS): a new sample selection strategy that prioritizes purity on the current task while retaining relevant samples from the past. Through extensive computational comparisons, we demonstrate the effectiveness of our approach in terms of both accuracy and purity of the obtained buffer, resulting in a remarkable average gain of 4.71% points in accuracy with respect to existing loss-based purification strategies. Code is available at https://github.com/aimagelab/mammoth.

View on arXivView PDF

Code Implementations(1)

aimagelab/mammothOfficial100%
PyTorchMIT

An Extendible (General) Continual Learning Framework based on Pytorch - official codebase of Dark Experience for General Continual Learning

841157PythonOct 5, 20206 months agoMIT
continual-learningdark-experience-replaydeep-learningderexperience-replay+3 more

Similar Papers

Cite this paper

@article{millunzi2024forgetting,
  title  = {May the Forgetting Be with You: Alternate Replay for Learning with Noisy Labels},
  author = {Monica Millunzi and Lorenzo Bonicelli and Angelo Porrello and Jacopo Credi and Petter N. Kolm and Simone Calderara},
  year   = {2024},
  eprint = {2408.14284},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2408.14284v1}
}

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/2408.14284.svg)](https://codeofpaper.com/paper/2408.14284)

Discussion