Papers
arxiv:2505.12781

A Token is Worth over 1,000 Tokens: Efficient Knowledge Distillation through Low-Rank Clone

Published on May 19
· Submitted by JitaiHao on May 20
Authors:
,
,
,
,

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.

AI-generated summary

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

Paper author Paper submitter

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

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

Sign up or log in to comment

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2505.12781 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2505.12781 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2505.12781 in a Space README.md to link it from this page.

Collections including this paper 3