Model-Aware Schedules Improve Generation via Fiberwise Optimal Transport

arXiv:2609.11842 · cs.LG, cs.AI · Submitted 2026-09-10 · Read on arXiv

cs.LG, cs.AI

Submitted: 2026-09-10

Updated: 2026-09-10

Comments: 11 pages, 2 figures. Keywords: Diffusion models; flow matching; schedule optimization; fiberwise optimal transport; time reparameterization; empirical universality

License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/

Terminology

• Optimal Transport — 3× in this paper · explained in 5 episodes
• Flow Matching — 2× in this paper · explained in 8 episodes
• Model-Agnostic — 2× in this paper · explained in 2 episodes

Related papers

• Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
• AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
• Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
• Convergence issues in Relational Concept Analysis based on AOC-posets
• Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
• Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks