提出可任意位置插入词元的生成模型,突破传统顺序生成限制。
Insertion Language Models: Sequence Generation with Arbitrary-Position Insertions
- 通过联合选择插入位置和词元,实现任意顺序序列生成
- 在规划任务中优于自回归与掩码扩散模型,在文本补全上灵活且准确
- 适合需要复杂约束或非顺序依赖的生成场景
自回归模型(ARMs)虽在序列生成中表现优异,但难以处理需满足复杂约束或非左到右依赖的序列。掩码扩散模型(MDMs)虽部分缓解此问题,但在同时解掩码多个词元时易引入不连贯性,且无法处理未知数量词元的任意填充。本文提出插入语言模型(ILMs),学习在序列任意位置插入词元——即同时选择插入位置与词元。通过逐个插入词元,ILMs能建模强词元依赖关系,并支持任意顺序生成,适用于非线性依赖结构的序列。我们设计专用网络参数化并采用简单去噪目标训练ILMs。实证表明,ILMs在常见规划任务中优于ARMs和MDMs;在无条件文本生成中性能接近ARMs,同时在任意长度文本补全任务中比MDMs更具灵活性。代码已公开:https://dhruveshp.com/projects/ilm。
原文摘要 · Abstract (English)
Autoregressive models (ARMs), which predict subsequent tokens one-by-one ``from left to right,'' have achieved significant success across a wide range of sequence generation tasks. However, they struggle to accurately represent sequences that require satisfying sophisticated constraints or whose sequential dependencies are better addressed by out-of-order generation. Masked Diffusion Models (MDMs) address some of these limitations, but the process of unmasking multiple tokens simultaneously in MDMs can introduce incoherences, and MDMs cannot handle arbitrary infilling constraints when the number of tokens to be filled in is not known in advance. In this work, we introduce Insertion Language Models (ILMs), which learn to insert tokens at arbitrary positions in a sequence -- that is, they select jointly both the position and the vocabulary element to be inserted. By inserting tokens one at a time, ILMs can represent strong dependencies between tokens, and their ability to generate sequences in arbitrary order allows them to accurately model sequences where token dependencies do not follow a left-to-right sequential structure. To train ILMs, we propose a tailored network parameterization and use a simple denoising objective. Our empirical evaluation demonstrates that ILMs outperform both ARMs and MDMs on common planning tasks. Furthermore, we show that ILMs outperform MDMs and perform on par with ARMs in an unconditional text generation task while offering greater flexibility than MDMs in arbitrary-length text infilling. The code is available at: https://dhruveshp.com/projects/ilm .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。