Edit Flows: Flow Matching with Edit Operations

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Havasi, Marton, Karrer, Brian, Gat, Itai, Chen, Ricky T. Q.
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915612466348032
author Havasi, Marton
Karrer, Brian
Gat, Itai
Chen, Ricky T. Q.
author_facet Havasi, Marton
Karrer, Brian
Gat, Itai
Chen, Ricky T. Q.
contents Autoregressive generative models naturally generate variable-length sequences, while non-autoregressive models struggle, often imposing rigid, token-wise structures. We propose Edit Flows, a non-autoregressive model that overcomes these limitations by defining a discrete flow over sequences through edit operations$\unicode{x2013}$insertions, deletions, and substitutions. By modeling these operations within a Continuous-time Markov Chain over the sequence space, Edit Flows enable flexible, position-relative generation that aligns more closely with the structure of sequence data. Our training method leverages an expanded state space with auxiliary variables, making the learning process efficient and tractable. Empirical results show that Edit Flows outperforms both autoregressive and mask models on image captioning and significantly outperforms the mask construction in text and code generation.
format Preprint
id arxiv_https___arxiv_org_abs_2506_09018
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Edit Flows: Flow Matching with Edit Operations
Havasi, Marton
Karrer, Brian
Gat, Itai
Chen, Ricky T. Q.
Machine Learning
Artificial Intelligence
Autoregressive generative models naturally generate variable-length sequences, while non-autoregressive models struggle, often imposing rigid, token-wise structures. We propose Edit Flows, a non-autoregressive model that overcomes these limitations by defining a discrete flow over sequences through edit operations$\unicode{x2013}$insertions, deletions, and substitutions. By modeling these operations within a Continuous-time Markov Chain over the sequence space, Edit Flows enable flexible, position-relative generation that aligns more closely with the structure of sequence data. Our training method leverages an expanded state space with auxiliary variables, making the learning process efficient and tractable. Empirical results show that Edit Flows outperforms both autoregressive and mask models on image captioning and significantly outperforms the mask construction in text and code generation.
title Edit Flows: Flow Matching with Edit Operations
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2506.09018