Preprints & Publications

2025

  1. Complexity Analysis of Normalizing Constant Estimation: from Jarzynski Equality to Annealed Importance Sampling and beyond
    Wei Guo, Molei Tao, and Yongxin Chen
    arXiv preprint arXiv:2502.04575, 2025
    arXiv
  2. Provable Benefit of Annealed Langevin Monte Carlo for Non-log-concave Sampling
    Wei Guo, Molei Tao, and Yongxin Chen
    In The Thirteenth International Conference on Learning Representations, 2025
    arXiv HTML Poster
  3. Fast solvers for discrete diffusion models: Theory and applications of high-order algorithms
    Yinuo Ren*, Haoxuan Chen*, Yuchen Zhu*, Wei Guo*, Yongxin Chen, Grant M Rotskoff, Molei Tao, and Lexing Ying
    arXiv preprint arXiv:2502.00234, 2025
    arXiv
  4. MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal Control
    Yuchen Zhu*, Wei Guo*, Jaemoo Choi, Guan-Horng Liu, Yongxin Chen, and Molei Tao
    arXiv preprint arXiv:2508.10684, 2025
    arXiv

2024

  1. Plug-and-Play Controllable Generation for Discrete Masked Models
    Wei Guo*, Yuchen Zhu*, Molei Tao, and Yongxin Chen
    arXiv preprint arXiv:2410.02143, 2024
    arXiv