5 papers
Relative Score Policy Optimization for Diffusion Language Models
Zichao Yu, Shengze Xu, Bingqing Jiang +2
Diffusion large language models (dLLMs) offer a promising route to parallel and efficient text generation, but improving their reasoning ability requires effective post-training. R…
ProFlow: Zero-Shot Physics-Consistent Sampling via Proximal Flow Guidance
Zichao Yu, Ming Li, Wenyi Zhang +2
Inferring physical fields from sparse observations while strictly satisfying partial differential equations (PDEs) is a fundamental challenge in computational physics. Recently, de…
Tree Reward-Aligned Search for TReASURe in Masked Diffusion Language Models
Zichao Yu, Ming Li, Wenyi Zhang +1
Tree search has recently emerged as a powerful framework for aligning generative models with task-specific rewards at test time. Applying tree search to Masked Diffusion Language M…
AB-Cache: Training-Free Acceleration of Diffusion Models via Adams-Bashforth Cached Feature Reuse
Zichao Yu, Zhen Zou, Guojiang Shao +6
Diffusion models have demonstrated remarkable success in generative tasks, yet their iterative denoising process results in slow inference, limiting their practicality. While exist…
Data-Driven Neural Estimation of Indirect Rate-Distortion Function
Zichao Yu, Qiang Sun, Wenyi Zhang
The rate-distortion function (RDF) has long been an information-theoretic benchmark for data compression. As its natural extension, the indirect rate-distortion function (iRDF) cor…