SD-LoRA: Scalable Decoupled Low-Rank Adaptation for Class Incremental Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wu, Yichen, Piao, Hongming, Huang, Long-Kai, Wang, Renzhen, Li, Wanhua, Pfister, Hanspeter, Meng, Deyu, Ma, Kede, Wei, Ying
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916645555929088
author Wu, Yichen
Piao, Hongming
Huang, Long-Kai
Wang, Renzhen
Li, Wanhua
Pfister, Hanspeter
Meng, Deyu
Ma, Kede
Wei, Ying
author_facet Wu, Yichen
Piao, Hongming
Huang, Long-Kai
Wang, Renzhen
Li, Wanhua
Pfister, Hanspeter
Meng, Deyu
Ma, Kede
Wei, Ying
contents Continual Learning (CL) with foundation models has recently emerged as a promising paradigm to exploit abundant knowledge acquired during pre-training for tackling sequential tasks. However, existing prompt-based and Low-Rank Adaptation-based (LoRA-based) methods often require expanding a prompt/LoRA pool or retaining samples of previous tasks, which poses significant scalability challenges as the number of tasks grows. To address these limitations, we propose Scalable Decoupled LoRA (SD-LoRA) for class incremental learning, which continually separates the learning of the magnitude and direction of LoRA components without rehearsal. Our empirical and theoretical analysis reveals that SD-LoRA tends to follow a low-loss trajectory and converges to an overlapping low-loss region for all learned tasks, resulting in an excellent stability-plasticity trade-off. Building upon these insights, we introduce two variants of SD-LoRA with further improved parameter efficiency. All parameters of SD-LoRAs can be end-to-end optimized for CL objectives. Meanwhile, they support efficient inference by allowing direct evaluation with the finally trained model, obviating the need for component selection. Extensive experiments across multiple CL benchmarks and foundation models consistently validate the effectiveness of SD-LoRA. The code is available at https://github.com/WuYichen-97/SD-Lora-CL.
format Preprint
id arxiv_https___arxiv_org_abs_2501_13198
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SD-LoRA: Scalable Decoupled Low-Rank Adaptation for Class Incremental Learning
Wu, Yichen
Piao, Hongming
Huang, Long-Kai
Wang, Renzhen
Li, Wanhua
Pfister, Hanspeter
Meng, Deyu
Ma, Kede
Wei, Ying
Machine Learning
Continual Learning (CL) with foundation models has recently emerged as a promising paradigm to exploit abundant knowledge acquired during pre-training for tackling sequential tasks. However, existing prompt-based and Low-Rank Adaptation-based (LoRA-based) methods often require expanding a prompt/LoRA pool or retaining samples of previous tasks, which poses significant scalability challenges as the number of tasks grows. To address these limitations, we propose Scalable Decoupled LoRA (SD-LoRA) for class incremental learning, which continually separates the learning of the magnitude and direction of LoRA components without rehearsal. Our empirical and theoretical analysis reveals that SD-LoRA tends to follow a low-loss trajectory and converges to an overlapping low-loss region for all learned tasks, resulting in an excellent stability-plasticity trade-off. Building upon these insights, we introduce two variants of SD-LoRA with further improved parameter efficiency. All parameters of SD-LoRAs can be end-to-end optimized for CL objectives. Meanwhile, they support efficient inference by allowing direct evaluation with the finally trained model, obviating the need for component selection. Extensive experiments across multiple CL benchmarks and foundation models consistently validate the effectiveness of SD-LoRA. The code is available at https://github.com/WuYichen-97/SD-Lora-CL.
title SD-LoRA: Scalable Decoupled Low-Rank Adaptation for Class Incremental Learning
topic Machine Learning
url https://arxiv.org/abs/2501.13198