通过共享适配器提升小模型知识蒸馏效果,训练速度翻倍。
SLAD : Shared LoRA Adapters for Task Specific Distillation

- 用低秩适配器保持师生模型特征对齐
- 共享适配器使小模型准确率显著提升
- 适合嵌入式设备上轻量模型的高效微调
在资源受限的嵌入式系统中,将小型基础模型适配到下游任务日益流行。近期兴起的任务特定蒸馏框架中,使用相同架构的大、小模型共同适配同一任务,旨在将大模型的知识迁移到小模型。现有方法通常先微调大模型(教师)再进行蒸馏,但研究发现仅用线性探测反而效果更好。我们发现这是因为微调导致师生模型特征表示错位。为此,我们提出采用低秩适配器(LoRA)以改善特征对齐。进一步地,在联合训练中共享两个编码器的适配器参数,强化对齐效果。所提方法SLAD显著提升了师生模型间的特征一致性,不仅提高小模型性能,也增强了大模型表现,且训练速度比传统微调快2倍。在多个分类与分割数据集上的实验表明,该方法在任务特定蒸馏框架中达到当前最优性能。
原文摘要 · Abstract (English)
In the context of resource-constrained environments such as embedded systems, adapting reduced-size foundation models to downstream tasks has become increasingly popular. This has recently motivated the emerging setting of task-specific distillation, where a larger and a smaller version of the same foundation model are both adapted to the same downstream task, with the goal of transferring knowledge from the former to the latter. Recent work has demonstrated the benefits of using a larger version of the same foundation model to assist the adaptation of a smaller one. Typically, the larger model (teacher) is first adapted via fine-tuning or linear probing before its knowledge is distilled into the smaller model (student). While fine-tuning the teacher often increases its performance, recent work showed that probing it leads to better knowledge distillation to the student. Our findings show that this is mainly due to a mis-alignment in feature representation between the teacher and the student which occurs during the teacher's fine-tuning. Inspired by existing efforts to preserve previously learned knowledge, we first propose leveraging low-rank adaptation, resulting in better feature alignment and therefore better knowledge transfer. Drawing from this insight, we further enhance the feature alignment through a parameter-sharing strategy of the adapters between the two encoders during joint training. Our proposed method, SLAD, shows better feature alignment between the teacher and student, which results in increased performance for not only the student but also the teacher model, while being 2x faster to train than fine-tuning. Through extensive experiments on multiple classification and segmentation datasets, we demonstrate the improved accuracy and transfer efficiency of our method, achieving state-of-the-art performance in the task-specific distillation framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。