arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2403.18886cs.LGcs.CV

预训练模型的自扩展与适配器混合用于持续学习

Self-Expansion of Pre-trained Models with Mixture of Adapters for Continual Learning

University of New South Wales(新南威尔士大学);CSIRO’s Data61(澳大利亚联邦科学与工业研究组织Data61)

Huiyi Wang, Haodong Lu, Lina Yao, Dong Gong

AI总结:

提出SEMA方法,通过模块化适配器自动检测分布偏移并决定重用或添加适配器,实现次线性扩展,提升持续学习性能。

摘要:

持续学习(CL)旨在从非平稳数据流中持续积累知识,同时避免灾难性遗忘已学知识,这需要在稳定性和适应性之间取得平衡。基于预训练模型(PTMs)的通用表示,PTM-based CL方法通过在冻结的PTMs上添加可学习的适配器或提示,对下游任务进行有效的持续适应。然而,许多现有的PTM-based CL方法在固定的模块集合上使用受限的适应以避免遗忘,导致CL能力受限。周期性地添加任务特定模块会导致模型线性增长和知识重用受损。我们提出了SEMA(Self-Expansion of pre-trained models with Modularized Adaptation),一种新颖的方法,以增强PTM-based CL中稳定性-塑性平衡的控制。SEMA根据在不同表示级别上是否检测到无法处理的显著分布偏移,自动决定在CL中重用或按需添加适配器模块。我们设计了模块化适配器,由功能适配器和表示描述符组成。表示描述符被训练为分布偏移指示器,并用于触发自扩展信号。为了更好地组合适配器,我们联合学习了一个可扩展的加权路由器,用于混合适配器输出。SEMA实现了更好的知识重用和次线性扩展率。大量实验证明了所提出的自扩展方法的有效性,在无记忆重放的PTM-based CL方法中达到了最先进的性能。代码可在https://github.com/huiyiwang01/SEMA-CL获取。

英文摘要:

Continual learning (CL) aims to continually accumulate knowledge from a non-stationary data stream without catastrophic forgetting of learned knowledge, requiring a balance between stability and adaptability. Relying on the generalizable representation in pre-trained models (PTMs), PTM-based CL methods perform effective continual adaptation on downstream tasks by adding learnable adapters or prompts upon the frozen PTMs. However, many existing PTM-based CL methods use restricted adaptation on a fixed set of these modules to avoid forgetting, suffering from limited CL ability. Periodically adding task-specific modules results in linear model growth rate and impaired knowledge reuse. We propose Self-Expansion of pre-trained models with Modularized Adaptation (SEMA), a novel approach to enhance the control of stability-plasticity balance in PTM-based CL. SEMA automatically decides to reuse or add adapter modules on demand in CL, depending on whether significant distribution shift that cannot be handled is detected at different representation levels. We design modular adapter consisting of a functional adapter and a representation descriptor. The representation descriptors are trained as a distribution shift indicator and used to trigger self-expansion signals. For better composing the adapters, an expandable weighting router is learned jointly for mixture of adapter outputs. SEMA enables better knowledge reuse and sub-linear expansion rate. Extensive experiments demonstrate the effectiveness of the proposed self-expansion method, achieving state-of-the-art performance compared to PTM-based CL methods without memory rehearsal. Code is available at https://github.com/huiyiwang01/SEMA-CL.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑