Researchers from Apple Inc. have developed a new method for distilling knowledge from Transformer models into Mamba-like architectures. Their two-stage approach involves first distilling knowledge into a linearized Attention model using a kernel trick, and then distilling that into an adapted Mamba model. This technique allows the distilled Mamba model to retain most of the performance of the original Transformer, achieving a perplexity of 14.01 compared to the teacher's 13.86, though some gaps appear on tasks requiring strong in-context recall. AI
IMPACT This research offers a method to leverage existing Transformer knowledge for Mamba-like architectures, potentially improving efficiency and adoption of State Space Models.
RANK_REASON The cluster contains a research paper detailing a new method for cross-architecture distillation of AI models. [lever_c_demoted from research: ic=1 ai=1.0]
Read on Apple Machine Learning Research →
- Apple Inc.
- Concordia University
- Flatiron Institute
- International Conference on Machine Learning
- Johns Hopkins University
- Mamba
- MILA Research Institute
- Pythia-1B
- transformer
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →