StarCoder: May the Source Be with You!
Abstract
The BigCode community, an open-scientific collaboration working on the responsible development of Large Language Models for Code (Code LLMs), introduces StarCoder and StarCoderBase: 15.5B parameter models with 8K context length, infilling capabilities and fast large-batch inference enabled by multi-query attention. StarCoderBase is trained on 1 trillion tokens sourced from The Stack, a large collection of permissively licensed GitHub repositories with inspection tools and an opt-out process. We fine-tuned StarCoderBase on 35B Python tokens, resulting in the creation of StarCoder. We perform the most comprehensive evaluation of Code LLMs to date and show that StarCoderBase outperforms every open Code LLM that supports multiple programming languages and matches or outperforms the OpenAI code-cushman-001 model. Furthermore, StarCoder outperforms every model that is fine-tuned on Python and still retains its performance on other programming languages. We take several important steps towards a safe open-access model release, including an improved PII redaction pipeline and a novel attribution tracing tool, and make the StarCoder models publicly available under a more commercially viable version of the Open Responsible AI Model license.
Cite
Text
Li et al. "StarCoder: May the Source Be with You!." Transactions on Machine Learning Research, 2023.Markdown
[Li et al. "StarCoder: May the Source Be with You!." Transactions on Machine Learning Research, 2023.](https://mlanthology.org/tmlr/2023/li2023tmlr-starcoder/)BibTeX
@article{li2023tmlr-starcoder,
title = {{StarCoder: May the Source Be with You!}},
author = {Li, Raymond and Allal, Loubna Ben and Zi, Yangtian and Muennighoff, Niklas and Kocetkov, Denis and Mou, Chenghao and Marone, Marc and Akiki, Christopher and Li, Jia and Chim, Jenny and Liu, Qian and Zheltonozhskii, Evgenii and Zhuo, Terry Yue and Wang, Thomas and Dehaene, Olivier and Lamy-Poirier, Joel and Monteiro, Joao and Gontier, Nicolas and Yee, Ming-Ho and Umapathi, Logesh Kumar and Zhu, Jian and Lipkin, Ben and Oblokulov, Muhtasham and Wang, Zhiruo and Murthy, Rudra and Stillerman, Jason T and Patel, Siva Sankalp and Abulkhanov, Dmitry and Zocca, Marco and Dey, Manan and Zhang, Zhihan and Bhattacharyya, Urvashi and Yu, Wenhao and Luccioni, Sasha and Villegas, Paulo and Zhdanov, Fedor and Lee, Tony and Timor, Nadav and Ding, Jennifer and Schlesinger, Claire S and Schoelkopf, Hailey and Ebert, Jan and Dao, Tri and Mishra, Mayank and Gu, Alex and Anderson, Carolyn Jane and Dolan-Gavitt, Brendan and Contractor, Danish and Reddy, Siva and Fried, Daniel and Bahdanau, Dzmitry and Jernite, Yacine and Ferrandis, Carlos Muñoz and Hughes, Sean and Wolf, Thomas and Guha, Arjun and Von Werra, Leandro and de Vries, Harm},
journal = {Transactions on Machine Learning Research},
year = {2023},
url = {https://mlanthology.org/tmlr/2023/li2023tmlr-starcoder/}
}