Competition-Level Code Generation with AlphaCode

• Published in: arXiv.org
• Main Authors: Li, Yujia, Choi, David, Chung, Junyoung, Kushman, Nate, Schrittwieser, Julian, Leblond, Rémi, Eccles, Tom, Keeling, James, Gimeno, Felix, Agustin Dal Lago, Thomas, Hubert, Choy, Peter, Cyprien de Masson d'Autume, Babuschkin, Igor, Chen, Xinyun, Po-Sen, Huang, Welbl, Johannes, Gowal, Sven, Cherepanov, Alexey, Molloy, James, Mankowitz, Daniel J, Esme Sutherland Robson, Kohli, Pushmeet, Nando de Freitas, Kavukcuoglu, Koray, Vinyals, Oriol
• Format: Paper Journal Article
• Language: English
• Published: Ithaca Cornell University Library, arXiv.org 08.02.2022
• Subjects: Algorithms; Computer Science - Artificial Intelligence; Computer Science - Learning; Computer Science - Programming Languages; Natural language (computers); Problem solving; Programming; Scale models; Task complexity; Translating
• ISSN: 2331-8422
• DOI: 10.48550/arxiv.2203.07814

Programming is a powerful and ubiquitous problem-solving tool. Developing systems that can assist programmers or even generate programs independently could make programming more productive and accessible, yet so far incorporating innovations in AI has proven challenging. Recent large-scale language models have demonstrated an impressive ability to generate code, and are now able to complete simple programming tasks. However, these models still perform poorly when evaluated on more complex, unseen problems that require problem-solving skills beyond simply translating instructions into code. For example, competitive programming problems which require an understanding of algorithms and complex natural language remain extremely challenging. To address this gap, we introduce AlphaCode, a system for code generation that can create novel solutions to these problems that require deeper reasoning. In simulated evaluations on recent programming competitions on the Codeforces platform, AlphaCode achieved on average a ranking of top 54.3% in competitions with more than 5,000 participants. We found that three key components were critical to achieve good and reliable performance: (1) an extensive and clean competitive programming dataset for training and evaluation, (2) large and efficient-to-sample transformer-based architectures, and (3) large-scale model sampling to explore the search space, followed by filtering based on program behavior to a small set of submissions.