A Token is Worth over 1,000 Tokens: Efficient Knowledge Distillation through Low-Rank Clone
Abstract
Low-Rank Clone (LRC) is an efficient pre-training method that enhances Small Language Models (SLMs) by combining soft pruning and activation alignment, achieving high performance with minimal resources.
Training high-performing Small Language Models (SLMs) remains costly, even with knowledge distillation and pruning from larger teacher models. Existing work often faces three key challenges: (1) information loss from hard pruning, (2) inefficient alignment of representations, and (3) underutilization of informative activations, particularly from Feed-Forward Networks (FFNs). To address these challenges, we introduce Low-Rank Clone (LRC), an efficient pre-training method that constructs SLMs aspiring to behavioral equivalence with strong teacher models. LRC trains a set of low-rank projection matrices that jointly enable soft pruning by compressing teacher weights, and activation clone by aligning student activations, including FFN signals, with those of the teacher. This unified design maximizes knowledge transfer while removing the need for explicit alignment modules. Extensive experiments with open-source teachers (e.g., Llama-3.2-3B-Instruct, Qwen2.5-3B/7B-Instruct) show that LRC matches or surpasses state-of-the-art models trained on trillions of tokens--while using only 20B tokens, achieving over 1,000x training efficiency. Our codes and model checkpoints are available at https://github.com/CURRENTF/LowRankClone and https://huggingface.co/collections/JitaiHao/low-rank-clone-lrc-6828389e96a93f1d4219dfaf.
Community
Training high-performance Small Language Models (SLMs) remains expensive. We introduce Low-Rank Clone (LRC), a knowledge distillation method that uses low-rank matrices to simultaneously soft-prune teacher weights and align student activations, effectively "cloning" a strong teacher model. LRC matches or surpasses state-of-the-art models using only 20B tokens (compared to trillions), achieving over 1,000x training efficiency.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Empirical Evaluation of Knowledge Distillation from Transformers to Subquadratic Language Models (2025)
- ShortV: Efficient Multimodal Large Language Models by Freezing Visual Tokens in Ineffective Layers (2025)
- Penrose Tiled Low-Rank Compression and Section-Wise Q&A Fine-Tuning: A General Framework for Domain-Specific Large Language Model Adaptation (2025)
- Scalable LLM Math Reasoning Acceleration with Low-rank Distillation (2025)
- Overtrained Language Models Are Harder to Fine-Tune (2025)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment:
@librarian-bot
recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper