中文

ResLoRA:低秩自适应中的恒等残差映射

计算与语言 2024-02-29 v1 人工智能

摘要

作为最流行的参数高效微调(PEFT)方法之一,低秩自适应(LoRA)常用于微调大型语言模型(LLMs)。然而,由于原始模型中的计算路径较长,有效且迅速地更新 LoRA 块的权重具有挑战性。为解决这一问题,我们提出了 ResLoRA,一种改进的 LoRA 框架。通过在训练期间添加残差路径,并在推理期间使用合并方法消除这些额外路径,我们的方法能够在不增加任何额外可训练参数或推理成本的情况下,比 LoRA 用更少的训练步骤获得更好的结果。在自然语言生成(NLG)、自然语言理解(NLU)和文本到图像任务上的实验证明了该方法的有效性。据我们所知,ResLoRA 是首个将残差路径与 LoRA 结合的工作。我们方法的代码可在 https://github.com/microsoft/LMOps/tree/main/reslora 获取。

关键词

引用

@article{arxiv.2402.18039,
  title  = {ResLoRA: Identity Residual Mapping in Low-Rank Adaption},
  author = {Shuhua Shi and Shaohan Huang and Minghui Song and Zhoujun Li and Zihan Zhang and Haizhen Huang and Furu Wei and Weiwei Deng and Feng Sun and Qi Zhang},
  journal= {arXiv preprint arXiv:2402.18039},
  year   = {2024}
}

备注

14 pages, 7 figures