Energy-based Latent Aligner for Incremental Learning

Citations Over TimeTop 10% of 2022 papers

Abstract

Deep learning models tend to forget their earlier knowledge while incrementally learning new tasks. This behavior emerges because the parameter updates optimized for the new tasks may not align well with the updates suitable for older tasks. The resulting latent representation mismatch causes forgetting. In this work, we propose ELI: Energy-based Latent Aligner for Incremental Learning, which first learns an energy manifold for the latent representations such that previous task latents will have low energy and the current task latents have high energy values. This learned manifold is used to counter the representational shift that happens during incremental learning. The implicit regularization that is offered by our proposed methodology can be used as a plug-and-play module in existing incremental learning methodologies. We validate this through extensive evaluation on CIFAR-100, ImageNet subset, ImageNet1k and Pascal VOC datasets. We observe consistent improvement when ELI is added to three prominent methodologies in class-incremental learning, across multiple incremental settings. Further, when added to the state-of-the-art incremental object detector, ELI provides over 5% improvement in detection accuracy, corroborating its effectiveness and complementary advantage to the existing art. Code is available at: https://github.com/JosephKJ/ELI.

Related Papers

• → Incremental learning with open set based discrimination enhancement(2021)5 cited
• → Dynamic Replay Training for Class-Incremental Learning(2024)3 cited
• → In Defense of the Learning Without Forgetting for Task Incremental Learning(2021)3 cited
• → Feature Forgetting in Continual Representation Learning(2022)2 cited
• → In Defense of the Learning Without Forgetting for Task Incremental\n Learning(2021)