Skip to content

Files

Latest commit

2435c35 · Jun 7, 2024

History

History
20 lines (13 loc) · 762 Bytes

README.md

File metadata and controls

20 lines (13 loc) · 762 Bytes

Multi-layer Rehearsal Feature Augmentation for Class-Incremental Learning

This is the code repo for Multi-layer Rehearsal Feature Augmentation for Class-Incremental Learning (ICML 2024).

[paper]

TL; DR: We analyze the overfitting for rehearsal samples in class-incremental learning (CIL) from the perspective of all-layer margin. We find that such margin is getting smaller and smaller during the training of CIL. We propose the Multi-layer Rehearsal Feature Augmentation (MRFA) to enlarge the margin of the rehearsal samples during CIL.

teaser

To Run the Code

python main.py --config {config_file_path}

Acknowledgments

This repo uses PyCIL.