---
datasets:
- togethercomputer/RedPajama-Data-1T
language:
- en
library_name: transformers
license: apache-2.0
pipeline_tag: text-generation
---
## PDS-470M
[paper](https://arxiv.org/abs/2410.07064) | [code](https://github.com/microsoft/LMOps/tree/main/data_selection) | [project page](https://github.com/microsoft/LMOps/tree/main/data_selection)
**PDS-470M** is a 470M parameter Mistral architecture model **pretrained from scratch** using the PDS framework on data selected from the CC split of [Redpajama](https://github.com/togethercomputer/RedPajama-Data).
The PDS framework is based on the [Pontryagin's maximum principle](https://en.wikipedia.org/wiki/Pontryagin%27s_maximum_principle#:~:text=Pontryagin's%20maximum%20principle%20is%20used,the%20state%20or%20input%20controls.) for optimal pre-training data selection, offering strong theoretical support and scalability for training large language models.
Please refer to our [paper](https://arxiv.org/abs/2410.07064) for more details.
### Overview of the theory:
### Overview of the PDS framework:
### 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.
### Baseline
[Conventional Pre-training](https://huggingface.co/Data-Selection/BSL-470M)
### 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}
}
```