publications

2025

  1. dmpo.png
    Enhancing Reasoning for Diffusion LLMs via Distribution Matching Policy Optimization
    Yuchen Zhu*, Wei Guo*, Jaemoo Choi, Petr Molodyk, Bo Yuan, Molei Tao, and Yongxin Chen
    Preprint, 2025
    HTML Code
  2. pdns.png
    Proximal Diffusion Neural Sampler
    Wei Guo*, Jaemoo Choi*, Yuchen Zhu, Molei Tao, and Yongxin Chen
    Preprint, 2025
    HTML
  3. mdns.png
    MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal Control
    Yuchen Zhu*, Wei Guo*, Jaemoo Choi, Guan-Horng Liu, Yongxin Chen, and Molei Tao
    Advances in Neural Information Processing Systems (NeurIPS), 2025
    HTML Code
  4. tr2d2.png
    TR2-D2: Tree Search Guided Trajectory-Aware Fine-Tuning for Discrete Diffusion
    Sophia Tang*, Yuchen Zhu*, Molei Tao, and Pranam Chatterjee
    Preprint, 2025
    HTML Code
  5. fast-solver.png
    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
    Advances in Neural Information Processing Systems (NeurIPS), 2025
    HTML
  6. mimicking-reasoning.png
    Mimicking or Reasoning: Rethinking Multi-Modal In-Context Learning in Vision-Language Models
    Chengyue Huang*, Yuchen Zhu*, Sichen Zhu*, Jingyun Xiao, Moises Andrade, Shivang Chopra, and Zsolt Kira
    ICCV 2025 MARS2 Workshop, 2025
    HTML
  7. diffuse-everything.jpg
    Diffuse Everything: Multimodal Diffusion Models on Arbitrary State Spaces
    Kevin Rojas*, Yuchen Zhu*, Sichen Zhu, Felix Ye, and Molei Tao
    International Conference on Machine Learning (ICML), 2025
    HTML Code Poster
  8. learning-to-stop.png
    Learning to Stop: Deep Learning for Mean Field Optimal Stopping
    Lorenzo Magnino*, Yuchen Zhu*, and Mathieu Lauriere
    International Conference on Machine Learning (ICML), 2025
    HTML Code Poster
  9. trivialized-momentum.png
    Trivialized Momentum Facilitates Diffusion Generative Modeling on Lie Groups
    Yuchen Zhu*, Tianrong Chen*, Lingkai Kong, Evangelos Theodorou, and Molei Tao
    International Conference on Learning Representations (ICLR), 2025
    HTML Code Poster
  10. diffusion-gene-expression.png
    Diffusion Generative Modeling for Spatially Resolved Gene Expression Inference from Histology Images
    Sichen Zhu*, Yuchen Zhu*, Molei Tao, and Peng Qiu
    International Conference on Learning Representations (ICLR), 2025
    HTML Code Poster

2024

  1. plug-play-generation.png
    Plug-and-Play Controllable Generation for Discrete Masked Models
    Wei Guo*, Yuchen Zhu*, Molei Tao, and Yongxin Chen
    Preprint, 2024
    HTML
  2. quantum-generation.png
    Quantum Generation with Structure Preserving Diffusion Model
    Yuchen Zhu, Tianrong Chen, Evangelos A Theodorou, Xie Chen, and Molei Tao
    Preprint, 2024
    HTML
  3. mfot.png
    Deep Learning for Mean Field Optimal Transport
    Sebastian Baudelet†, Brieuc Frenais†, Mathieu Lauriere†, Amal Machtalay†, and Yuchen Zhu†
    ESAIM: PROCEEDINGS AND SURVEYS 77, 2024
    HTML
  4. mean-field.png
    A Mean-Field Analysis of Neural Stochastic Gradient Descent-Ascent for Functional Minimax Optimization
    Yuchen Zhu, Yufeng Zhang, Zhaoran Wang, Zhuoran Yang, and Xiaohong Chen
    Preprint, 2024
    HTML Poster