arXiv · 2506.09018
Edit Flows: Flow Matching with Edit Operations
Abstract
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.
Explore related subjects
Keep this discovery
Marton Havasi, Brian Karrer, Itai Gat, Ricky T. Q. Chen. 2025-06-10. Edit Flows: Flow Matching with Edit Operations. https://arxiv.org/abs/2506.09018
Cite the original work for its findings. Save a collection to share your selection of sources.