arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.06322cs.LGcs.AIcs.CV
收藏

MagMax:利用模型合并实现无缝持续学习

MagMax: Leveraging Model Merging for Seamless Continual Learning

IDEAS NCBR;Warsaw University of Technology(华沙工业大学);Autonomous University of Barcelona(巴塞罗那自治大学);Computer Vision Center(计算机视觉中心);Tooploox;Gdańsk University of Technology(格但斯克工业大学)

Daniel Marczak, Bartłomiej Twardowski, Tomasz Trzciński, Sebastian Cygert

AI总结:

本文提出MagMax持续学习方法,通过顺序微调与最大幅度权重选择相结合的模型合并策略,使大型预训练模型在连续任务中有效整合知识并减少遗忘,在类别和领域增量学习场景中表现优越。

摘要:

本文介绍了一种名为 MagMax 的持续学习方法,该方法利用模型合并使大型预训练模型能够从新数据中持续学习,而不会遗忘先前获得的知识。与旨在任务训练期间减少遗忘的传统持续学习方法不同,MagMax 将顺序微调与最大幅度权重选择相结合,以实现跨任务的有效知识整合。我们的首个贡献是对模型合并技术进行了广泛研究,揭示了诸如权重平均和随机权重选择等简单方法在各种持续学习场景中出人意料地表现良好。更重要的是,我们提出了 MagMax,这是一种新颖的模型合并策略,能够使大型预训练模型针对连续任务进行持续学习。我们的全面评估证明了 MagMax 在包括类别增量和领域增量学习设置在内的各种场景中的优越性。代码可在以下网址获取:https://github.com/danielm1405/magmax。

英文摘要:

This paper introduces a continual learning approach named MagMax, which utilizes model merging to enable large pre-trained models to continuously learn from new data without forgetting previously acquired knowledge. Distinct from traditional continual learning methods that aim to reduce forgetting during task training, MagMax combines sequential fine-tuning with a maximum magnitude weight selection for effective knowledge integration across tasks. Our initial contribution is an extensive examination of model merging techniques, revealing that simple approaches like weight averaging and random weight selection surprisingly hold up well in various continual learning contexts. More importantly, we present MagMax, a novel model-merging strategy that enables continual learning of large pre-trained models for successive tasks. Our thorough evaluation demonstrates the superiority of MagMax in various scenarios, including class- and domain-incremental learning settings. The code is available at this URL: https://github.com/danielm1405/magmax.

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

↑