Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2026
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2602.18695 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911461193809920 |
|---|---|
| author | Patel, Dhruvesh Rozonoyer, Benjamin Pandey, Gaurav Naseem, Tahira Astudillo, Ramón Fernandez McCallum, Andrew |
| author_facet | Patel, Dhruvesh Rozonoyer, Benjamin Pandey, Gaurav Naseem, Tahira Astudillo, Ramón Fernandez McCallum, Andrew |
| contents | In many domains generating variable length sequences through insertions provides greater flexibility over autoregressive models. However, the action space of insertion models is much larger than that of autoregressive models (ARMs) making the learning challenging. To address this, we incorporate trainable order dynamics into the target rates for discrete flow matching, and show that with suitable choices of parameterizations, joint training of the target order dynamics and the generator is tractable without the need for numerical simulation. As the generative insertion model, we use a variable length masked diffusion model, which generates by inserting and filling mask tokens. On graph traversal tasks for which a locally optimal insertion order is known, we explore the choices of parameterization empirically and demonstrate the trade-offs between flexibility, training stability and generation quality. On de novo small molecule generation, we find that the learned order dynamics leads to an increase in the number of valid molecules generated and improved quality, when compared to uniform order dynamics. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2602_18695 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Insertion Based Sequence Generation with Learnable Order Dynamics Patel, Dhruvesh Rozonoyer, Benjamin Pandey, Gaurav Naseem, Tahira Astudillo, Ramón Fernandez McCallum, Andrew Machine Learning In many domains generating variable length sequences through insertions provides greater flexibility over autoregressive models. However, the action space of insertion models is much larger than that of autoregressive models (ARMs) making the learning challenging. To address this, we incorporate trainable order dynamics into the target rates for discrete flow matching, and show that with suitable choices of parameterizations, joint training of the target order dynamics and the generator is tractable without the need for numerical simulation. As the generative insertion model, we use a variable length masked diffusion model, which generates by inserting and filling mask tokens. On graph traversal tasks for which a locally optimal insertion order is known, we explore the choices of parameterization empirically and demonstrate the trade-offs between flexibility, training stability and generation quality. On de novo small molecule generation, we find that the learned order dynamics leads to an increase in the number of valid molecules generated and improved quality, when compared to uniform order dynamics. |
| title | Insertion Based Sequence Generation with Learnable Order Dynamics |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2602.18695 |