Saved in:
Bibliographic Details
Main Author: Iyer, Srikrishna
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.16487
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916494851440640
author Iyer, Srikrishna
author_facet Iyer, Srikrishna
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2411_16487
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle When Babies Teach Babies: Can student knowledge sharing outperform Teacher-Guided Distillation on small datasets?
Iyer, Srikrishna
Computation and Language
Artificial Intelligence
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.
title When Babies Teach Babies: Can student knowledge sharing outperform Teacher-Guided Distillation on small datasets?
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2411.16487