用分层梯度相似树高效实现大模型持续学习,防止遗忘且节省计算。
TreeLoRA: Efficient Continual Learning via Layer-Wise LoRAs Guided by a Hierarchical Gradient-Similarity Tree
- 基于层级梯度相似性构建分层低秩适配器,动态组织任务
- 在ViT和LLM上实现90%以上参数更新效率提升,性能稳定
- 适合资源受限场景下的大模型在线学习,如边缘设备部署
许多真实应用场景中数据以流式方式持续到来,需通过持续学习(CL)在线更新模型,适应新任务同时保留旧知识,避免灾难性遗忘。随着大规模预训练模型(LPMs)的兴起,其庞大的计算开销与参数量使效率成为关键挑战。本文提出TreeLoRA(K-D树型低秩适配器),通过分层梯度相似性构建逐层适配器,实现高效持续学习,尤其适用于LPMs。为降低任务相似性估计的计算负担,采用基于置信下界的探索策略,结合带宽优化的贝叶斯上下文选择方法;同时引入稀疏梯度更新机制,进一步优化参数更新效率。理论分析验证了方法设计合理性。在视觉变压器(ViTs)与大语言模型(LLMs)上的实验表明,该方法在视觉与自然语言处理任务中均具备优异的性能与效率表现,显著减少计算开销。
原文摘要 · Abstract (English)
Many real-world applications collect data in a streaming environment, where learning tasks are encountered sequentially. This necessitates continual learning (CL) to update models online, enabling adaptation to new tasks while preserving past knowledge to prevent catastrophic forgetting. Nowadays, with the flourish of large pre-trained models (LPMs), efficiency has become increasingly critical for CL, due to their substantial computational demands and growing parameter sizes. In this paper, we introduce TreeLoRA (K-D Tree of Low-Rank Adapters), a novel approach that constructs layer-wise adapters by leveraging hierarchical gradient similarity to enable efficient CL, particularly for LPMs. To reduce the computational burden of task similarity estimation, we employ bandit techniques to develop an algorithm based on lower confidence bounds to efficiently explore the task structure. Furthermore, we use sparse gradient updates to facilitate parameter optimization, making the approach better suited for LPMs. Theoretical analysis is provided to justify the rationale behind our approach, and experiments on both vision transformers (ViTs) and large language models (LLMs) demonstrate the effectiveness and efficiency of our approach across various domains, including vision and natural language processing tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。