Skip to content
Sagan

Paper

Echo-LoRA: Parameter-Efficient Fine-Tuning via Cross-Layer Representation Injection

Unreadunread

AI summary

Echo-LoRA is a parameter-efficient fine-tuning method that injects information from deeper layers back into shallower LoRA modules during training. It collects hidden states from late layers, aggregates them into an "echo representation," and uses lightweight projection and gating networks to feed that signal into early LoRA/DoRA updates. The echo path is discarded after training, so the deployed model is identical to standard LoRA at inference. On eight commonsense reasoning benchmarks, Echo-LoRA improves over LoRA by 5.7 percentage points (against reported baselines) or 3.0 points (against reproduced baselines) across LLaMA-7B/LLaMA2-7B/LLaMA3-8B.

Main takeaways:

  • Echo-LoRA collects boundary hidden states from deeper layers and injects them into shallow LoRA modules via lightweight projection and gating, creating a training-only cross-layer feedback loop.
  • The echo path is removed after training, so inference keeps the original low-rank LoRA form with no added parameters or compute.
  • On eight commonsense reasoning tasks, the method improves over LoRA by 3–5.7 percentage points depending on baseline, and by 2.7 points when combined with DoRA.
  • Answer-only masking, masked distillation, and stochastic routing are used to stabilize the auxiliary path and reduce train-test mismatch.