Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SantaCoder: don't reach for the stars!

About

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 (InCoder-6.7B and CodeGen-Multi-2.7B) in both left-to-right generation and infilling on the Java, JavaScript, and Python portions of MultiPL-E, despite being a substantially smaller model. All models are released under an OpenRAIL license at https://hf.co/bigcode.

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, Dmitry Abulkhanov, Manuel Romero, Michael Lappert, Francesco De Toni, Bernardo Garc\'ia del R\'io, Qian Liu, Shamik Bose, Urvashi Bhattacharyya, Terry Yue Zhuo, Ian Yu, Paulo Villegas, Marco Zocca, Sourab Mangrulkar, David Lansky, Huu Nguyen, Danish Contractor, Luis Villa, Jia Li, Dzmitry Bahdanau, Yacine Jernite, Sean Hughes, Daniel Fried, Arjun Guha, Harm de Vries, Leandro von Werra• 2023

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval (test)
Pass@118
444
Code GenerationMBPP (test)
Pass@135
276
Code GenerationDS-1000 1.0 (test)
Matplotlib21.6
19
Docstring GenerationCodeXGLUE Python (test)
BLEU19.74
11
Fill-in-the-Middle Code CompletionFIM-Tasks Single-Line Infilling (test)
Python Score44
7
Code GenerationODEX
English Overall Pass@137.7
6
Secret DetectionGitHub Issue Reports (test)
Precision93.05
5
Code SummarizationCodeXGLUE Python
BLEU19.74
4
Python return type predictionPradel benchmarks Fried adaptation (test)
Non-None F166.9
4
Single-line fill-in-the-middleFIM Java (test)
Line Exact Match62
3
Showing 10 of 12 rows

Other info

Follow for update