Cross-Class Feature Augmentation for Class Incremental Learning

Authors

  • Taehoon Kim Seoul National University
  • Jaeyoo Park Seoul National University
  • Bohyung Han Seoul National University

DOI:

https://doi.org/10.1609/aaai.v38i12.29216

Keywords:

ML: Life-Long and Continual Learning, ML: Representation Learning

Abstract

We propose a novel class incremental learning approach, which incorporates a feature augmentation technique motivated by adversarial attacks. We employ a classifier learned in the past to complement training examples of previous tasks. The proposed approach has an unique perspective to utilize the previous knowledge in class incremental learning since it augments features of arbitrary target classes using examples in other classes via adversarial attacks on a previously learned classifier. By allowing the Cross-Class Feature Augmentations (CCFA), each class in the old tasks conveniently populates samples in the feature space, which alleviates the collapse of the decision boundaries caused by sample deficiency for the previous tasks, especially when the number of stored exemplars is small. This idea can be easily incorporated into existing class incremental learning algorithms without any architecture modification. Extensive experiments on the standard benchmarks show that our method consistently outperforms existing class incremental learning methods by significant margins in various scenarios, especially under an environment with an extremely limited memory budget.

Downloads

Published

2024-03-24

How to Cite

Kim, T., Park, J., & Han, B. (2024). Cross-Class Feature Augmentation for Class Incremental Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 38(12), 13168–13176. https://doi.org/10.1609/aaai.v38i12.29216

Issue

Section

AAAI Technical Track on Machine Learning III