publications

2025

  1. arXiv
    MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal Control
    Yuchen Zhu*, Wei Guo*, Jaemoo Choi, Guan-Horng Liu, Yongxin Chen, and Molei Tao
    Preprint, 2025
    HTML
  2. 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
    Preprint, 2025
    HTML
  3. 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
  4. 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
  5. 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
  6. 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
  7. 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
    ICLR FPI Workshop, 2025
    Long version under review
    HTML

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. ESAIM
    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. arXiv
    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