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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Millunzi, Monica, Bonicelli, Lorenzo, Porrello, Angelo, Credi, Jacopo, Kolm, Petter N., Calderara, Simone
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913480802566144
author Millunzi, Monica
Bonicelli, Lorenzo
Porrello, Angelo
Credi, Jacopo
Kolm, Petter N.
Calderara, Simone
author_facet Millunzi, Monica
Bonicelli, Lorenzo
Porrello, Angelo
Credi, Jacopo
Kolm, Petter N.
Calderara, Simone
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2408_14284
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle May the Forgetting Be with You: Alternate Replay for Learning with Noisy Labels
Millunzi, Monica
Bonicelli, Lorenzo
Porrello, Angelo
Credi, Jacopo
Kolm, Petter N.
Calderara, Simone
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
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.
title May the Forgetting Be with You: Alternate Replay for Learning with Noisy Labels
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.14284