about
In the Middle: A New Pre-Training Paradigm (arxiv.org)
1 point by tosh on Apr 26, 2023 | hide | past | pdf | discuss on HN

In plain words: Two models read the same text in opposite directions and are pushed to agree; at test time both write toward the middle to fill a gap. It beat strong baselines on code and language while learning from less data.

Abstract · Meet in the Middle: A New Pre-training Paradigm

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 sequence information during training, and the possibility of having context from both sides during inference. In this paper, we propose a new pre-training paradigm with techniques that jointly improve the training data efficiency and the capabilities of the LMs in the infilling task. The first is a training objective that aligns the predictions of a left-to-right LM with those of a right-to-left LM, trained on the same data but in reverse order. The second is a bidirectional inference procedure that enables both LMs to meet in the middle. We show the effectiveness of our pre-training paradigm with extensive experiments on both programming and natural language models, outperforming strong baselines.

Anh Nguyen, Nikos Karampatziakis, Weizhu Chen
arXiv:2303.07295 · cs.CL, cs.LG · submitted Mar 13, 2023
abstract · pdf · html · 24 pages, 2 figures

add comment on HN