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

通过均场注意力动力学理解LoRA中的灾难性遗忘

Understanding Catastrophic Forgetting In LoRA via Mean-Field Attention Dynamics

Hugo Koubbi, Louis Hernandez, Matthieu Boussard

AI总结:

本文通过均场自注意力玩具模型研究LoRA的灾难性遗忘问题,分析扰动范数和Transformer深度对遗忘行为的影响,并通过实验验证理论结果。

摘要:

低秩适应(LoRA)是由于其有利的计算-性能权衡而成为主导的参数高效微调方法,但其却遭受灾难性遗忘。我们通过一个可处理的均场自注意力玩具模型研究遗忘现象,其中令牌作为相互作用粒子系统演变,LoRA作为低秩扰动。利用偏微分方程和动力系统工具,我们表征了暗示遗忘与非遗忘行为之间相变的区域。我们显示,一个相变出现在扰动范数,另一个出现在Transformer的深度。我们进一步将时间到偏差的界限与扰动大小和谱量相关联,并通过LoRA微调的现实模型上的实验和探索性分析验证预测趋势。

英文摘要:

Low-Rank Adaptation (LoRA) is the dominant parameter-efficient fine-tuning method due to its favorable compute-performance trade-off, yet it suffers from catastrophic forgetting. We study forgetting through a tractable _mean-field self-attention_ toy model, where tokens evolve as an interacting particle system and LoRA acts as a low-rank perturbation. Using tools from partial differential equations and dynamical systems, we characterize regimes suggesting a phase transition between forgetting and non-forgetting behavior. We show that one phase transition appears with respect to the norm of the perturbation, and the other with respect to the depth of the Transformers. We further bound the time-to-deviation in terms of the perturbation size and spectral quantities, and corroborate the predicted trends with experiments and exploratory analyses on real models under LoRA fine-tuning.

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

↑