SantaCoder: don't reach for the stars!
Loubna Ben Allal, Raymond Li, Denis Kocetkov, Chenghao Mou,, Christopher Akiki, Carlos Munoz Ferrandis, Niklas Muennighoff, Mayank Mishra,, Alex Gu, Manan Dey, Logesh Kumar Umapathi, Carolyn Jane Anderson, Yangtian, Zi, Joel Lamy Poirier, Hailey Schoelkopf, Sergey Troshin

TL;DR
This paper reports on the BigCode project's progress in developing large language models for code, highlighting data filtering techniques, model training on multiple languages, and outperforming larger models on code generation benchmarks.
Contribution
Introduces a 1.1B parameter multilingual code generation model trained on The Stack, with improved data filtering and evaluation showing superior performance over larger models.
Findings
Aggressive filtering of near-duplicate data improves performance.
Selecting files from repositories with fewer GitHub stars can enhance model quality.
The models outperform larger open-source models on code generation benchmarks.
Abstract
The BigCode project is an open-scientific collaboration working on the responsible development of large language models for code. This tech report describes the progress of the collaboration until December 2022, outlining the current state of the Personally Identifiable Information (PII) redaction pipeline, the experiments conducted to de-risk the model architecture, and the experiments investigating better preprocessing methods for the training data. We train 1.1B parameter models on the Java, JavaScript, and Python subsets of The Stack and evaluate them on the MultiPL-E text-to-code benchmark. We find that more aggressive filtering of near-duplicates can further boost performance and, surprisingly, that selecting files from repositories with 5+ GitHub stars deteriorates performance significantly. Our best model outperforms previous open-source multilingual code generation models…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Code & Models
- 🤗bigcode/santacodermodel· 7.0k dl· ♡ 3357.0k dl♡ 335
- 🤗olivierdehaene/optimized-santacodermodel· 19 dl· ♡ 819 dl♡ 8
- 🤗bigcode/starpiimodel· 5.4k dl· ♡ 1235.4k dl♡ 123
- 🤗mayank-mishra/santacoder-GPTQ-8bit-128gmodel· ♡ 1♡ 1
- 🤗mayank-mishra/santacoder-GPTQ-4bit-128gmodel· ♡ 2♡ 2
- 🤗prashrex/Santacoder-ggufmodel· 8 dl8 dl
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Taxonomy
TopicsSoftware Engineering Research · Topic Modeling · Machine Learning and Data Classification
