arXiv:2602.05988cs.LG2026-02中稿 · the International …被引 2

按层选择性微调,用相似度指标减少50%参数量仍保持性能

Layer-wise LoRA fine-tuning: a similarity metric approach

  • 基于内部表征相似度,筛选最相关的模型层进行微调
  • 参数量减少50%,在GLUE等任务上性能几乎无损
  • 适用于编码器、解码器及多模态模型,兼容现有低秩方法

在超大规模数据集上预训练大语言模型已成为通用人工智能发展的基础。而提升其下游任务表现通常需通过微调来适配知识。参数高效微调技术(如低秩适应,LoRA)通过冻结预训练模型并更新少量参数,显著降低计算成本,相比全量微调可减少超过99%的可训练参数量。然而,随着模型规模持续增长,这一节省仍显不足。本文提出一种分层选择性微调方法,仅对少数关键层应用LoRA或其变体。我们主张并非所有层对模型适配贡献均等,通过测量各层对内部表征变化的贡献,识别出最具相关性的层。该方法与现有低秩适应技术正交且兼容。实验显示,该方法使基于LoRA的技术可训练参数减少高达50%,同时在不同模型和任务中维持预测性能。在仅编码器架构上,参数减少带来对GLUE基准测试性能的可忽略下降;在仅解码器架构上,数学求解与代码生成任务中性能下降极小甚至有所提升。该有效性同样扩展至多模态模型,其结果与全层微调的LoRA模块相当。代码已公开:https://github.com/c2d-usp/Layer-wise-LoRA-with-CKA

原文摘要 · Abstract (English)

Pre-training Large Language Models (LLMs) on web-scale datasets becomes fundamental for advancing general-purpose AI. In contrast, enhancing their predictive performance on downstream tasks typically involves adapting their knowledge through fine-tuning. Parameter-efficient fine-tuning techniques, such as Low-Rank Adaptation (LoRA), aim to reduce the computational cost of this process by freezing the pre-trained model and updating a smaller number of parameters. In comparison to full fine-tuning, these methods achieve over 99\% reduction in trainable parameter count, depending on the configuration. Unfortunately, such a reduction may prove insufficient as LLMs continue to grow in scale. In this work, we address the previous problem by systematically selecting only a few layers to fine-tune using LoRA or its variants. We argue that not all layers contribute equally to the model adaptation. Leveraging this, we identify the most relevant layers to fine-tune by measuring their contribution to changes in internal representations. Our method is orthogonal to and readily compatible with existing low-rank adaptation techniques. We reduce the trainable parameters in LoRA-based techniques by up to 50\%, while maintaining the predictive performance across different models and tasks. Specifically, on encoder-only architectures, this reduction in trainable parameters leads to a negligible predictive performance drop on the GLUE benchmark. On decoder-only architectures, we achieve a small drop or even improvements in the predictive performance on mathematical problem-solving capabilities and coding tasks. Finally, this effectiveness extends to multimodal models, for which we also observe competitive results relative to fine-tuning with LoRA modules in all layers. Code is available at: https://github.com/c2d-usp/Layer-wise-LoRA-with-CKA

参数高效微调优化低秩适应分层选择

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。