Efficient Training of Language Models to Fill in the Middle

Authors: Mohammad Bavarian, Heewoo Jun, Nikolas Tezak, John Schulman, Christine McLeavey, Jerry Tworek, Mark Chen

License: CC BY 4.0
AI-generated keywords: Autoregressive language models Data augmentation Fill-in-the-middle (FIM) Training methodology Natural language processing

AI-generated Key Points

  • Autoregressive language models are effective in filling missing text using the fill-in-the-middle (FIM) technique.
  • Using FIM does not compromise the generative capabilities of autoregressive language models.
  • Future autoregressive language models should be trained using FIM by default for practicality and efficiency.
  • Best practices for training FIM models include exploring key hyperparameters and making best-performing infilling model available through an API.
  • The study discusses techniques and results from pretraining and finetuning experiments, providing recommendations for future research directions.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Abstract: 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 has garnered much interest in recent years, we provide extensive evidence that training models with a large fraction of data transformed in this way does not harm the original left-to-right generative capability, as measured by perplexity and sampling evaluations across a wide range of scales. Given the usefulness, simplicity, and efficiency of training models to fill-in-the-middle (FIM), we suggest that future autoregressive language models be trained with FIM by default. To this end, we run a series of ablations on key hyperparameters, such as the data transformation frequency, the structure of the transformation, and the method of selecting the infill span. We use these ablations to prescribe strong default settings and best practices to train FIM models. We have released our best infilling model trained with best practices in our API, and release our infilling benchmarks to aid future research.

Submitted to arXiv on 28 Jul. 2022

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2207.14255v1

  • Comprehensive Summary
  • Key points
  • Layman's Summary
  • Blog article
In this study, we demonstrate the effectiveness of autoregressive language models in filling missing text through a simple data transformation technique known as fill-in-the-middle (FIM). Despite its growing popularity, our research shows that using FIM does not compromise the original generative capabilities of these models. We propose that future autoregressive language models should be trained using FIM by default due to its practicality and efficiency. To establish best practices for training FIM models, we conducted experiments exploring key hyperparameters and have made our best-performing infilling model available through our API. Additionally, we discuss various techniques and results from pretraining and finetuning experiments and provide recommendations for future research directions. Our study contributes valuable insights into enhancing autoregressive language models through FIM training methodology.

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.