ARTFEED — Contemporary Art Intelligence

Entropy-Based Scheduler Boosts Flow Model Sampling Efficiency

ai-technology · 2026-05-18

Researchers have developed a training-free entropic inference-time scheduler for flow-based generative models, improving sample quality under a small inference budget. The method uses a conditional-marginal entropy-rate objective to discretize the probability path, separating bridge geometry from marginal flow evolution. For Gaussian Brownian bridges, the entropy rate is closed-form and U-shaped, motivating nonuniform grids that concentrate evaluations near boundaries. On trained 2D bridge/flow models, the scheduler improved 10-step ODE-Heun MMD by 18.1% and SDE-Heun by 22.7% over linear discretization. On EDM/CIFAR-10, the entropic time schedule also showed gains. The work is published on arXiv (2605.16126) and addresses a key challenge in flow matching and Schrödinger bridges, where inference grids are typically heuristic.

Key facts

  • Training-free entropic inference-time scheduler for flow-based generative models
  • Uses conditional-marginal entropy-rate objective for bridge-aware discretization
  • Gaussian Brownian bridges yield closed-form U-shaped entropy rate
  • Improves 10-step ODE-Heun MMD by 18.1% over linear on 2D models
  • SDE-Heun improvement of 22.7% in same low-NFE sweep
  • Tested on EDM/CIFAR-10 with positive results
  • Published on arXiv with ID 2605.16126
  • Addresses heuristic inference grids in flow matching and Schrödinger bridges

Entities

Institutions

  • arXiv

Sources