English

DistillLens: Symmetric Knowledge Distillation Through Logit Lens

Computation and Language 2026-02-17 v1

Abstract

Standard Knowledge Distillation (KD) compresses Large Language Models (LLMs) by optimizing final outputs, yet it typically treats the teacher's intermediate layer's thought process as a black box. While feature-based distillation attempts to bridge this gap, existing methods (e.g., MSE and asymmetric KL divergence) ignore the rich uncertainty profiles required for the final output. In this paper, we introduce DistillLens, a framework that symmetrically aligns the evolving thought processes of student and teacher models. By projecting intermediate hidden states into the vocabulary space via the Logit Lens, we enforce structural alignment using a symmetric divergence objective. Our analysis proves that this constraint imposes a dual-sided penalty, preventing both overconfidence and underconfidence while preserving the high-entropy information conduits essential for final deduction. Extensive experiments on GPT-2 and Llama architectures demonstrate that DistillLens consistently outperforms standard KD and feature-transfer baselines on diverse instruction-following benchmarks. The code is available at https://github.com/manishdhakal/DistillLens.

Keywords

Cite

@article{arxiv.2602.13567,
  title  = {DistillLens: Symmetric Knowledge Distillation Through Logit Lens},
  author = {Manish Dhakal and Uthman Jinadu and Anjila Budathoki and Rajshekhar Sunderraman and Yi Ding},
  journal= {arXiv preprint arXiv:2602.13567},
  year   = {2026}
}

Comments

Knowledge Distillation in LLMs

R2 v1 2026-07-01T10:36:29.749Z