Self-Infilling Code Generation
Not on the map. Why?
Abstract (arXiv)
This work introduces self-infilling code generation, a general framework that incorporates infilling operations into auto-regressive decoding. Our approach capitalizes on the observation that recent infilling-capable code language models can self-infill: whereas infilling operations aim to fill in the middle based on a predefined prefix and suffix, self-infilling sequentially generates both such surrounding context and the infilled content. We utilize this capability to introduce novel interruption and looping mechanisms in conventional decoding, evolving it into a non-monotonic process. Interruptions allow for postponing the generation of specific code until a definitive suffix is established, enhancing control over the output. Meanwhile, the looping mechanism, which leverages the complementary nature of self-infilling and left-to-right decoding, can iteratively update and synchronize each piece of generation cyclically. Extensive experiments are conducted to demonstrate that our proposed decoding process is effective in enhancing both regularity and quality across several code generation benchmarks.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 59 of its 77 references that are in the map. First posted on arXiv 29 Nov 2023.
Builds on
59 of 77 references.
References · 59
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,893
38 more
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,374
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,442
- 2019The Curious Case of Neural Text Degenerationcited by 2,916
- 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,811
- 2023Code Llama: Open Foundation Models for Codecited by 2,139
- 2022Competition-Level Code Generation with AlphaCodecited by 1,462
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
- 2023PaLM 2 Technical Reportcited by 1,162
- 2021GLM: General Language Model Pretraining with Autoregressive Blank Infillingcited by 1,031
- 2023Large Language Models Cannot Self-Correct Reasoning Yetcited by 1,017
- 2023Teaching Large Language Models to Self-Debugcited by 988
- 2022CodeGen: An Open Large Language Model for Code with Multi-Turn Program Synthesiscited by 885
- 2022GLM-130B: An Open Bilingual Pre-trained Modelcited by 843
- 2019SpanBERT: Improving Pre-training by Representing and Predicting Spanscited by 834
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
- 2023StarCoder: may the source be with you!cited by 799
- 2023WizardCoder: Empowering Code Large Language Models with Evol-Instructcited by 733
- 2022PAL: Program-aided Language Modelscited by 652
- 2015Sequence Level Training with Recurrent Neural Networkscited by 576
- 2022InCoder: A Generative Model for Code Infilling and Synthesiscited by 498
- 2022DS-1000: A Natural and Reliable Benchmark for Data Science Code Generationcited by 452
- 2022UL2: Unifying Language Learning Paradigmscited by 295
- 2022Efficient Training of Language Models to Fill in the Middlecited by 255
- 2023LEVER: Learning to Verify Language-to-Code Generation with Executioncited by 239
- 2023Automatically Correcting Large Language Models: Surveying the landscape of diverse self-correction strategiescited by 238
- 2023OctoPack: Instruction Tuning Code Large Language Modelscited by 189
- 2023SantaCoder: don't reach for the stars!cited by 170
- 2023Scaling Autoregressive Multi-Modal Models: Pretraining and Instruction Tuningcited by 170
- 2022CM3: A Causal Masked Multimodal Model of the Internetcited by 142
- 2019Insertion Transformer: Flexible Sequence Generation via Insertion Operationscited by 125
- 2022Retrieval-Augmented Multimodal Language Modelingcited by 125
- 2023CodeGen2: Lessons for Training LLMs on Programming and Natural Languagescited by 125
- 2020Enabling Language Models to Fill in the Blankscited by 122
- 2022Coder Reviewer Reranking for Code Generationcited by 84
- 2023Lemur: Harmonizing Natural Language and Code for Language Agentscited by 72
- 2021HTLM: Hyper-Text Pre-Training and Prompting of Language Modelscited by 58
- 2022Transcending Scaling Laws with 0.1% Extra Computecited by 57
50 most cited of 59 All 59 (JSON)
Built on by
Citing papers per year: 2024: 2; 2025: 0; 2026: 1.
3 papers.