Multi-criteria Selection of Rehearsal Samples for Continual Learning

•We present a multi-criteria subset selection strategy to overcome the unstable learning issue of using singular criteria.•Two novel subset selection strategies are introduced: intra-class cluster variation and classifier loss to replay-based continual learning framework.•Proposed method achieves ne...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Pattern recognition 2022-12, Vol.132, p.108907, Article 108907
Hauptverfasser: Zhuang, Chen, Huang, Shaoli, Cheng, Gong, Ning, Jifeng
Format: Artikel
Sprache:eng
Schlagworte:
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:•We present a multi-criteria subset selection strategy to overcome the unstable learning issue of using singular criteria.•Two novel subset selection strategies are introduced: intra-class cluster variation and classifier loss to replay-based continual learning framework.•Proposed method achieves new state-of-the-art results and provides an insight into sample selection. Retaining a small subset to replay is a direct and effective way to prevent catastrophic forgetting in continual learning. However, due to data complexity and restricted memory, picking a proper subset for rehearsal is challenging and still being explored. In this work, we present a Multi-criteria Subset Selection approach that can stabilize and advance replay-based continual learning. The method picks rehearsal samples by integrating multiple criteria, including distance to prototype, intra-class cluster variation, and classifier loss. By doing so, it maximizes the comprehensive representation power of the sampled subset by ensuring its representativeness, diversity, and discriminability. We empirically find that singular criteria are likely to fail in particular tasks, while multi-criteria minimizes this risk and stabilizes task training throughout the continual learning process. Moreover, our method improves replay-based methods consistently and achieves state-of-the-art performance on both CIFAR100 and Tiny-Imagenet datasets.
ISSN:0031-3203
1873-5142
DOI:10.1016/j.patcog.2022.108907