LoRA(Low-Rank Adaptation) ,也叫低秩适配器,是一种用于迁移学习的方法;
通过低秩适应矩阵来调整源领域和目标领域之间的特征表示;
形象的解释是:
在LLM大模型的训练中,两个全连接层的神经元个数可能非常大,比如上图的全连接in层、out层各有10,000个神经元,那么在全参数训练中就有10,000 * 10,000个参数需要迭代;
而插入的 LoRA 层是一个维度非常小的低秩矩阵,比如维度是32 * 32,那现在的计算参数就等于10,000 * 32 * 2 + 32 * 32,大大缩小了训练参数的个数;
LoRA 层就好比是“中介”,将买家与卖家(in层与out层)各自的需求进行汇总,在“中介”内部完成信息交换,加快匹配速度。
from peft import LoraConfig, get_peft_model
config = LoraConfig(
r=64,
target_modules=target_modules,
lora_alpha=16,
lora_dropout=0.05,
fan_in_fan_out=False,
bias="none",
modules_to_save=None,
)
常用参数解析:
int
) :低秩矩阵的维度,通常是4、8、16、32、64;Union[List[str],str]
):添加lora的目标模块名称;int
):缩放参数,控制低秩矩阵的适应程度——越小的值对模型参数进行压缩的越强烈,可能会影响模型性能;越大的值,则减轻了对模型参数的压缩,可能保留了更多的模型性能。不同的模型可能有不同的默认值和具体用法;float
):防止过拟合的dropout;bool
):是否将层的存储形式替换成 (fan_in, fan_out) 的样子,默认为False;str
):是否添加偏置。参数选择为:[“none”,“all”,“lora_only”]。如果为"all"或"lora_only",则相应的偏差将在训练期间更新;List[str]
):要在训练过程中保存的模块列表,默认为None 表示不保存;除此之外还有以下不常用的参数:
Union[List[int],int]
):str
):dict
):dict
):