English

GLASS Flows: Transition Sampling for Alignment of Flow and Diffusion Models

Machine Learning 2026-02-12 v3 Artificial Intelligence Machine Learning

Abstract

The performance of flow matching and diffusion models can be greatly improved at inference time using reward alignment algorithms, yet efficiency remains a major limitation. While several algorithms were proposed, we demonstrate that a common bottleneck is the sampling method these algorithms rely on: many algorithms require to sample Markov transitions via SDE sampling, which is significantly less efficient and often less performant than ODE sampling. To remove this bottleneck, we introduce GLASS Flows, a new sampling paradigm that simulates a "flow matching model within a flow matching model" to sample Markov transitions. As we show in this work, this "inner" flow matching model can be retrieved from a pre-trained model without any re-training, combining the efficiency of ODEs with the stochastic evolution of SDEs. On large-scale text-to-image models, we show that GLASS Flows eliminate the trade-off between stochastic evolution and efficiency. Combined with Feynman-Kac Steering, GLASS Flows improve state-of-the-art performance in text-to-image generation, making it a simple, drop-in solution for inference-time scaling of flow and diffusion models.

Keywords

Cite

@article{arxiv.2509.25170,
  title  = {GLASS Flows: Transition Sampling for Alignment of Flow and Diffusion Models},
  author = {Peter Holderrieth and Uriel Singer and Tommi Jaakkola and Ricky T. Q. Chen and Yaron Lipman and Brian Karrer},
  journal= {arXiv preprint arXiv:2509.25170},
  year   = {2026}
}
R2 v1 2026-07-01T06:05:26.659Z