Team Ai
Modelpublic

Data-Selection/BSL-160M

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes570downloads
Model Card

BSL-160M

paper | code

BSL-160M is a 160M model with Mistral achitecture pre-trained from scratch on the CC split of Redpajama.

It is used as the baseline for [PDS-160M](https://huggingface.co/Data-Selection/PDS-160M).

Evaluation

PDS-selected data improves the performance of language models pre-trained from scratch and saves pre-training comptation. The improvement scales up to large model sizes.

<p align='left'> <img src="https://cdn-uploads.huggingface.co/production/uploads/624ac662102fcdff87be51b9/6undIr37d10qD73TDiPDK.png" width="600"> </p>

Citation

bibtex
@article{gu2024data,
  title={Data Selection via Optimal Control for Language Models},
  author={Gu, Yuxian and Dong, Li and Wang, Hongning and Hao, Yaru and Dong, Qingxiu and Wei, Furu and Huang, Minlie},
  journal={arXiv preprint arXiv:2410.07064},
  year={2024}
}