{"id":"arxiv/2606.28301","url":"https://arxiv.org/abs/2606.28301","tags":["cs.LG","cs.DS","math.NA","math.PR","stat.ML"],"score":0,"title":"VGB for Masked Diffusion Model: Efficient Test-time Scaling for Reward Satisfaction and Sample Editing","domain":"cs.LG","source":"arxiv","authors":["Kijung Jeon","Thuy-Duong Vuong","Molei Tao"],"reviews":0,"subject":"cs.LG","summary":"Inference-time scaling is a promising paradigm to improve generative models, especially when outputs must satisfy structural constraints or optimize downstream rewards. We consider Masked Diffusion Model (MDM) and introduce MDM-VGB, a discrete diffusion sampler that augments unmasking generation with theoretically principled reward-guided remasking. Inspired by the recent success of the classical Jerrum-Sinclair backtracking Markov chain in reward-tilted generation, MDM-VGB extends the backtracking random walk from a fixed prefix tree to a masked-state graph, allowing tokens to be unmasked…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-06-26","reactions":{}}