让机器直接学习低秩微调权重,实现性能预测与模型编辑。
Learning on LoRAs: GL-Equivariant Processing of Low-Rank Weight Spaces for Large Finetuned Models
- 设计对称感知的等变模型处理低秩权重空间。
- 可准确预测微调后模型在下游任务的表现与数据属性。
- 适合模型压缩、安全检测与自动化微调研究者使用。
低秩适应(LoRAs)革新了大模型微调方式,可在资源有限时高效适配。本文提出学习于LoRA(LoL)范式,将LoRA权重本身作为机器学习模型的输入。例如,此类模型可预测微调后模型的下游性能、检测有害微调或生成新模型修改而无需传统训练。我们首先识别出低秩权重分解中的内在参数对称性,其与标准神经网络显著不同。为此,开发了多种对称感知的不变或等变LoL模型,采用规范化解析、不变特征提取和等变层等工具。通过微调数千个文生图扩散模型与语言模型,构建了LoRA数据集。在数值实验中,验证了所提架构能有效处理低秩权重分解,成功预测CLIP分数、微调数据属性、数据归属及下游任务准确率。
原文摘要 · Abstract (English)
Low-rank adaptations (LoRAs) have revolutionized the finetuning of large foundation models, enabling efficient adaptation even with limited computational resources. The resulting proliferation of LoRAs presents exciting opportunities for applying machine learning techniques that take these low-rank weights themselves as inputs. In this paper, we investigate the potential of Learning on LoRAs (LoL), a paradigm where LoRA weights serve as input to machine learning models. For instance, an LoL model that takes in LoRA weights as inputs could predict the performance of the finetuned model on downstream tasks, detect potentially harmful finetunes, or even generate novel model edits without traditional training methods. We first identify the inherent parameter symmetries of low rank decompositions of weights, which differ significantly from the parameter symmetries of standard neural networks. To efficiently process LoRA weights, we develop several symmetry-aware invariant or equivariant LoL models, using tools such as canonicalization, invariant featurization, and equivariant layers. We finetune thousands of text-to-image diffusion models and language models to collect datasets of LoRAs. In numerical experiments on these datasets, we show that our LoL architectures are capable of processing low rank weight decompositions to predict CLIP score, finetuning data attributes, finetuning data membership, and accuracy on downstream tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。