SAD-LoRA: Spectral Alignment for Low-Rank Knowledge Distillation
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- 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 and maintains it during training via a differentiable principal-angle loss on . 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 to nearly zero and lifts final subspace alignment from to . On RoBERTa-large to RoBERTa-base distillation across six GLUE tasks, SAD-LoRA improves rank efficiency: at , it matches or beats the strongest included spectral baseline on five of six tasks, and at 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