arXiv:2506.16456cs.LGcs.AI2025-06

通过联合张量分解提升LoRA表达能力,用更少参数实现更好效果

Joint Tensor-Train Parameterization for Efficient and Expressive Low-Rank Adaptation

  • 用统一张量结构生成相关联的低秩矩阵,增强模型表达力
  • 在量子点分类和GPT-2任务中,以更少参数达到更高精度
  • 适合追求高效微调且注重泛化性能的研究者使用

低秩适应(LoRA)因其参数高效性被广泛用于大规模神经网络微调。然而,标准LoRA独立优化低秩矩阵,限制了其表达能力和泛化性能。虽然可对单个LoRA矩阵单独使用经典张量分解(TT),但该方法并未显著提升参数效率或性能。本文提出TensorGuide,一种新型张量分解引导的适应框架:通过受控高斯噪声驱动的统一TT结构生成两个相关联的低秩LoRA矩阵。该联合TT表示天然具备结构化、低秩特性,显著提升表达力、泛化能力与参数效率,且不增加可训练参数数量。理论上,基于神经正切核分析证明其优化动态更优、泛化能力更强。大量实验在量子点分类与GPT-2微调任务中表明,基于TensorGuide的LoRA持续优于标准LoRA与TT-LoRA,在更少参数下实现更高准确率与可扩展性。

原文摘要 · Abstract (English)

Low-Rank Adaptation (LoRA) is widely recognized for its parameter-efficient fine-tuning of large-scale neural models. However, standard LoRA independently optimizes low-rank matrices, which inherently limits its expressivity and generalization capabilities. While classical tensor-train (TT) decomposition can be separately employed on individual LoRA matrices, this work demonstrates that the classical TT-based approach neither significantly improves parameter efficiency nor achieves substantial performance gains. This paper proposes TensorGuide, a novel tensor-train-guided adaptation framework to overcome these limitations. TensorGuide generates two correlated low-rank LoRA matrices through a unified TT structure driven by controlled Gaussian noise. The resulting joint TT representation inherently provides structured, low-rank adaptations, significantly enhancing expressivity, generalization, and parameter efficiency without increasing the number of trainable parameters. Theoretically, we justify these improvements through neural tangent kernel analyses, demonstrating superior optimization dynamics and enhanced generalization. Extensive experiments on quantum dot classification and GPT-2 fine-tuning benchmarks demonstrate that TensorGuide-based LoRA consistently outperforms standard LoRA and TT-LoRA, achieving improved accuracy and scalability with fewer parameters.

低秩微调张量分解参数效率

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