← Search

Baptiste Roziere

9 accepted papers

2025

TestGenEval: A Real World Unit Test Generation and Test Completion Benchmark

ICLR 2025poster

Code generation models can help improve many common software tasks ranging from code completion to defect prediction. Most of the existing benchmarks for code generation LLMs focus on code authoring or code completion. Surprisingly, there has been far less effort dedicated to benchmarking software t…

Cited by 13SourcePDFScholar
2024

Better & Faster Large Language Models via Multi-token Prediction

ICML 2024poster

Large language models such as GPT and Llama are trained with a next-token prediction loss. In this work, we suggest that training language models to predict multiple future tokens at once results in higher sample efficiency. More specifically, at each position in the training corpus, we ask the mode…

Cited by 85SourcePDFScholar
2024

CRUXEval: A Benchmark for Code Reasoning, Understanding and Execution

ICML 2024poster

We present Code Reasoning, Understanding, and eXecution Evaluation, a benchmark consisting of 800 Python functions (3-13 lines). Each function comes with an input-output pair, leading to two natural tasks: input prediction and output prediction. First, we propose a general recipe for generating our…

Cited by 84SourcePDFScholar
2024

Getting the most out of your tokenizer for pre-training and domain adaptation

ICML 2024poster

Tokenization is an understudied and often neglected component of modern LLMs. Most published works use a single tokenizer for all experiments, often borrowed from another model, without performing ablations or analysis to optimize tokenization. Moreover, the tokenizer is generally kept unchanged whe…

2023

Code Translation with Compiler Representations

ICLR 2023top-25%

In this paper, we leverage low-level compiler intermediate representations (IR) code translation. Traditional transpilers rely on syntactic information and handcrafted rules, which limits their applicability and produces unnatural-looking code. Applying neural machine translation (NMT) approaches to…

Cited by 86SourcePDFScholar
2022

Leveraging Automated Unit Tests for Unsupervised Code Translation

ICLR 2022spotlight

With little to no parallel data available for programming languages, unsupervised methods are well-suited to source code translation. However, the majority of unsupervised machine translation approaches rely on back-translation, a method developed in the context of natural language translation and o…

Cited by 137SourcePDFScholar
2021

DOBF: A Deobfuscation Pre-Training Objective for Programming Languages

NeurIPS 2021poster

Recent advances in self-supervised learning have dramatically improved the state of the art on a wide variety of tasks. However, research in language model pre-training has mostly focused on natural languages, and it is unclear whether models like BERT and its variants provide the best pre-training…

2020

Adversarial Attacks on Linear Contextual Bandits

NeurIPS 2020poster

Contextual bandit algorithms are applied in a wide range of domains, from advertising to recommender systems, from clinical trials to education. In many of these domains, malicious agents may have incentives to force a bandit algorithm into a desired behavior For instance, an unscrupulous ad publish…

Cited by 68SourcePDFScholar
2020

Unsupervised Translation of Programming Languages

NeurIPS 2020poster

A transcompiler, also known as source-to-source translator, is a system that converts source code from a high-level programming language (such as C++ or Python) to another. Transcompilers are primarily used for interoperability, and to port codebases written in an obsolete or deprecated language (e.…