English

SAD-LoRA: Spectral Alignment for Low-Rank Knowledge Distillation

Machine Learning 2026-07-05 v1 Computer Vision and Pattern Recognition

Abstract

Distilling a fine-tuned teacher into a LoRA-adapted student is a standard recipe for parameter-efficient compression, but output-level KD does not explicitly control which rank-rr weight subspace the adapter occupies. We propose \textbf{SAD-LoRA} (\textbf{S}pectral \textbf{A}lignment \textbf{D}istillation), which selects this subspace from the data-weighted student-space reference update \DWT\Sigx1/2\DWT\Sigx^{1/2} and maintains it during training via a differentiable principal-angle loss on \colspan(B)\colspan(B). We show that the data-weighted distillation error decomposes exactly into subspace misalignment, within-subspace coefficient mismatch, and irreducible rank residual; standard KD can affect the first term only indirectly through output gradients. On controlled synthetic problems with a flat teacher spectrum, SAD-LoRA reduces the subspace-misalignment term from 51%51\% to nearly zero and lifts final subspace alignment from 0.490.49 to 1.001.00. On RoBERTa-large to RoBERTa-base distillation across six GLUE tasks, SAD-LoRA improves rank efficiency: at r=4r{=}4, it matches or beats the strongest included spectral baseline on five of six tasks, and at r=8r{=}8 it gives the best result on SST-2 and CoLA. Ablations identify subspace alignment as the load-bearing component, while coefficient matching is auxiliary.

Cite

@article{arxiv.2607.04306,
  title  = {SAD-LoRA: Spectral Alignment for Low-Rank Knowledge Distillation},
  author = {Omer Tariq and Syed Muhammad Raza and Jeongbae Son},
  journal= {arXiv preprint arXiv:2607.04306},
  year   = {2026}
}

Comments

ICML'26 workshop on CoLoRAI - The 2nd Workshop on Connecting Low-rank Representations in AI, 15 pages