Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces a distillation pipeline for transferring knowledge from quadratic attention-based LLMs (Llama, Qwen, Olmo) to sub-quadratic xLSTM architectures, aiming for lossless distillation defined by tolerance-corrected Win-and-Tie rates. The pipeline includes a novel merging stage to combine individually linearized experts into a single xLSTM model. Results demonstrate that the distilled xLSTM models can recover most of the teacher's performance and even surpass it on some downstream tasks, suggesting a viable path to more efficient LLMs.
xLSTM models can now effectively learn from large attention-based models, even outperforming their teachers on some tasks through a novel distillation and merging pipeline.
There have been numerous attempts to distill quadratic attention-based large language models (LLMs) into sub-quadratic linearized architectures. However, despite extensive research, such distilled models often fail to match the performance of their teacher LLMs on various downstream tasks. We set out the goal of lossless distillation, which we define in terms of tolerance-corrected Win-and-Tie rates between student and teacher on sets of tasks. To this end, we introduce an effective distillation pipeline for xLSTM-based students. We propose an additional merging stage, where individually linearized experts are combined into a single model. We show the effectiveness of this pipeline by distilling base and instruction-tuned models from the Llama, Qwen, and Olmo families. In many settings, our xLSTM-based students recover most of the teacher's performance, and even exceed it on some downstream tasks. Our contributions are an important step towards more energy-efficient and cost-effective replacements for transformer-based LLMs.