arXiv:2501.03152cs.LGcs.AI2025-01

用互信息上界构建新指标,更准预测LoRA微调性能

The Scaling Law for LoRA Base on Mutual Information Upper Bound

  • 基于互信息上界设计内部评估指标
  • 在Llama3-8B和Phi3-3B上验证指标更贴合缩放规律
  • 适合关注LoRA可解释性与高效微调的研究者

LoRA(低秩适应)是一种广泛使用的模型微调方法。在微调过程中,模型性能、参数量与数据复杂度之间的关系一直是研究重点。现有方法通常依赖交叉熵或困惑度等外部指标评估性能,但这些指标难以捕捉预训练大模型的通用知识与LoRA模块从新数据中学习的特定知识之间的依赖关系。为此,我们基于互信息上界(MIUB)理论设计了一种内部评估指标,用于研究大模型LoRA微调的缩放规律。实验在基准数据集上使用Llama3-8B和Phi3-3B模型进行验证,结果表明,所提出的MIUB指标相较于交叉熵和困惑度,能更准确且稳定地反映LoRA微调的缩放规律。

原文摘要 · Abstract (English)

LoRA (Low-Rank Adaptation) is a widely used model fine-tuning method. In fine-tuning, the law among model performance, model parameters, and data complexity has been a focal issue in the field. Existing methods often leverage external metrics (such as cross-entropy or perplexity) to evaluate model performance. In the fine-tuning process for large models, two types of knowledge are typically involved: the frozen, general knowledge acquired by the model during pre-training and the new knowledge learned through the LoRA module from the current data. Generally, the less LoRA's learned knowledge relies on the large model, the more it captures the specific knowledge of new data, thereby enhancing its adaptability to new tasks. However, external metrics do not readily capture the dependency relationship between these two types of knowledge. Therefore, we designed an internal metric based on the Mutual Information Upper Bound (MIUB) theory to investigate the scaling law of large-model LoRA fine-tuning. In our experiments, we validated this approach on benchmark datasets, using the Llama3-8B and Phi3-3B models. The results show that the proposed MIUB metric aligns more accurately and stably with the scaling law of LoRA fine-tuning compared to cross-entropy and perplexity.

LoRA微调缩放律

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