NTK机制下的LoRA训练不存在虚假局部极小值
机器学习
2024-05-29 v3 最优化与控制
摘要
低秩适应(LoRA)已成为大型语言模型(LLM)参数高效微调的标准方法,但我们对LoRA的理论理解仍然有限。本文在神经正切核(NTK)机制下对具有个数据点的LoRA微调进行了理论分析,表明:(i)全微调(无LoRA)存在秩为的低秩解;(ii)使用秩的LoRA可消除虚假局部极小值,使梯度下降能够找到低秩解;(iii)使用LoRA找到的低秩解具有良好的泛化性能。
引用
@article{arxiv.2402.11867,
title = {LoRA Training in the NTK Regime has No Spurious Local Minima},
author = {Uijeong Jang and Jason D. Lee and Ernest K. Ryu},
journal= {arXiv preprint arXiv:2402.11867},
year = {2024}
}
备注
23 pages