通过跨层注入隐藏状态,用极小参数提升大模型推理能力
Echo-LoRA: Parameter-Efficient Fine-Tuning via Cross-Layer Representation Injection

- 从深层网络提取隐藏状态,生成样本级信号注入浅层模块
- 在多个基准上平均提升5.7个百分点,优于原有LoRA方法
- 训练后移除额外路径,部署时无额外计算负担
参数高效微调(PEFT)已成为适配大语言模型到下游任务的实用路径,其中LoRA类方法因其训练成本低、部署简单而备受青睐。然而,大多数LoRA变体仅在各层权重空间内修改更新规则,忽略了深层网络形成的中间表示。本文提出Echo-LoRA,一种基于跨层表示注入的参数高效微调方法:训练时,从深层源层收集边界隐藏状态,聚合为样本级的回声表示,并通过轻量级投影与门控网络将其注入浅层LoRA或DoRA模块。采用仅答案掩码、掩码蒸馏和随机路由策略,确保辅助路径稳定并缩小训练与推理差距。在八个常识推理基准上,Echo-LoRA在LLaMA-7B、LLaMA2-7B和LLaMA3-8B上平均超越报告的LoRA基线5.7个百分点;在统一实现下重现的基线中,平均提升3.0点;与DoRA结合时,提升2.7点。训练完成后,回声路径被丢弃,部署模型保持原始低秩形式,不引入额外推理参数或计算开销。
原文摘要 · Abstract (English)
Parameter-efficient fine-tuning (PEFT) has become a practical route for adapting large language models to downstream tasks, with LoRA-style methods being particularly attractive because they are inexpensive to train and easy to deploy. Most LoRA variants, however, revise the update rule within the weight space of each layer and leave the intermediate representations formed by deeper layers largely unused. We propose Echo-LoRA, a cross-layer representation injection method for parameter-efficient fine-tuning. During training, Echo-LoRA collects boundary hidden states from deeper source layers, aggregates them into a sample-level echo representation, and uses lightweight projection and gating networks to inject the resulting signal into shallow LoRA or DoRA modules. Answer-only masking, masked distillation, and stochastic routing are used to keep this auxiliary path stable and to reduce the gap between training and inference. On eight commonsense reasoning benchmarks, Echo-LoRA exceeds the reported LoRA baselines by 5.7 percentage points on average across LLaMA-7B, LLaMA2-7B, and LLaMA3-8B. Under reproduced LoRA baselines in our unified implementation, the average gain is 3.0 points; when combined with DoRA, the gain is 2.7 points. The Echo path is discarded after training, so the deployed model keeps the original low-rank LoRA/DoRA form and adds neither inference-time parameters nor inference computation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。