How to?

LoRA

Jan 1, 2025
技术学习models
1 Minutes
107 Words

假设原始线性层权重是:

  • W: [1000, 2000] # out_features=1000, in_features=2000

LoRA rank r=16 时,会训练两个小矩阵:

  • A: [16, 2000]
  • B: [1000, 16]

前向时原来是:

  • y = W x

加 LoRA 后变成:

  • y = W x + scale * B(Ax)

其中:

  • scale = lora_alpha / r = 32 / 16 = 2

transformer 的哪些线性层会被应用 lora?

Lora 应用于 qkv 的 proj,out_proj,以及 FFN 中的所有线性层 (e.g. 对于 GLU 来说, gate_proj, up_proj 和 down_proj 都有).

Article title:LoRA
Article author:Julyfun
Release time:Jan 1, 2025
Copyright 2026
Sitemap