English
Related papers

Related papers: Structure-Aware Fill-in-the-Middle Pretraining for…

200 papers

We introduce Syntax-Aware Fill-In-the-Middle (SAFIM), a new benchmark for evaluating Large Language Models (LLMs) on the code Fill-in-the-Middle (FIM) task. This benchmark focuses on syntax-aware completions of program structures such as…

Computation and Language · Computer Science 2024-06-25 Linyuan Gong , Sida Wang , Mostafa Elhoushi , Alvin Cheung

Large Language Models (LLMs) have significantly advanced code completion, yet they often fail when the developer's intent is underspecified in the code context. To address this, developers usually add natural language instructions (e.g.,…

Software Engineering · Computer Science 2025-10-14 Zhensu Sun , Chengran Yang , Chao Peng , Pengfei Gao , Xiaoning Du , Li Li , David Lo

Fill-in-the-Middle (FIM) models play a vital role in code completion tasks, leveraging both prefix and suffix context to provide more accurate and contextually relevant suggestions. This paper presents approaches to improve FIM code…

Information Retrieval · Computer Science 2024-12-24 Hitesh Sagtani , Rishabh Mehrotra , Beyang Liu

Fill-in-the-middle (FIM) is a pretraining objective widely used to equip causal language models with infilling ability, yet its effect on verbatim memorization remains underexplored. We study the memorization dynamics of FIM in a controlled…

Computation and Language · Computer Science 2026-05-25 Tobias von Arx , Tanguy Dieudonné

Code completion is a prominent application of Large Language Models (LLMs) in software engineering. Due to the near real-time response requirements of this task, base models with small to medium-sized parameters are typically employed,…

Software Engineering · Computer Science 2025-09-18 Dongjun Yu , Xiao Yan , Zhenrui Li , Jipeng Xiao , Haochuan He , Yongda Yu , Hao Zhang , Guoping Rong , Xiaobo Huang

Large Language Models are powerful tools for program synthesis and advanced auto-completion, but come with no guarantee that their output code is syntactically correct. This paper contributes an incremental parser that allows early…

Programming Languages · Computer Science 2024-09-06 Daniel Melcer , Nathan Fulton , Sanjay Krishna Gouda , Haifeng Qian

Fill-in-the-Middle (FIM), or infilling, has become integral to code language models, enabling generation of missing code given both left and right contexts. However, the current FIM training paradigm which performs next-token prediction…

Machine Learning · Computer Science 2025-11-20 Yifeng Ding , Hantian Ding , Shiqi Wang , Qing Sun , Varun Kumar , Zijian Wang

Large language models (LLMs) have made significant advancements in code-related tasks, yet many LLMs treat code as simple sequences, neglecting its structured nature. We introduce AST-T5, a novel pretraining paradigm that leverages the…

Software Engineering · Computer Science 2024-06-25 Linyuan Gong , Mostafa Elhoushi , Alvin Cheung

We show that autoregressive language models can learn to infill text after we apply a straightforward transformation to the dataset, which simply moves a span of text from the middle of a document to its end. While this data augmentation…

Computation and Language · Computer Science 2022-07-29 Mohammad Bavarian , Heewoo Jun , Nikolas Tezak , John Schulman , Christine McLeavey , Jerry Tworek , Mark Chen

Language models have become the backbone of today's AI systems. However, their predominant left-to-right generation limits the use of bidirectional context, which is essential for tasks that involve filling text in the middle. We propose…

Computation and Language · Computer Science 2023-10-17 Tianxiao Shen , Hao Peng , Ruoqi Shen , Yao Fu , Zaid Harchaoui , Yejin Choi

Post-processing is crucial for the automatic evaluation of LLMs in fill-in-the-middle (FIM) code generation due to the frequent presence of extraneous code in raw outputs. This extraneous generation suggests a lack of awareness regarding…

Software Engineering · Computer Science 2025-06-11 Wasi Uddin Ahmad , Somshubra Majumdar , Boris Ginsburg

Protein language models (pLMs), pre-trained via causal language modeling on protein sequences, have been a promising tool for protein sequence design. In real-world protein engineering, there are many cases where the amino acids in the…

Machine Learning · Computer Science 2023-03-30 Youhan Lee , Hasun Yu

The dominant Fill-in-the-Middle (FIM) paradigm for code completion is constrained by its rigid inability to correct contextual errors and reliance on unaligned, insecure Base models. While Chat LLMs offer safety and Agentic workflows…

Software Engineering · Computer Science 2026-01-21 Jiajun Zhang , Zeyu Cui , Jiaxi Yang , Lei Zhang , Yuheng Jing , Zeyao Ma , Tianyi Bai , Zilei Wang , Qiang Liu , Liang Wang , Binyuan Hui , Junyang Lin

The code generation capabilities of Large Language Models (LLMs) have advanced applications like tool invocation and problem-solving. However, improving performance in code-related tasks remains challenging due to limited training data that…

Computation and Language · Computer Science 2025-08-28 Houxing Ren , Zimu Lu , Weikang Shi , Haotian Hou , Yunqiao Yang , Ke Wang , Aojun Zhou , Junting Pan , Mingjie Zhan , Hongsheng Li

Most language models (LMs) are trained and applied in an autoregressive left-to-right fashion, assuming that the next token only depends on the preceding ones. However, this assumption ignores the potential benefits of using the full…

Computation and Language · Computer Science 2023-03-14 Anh Nguyen , Nikos Karampatziakis , Weizhu Chen

Summarizing source code into natural language descriptions (code summarization) helps developers better understand program functionality and reduce the burden of software maintenance. Abstract Syntax Trees (ASTs), as opposed to source code,…

Software Engineering · Computer Science 2026-02-09 Shijia Dong , Haoruo Zhao , Paul Harvey

Autoregressive (AR) models can generate high-quality low-poly meshes from point clouds, but they still operate in an all-or-nothing manner: when a local region is unsatisfactory, the entire mesh must be regenerated, wasting computation and…

In vision-language pre-training (VLP), masked image modeling (MIM) has recently been introduced for fine-grained cross-modal alignment. However, in most existing methods, the reconstruction targets for MIM lack high-level semantics, and…

Computer Vision and Pattern Recognition · Computer Science 2024-03-04 Haowei Liu , Yaya Shi , Haiyang Xu , Chunfeng Yuan , Qinghao Ye , Chenliang Li , Ming Yan , Ji Zhang , Fei Huang , Bing Li , Weiming Hu

Existing scene text removal (STR) task suffers from insufficient training data due to the expensive pixel-level labeling. In this paper, we aim to address this issue by introducing a Text-aware Masked Image Modeling algorithm (TMIM), which…

Computer Vision and Pattern Recognition · Computer Science 2024-09-23 Zixiao Wang , Hongtao Xie , YuXin Wang , Yadong Qu , Fengjun Guo , Pengwei Liu

We propose fill-in-the-blanks as a video understanding evaluation framework and introduce FIBER -- a novel dataset consisting of 28,000 videos and descriptions in support of this evaluation framework. The fill-in-the-blanks setting tests a…

Computer Vision and Pattern Recognition · Computer Science 2022-03-24 Santiago Castro , Ruoyao Wang , Pingxuan Huang , Ian Stewart , Oana Ignat , Nan Liu , Jonathan C. Stroud , Rada Mihalcea
‹ Prev 1 2 3 10 Next ›