{"id":"arxiv/2601.18681","url":"https://arxiv.org/abs/2601.18681","tags":["cs.LG","cs.AI","eess.SY","math.OC"],"score":0,"title":"ART for Diffusion Sampling: A Reinforcement Learning Approach to Timestep Schedule","domain":"cs.LG","source":"arxiv","authors":["Yilie Huang","Wenpin Tang","Xunyu Zhou"],"reviews":0,"subject":"cs.LG","summary":"We consider time discretization for score-based diffusion models to generate samples from a learned reverse-time dynamic on a finite grid. Uniform and hand-crafted grids can be suboptimal given a budget on the number of time steps. We introduce Adaptive Reparameterized Time (ART), which controls the clock speed of a reparameterized time variable to redistribute computation along the sampling trajectory while preserving the terminal time, with the objective of minimizing the aggregate Euler discretization error. We derive a randomized companion ART-RL that recasts ART as a continuous-time…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-01-26","reactions":{}}