When Babies Teach Babies: Can student knowledge sharing outperform Teacher-Guided Distillation on small datasets?
Not on the map. Why?
Abstract (arXiv)
We present our submission to the BabyLM challenge, aiming to push the boundaries of data-efficient language model pretraining. Our method builds upon deep mutual learning, introducing a student model search for diverse initialization. We address the limitation of treating students equally by formulating weighted mutual learning as a bi-level optimization problem. The inner loop learns compact students through online distillation, while the outer loop optimizes weights for better knowledge distillation from diverse students. This dynamic weighting strategy eliminates the need for a teacher model, reducing computational requirements. Our evaluations show that teacher-less methods can match or surpass teacher-supervised approaches.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 31 references that are in the map. First posted on arXiv 25 Nov 2024.
Builds on
21 of 31 references.
References · 21
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
- 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,718
- 2017Learning Efficient Convolutional Networks through Network Slimmingcited by 673
- 2017Deep Mutual Learningcited by 621
- 2015BinaryConnect: Training Deep Neural Networks with binary weights during propagationscited by 571
- 2019BLiMP: The Benchmark of Linguistic Minimal Pairs for Englishcited by 553
- 2019Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillationcited by 491
- 2020A Comprehensive Survey of Neural Architecture Search: Challenges and Solutionscited by 253
9 more
- 2018Neural Architecture Search with Bayesian Optimisation and Optimal Transportcited by 192
- 2019Online Knowledge Distillation with Diverse Peerscited by 127
- 2015The Goldilocks Principle: Reading Children's Books with Explicit Memory Representationscited by 108
- 2018A standardized Project Gutenberg corpus for statistical analysis of natural language and quantitative linguisticscited by 64
- 2023Baby Llama: knowledge distillation from an ensemble of teachers trained on a small dataset with no performance penaltycited by 57
- 2024Elements of World Knowledge (EWoK): A Cognition-Inspired Framework for Evaluating Basic World Knowledge in Language Modelscited by 55
- 2023Trained on 100 million words and still in shape: BERT meets British National Corpuscited by 39
- 2024[Call for Papers] The 2nd BabyLM Challenge: Sample-efficient pretraining on a developmentally plausible corpuscited by 26
- 2018Open Subtitles Paraphrase Corpus for Six Languagescited by 24
Built on by
Not cited yet.