LISA,比LoRA还快50%的微调方法

LoRA 非常侧重 LLM 的底层和顶层的权重。利用这一特性,LMFlow 团队提出一个极其简洁的算法:Layerwise Importance Sampled AdamW(LISA)。

论文链接:
https://arxiv.org/abs/2403.17919
开源地址:
https://github.com/OptimalScale/LMFlow

LISA 算法的核心在于:

  • 始终更新底层 embedding 和顶层 linear head;

  • 随机更新少数中间的 self-attention 层,比如 2-4 层。

在这里插入图片描述
实验发现该算法在指令微调任务上超过 LoRA 甚至全参数微调。LISA 每次中间只会激活一小部分参数,算法对更深的网络,以及梯度检查点技术(Gradient Checkpointing)也很友好,能够带来更大的空间节省。在指令微调任务上,LISA 的收敛性质比 LoRA 有很大提升。

LMFlow 现已集成 LISA,安装完成后只需一条指令就可以使用 LISA 进行微调。

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值