313 citations · 1.1k across the 28 of their papers we have counts for
Showing cs.LGShow all
2 papers · 1 filter
cs.LG2026
MEND: RL For Flow Models via Proximal Velocity Matching
Shreshth Saini, Neil Birkbeck, Yilin Wang +2
Reward post-training of flow models either reweights the model's own samples under a KL penalty or a frozen reference, often for thousands of updates, or backpropagates the reward…
cs.LG2026
TABES: Trajectory-Aware Backward-on-Entropy Steering for Masked Diffusion Models
Shreshth Saini, Avinab Saha, Balu Adsumilli +3
Masked Diffusion Models (MDMs) have emerged as a promising non-autoregressive paradigm for generative tasks, offering parallel decoding and bidirectional context utilization. Howev…