MT
Molei Tao
cs.LGstat.MLcs.NAmath.NAmath.DSphysics.comp-phmath.OCmath.PRcs.AIcs.CV
On Valency
published · living versionsW_gevsh4m2·v1 · currentpublished
Multi-Draft On-Policy Distillation for Diffusion Drafters
with Petr Molodyk, Yuchen Zhu, Jaemoo Choi, Yongxin Chen
1 version
W_rfpk473u·v1published
Preprints & journals
72 papers in the corpus · 2009–2026Imperfect Verifiers, Exact Targets: Inference-Time Scaling for Continuous Generation2610.15281v1 · Kijung Jeon, Molei Tao2026 · 0 citationsarXiv
Improving score-based sampling via affine post-processing2610.04773v1 · Curtis McDonald, Kevin Rojas, Molei Tao et al.2026 · 0 citationsarXiv
Large Language Continuous Diffusion Models2610.02665v1 · Zhihan Yang, Wei Guo, Jean-Marie Lemercier et al.2026 · 0 citationsarXiv
Fast Non-Log-Concave Sampling under Nonconvex Equality and Inequality Constraints with Landing2510.22044v1 · Kijung Jeon, Michael Muehlebach, Molei Tao2025 · 0 citationsNeurIPS 2025
AlignFlow: Improving Flow-based Generative Models with Semi-Discrete Optimal Transport2510.15038v2 · Lingkai Kong, Molei Tao, Yang Liu et al.2025 · 0 citationsarXiv
Enhancing Reasoning for Diffusion LLMs via Distribution Matching Policy Optimization2510.08233v3 · Yuchen Zhu, Wei Guo, Jaemoo Choi et al.2025 · 0 citationsarXiv
Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization2510.08554v3 · Kevin Rojas, Jiahe Lin, Kashif Rasul et al.2025 · 0 citationsarXiv
Proximal Diffusion Neural Sampler2510.03824v2 · Wei Guo, Jaemoo Choi, Yuchen Zhu et al.2025 · 0 citationsarXiv
TR2-D2: Tree Search Guided Trajectory-Aware Fine-Tuning for Discrete Diffusion.41256882 · Tang, Sophia, Zhu, Yuchen, Tao, Molei et al.2025 · 1 citationArXiv. 2025
A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspective2509.16499v3 · Lianghe Shi, Meng Wu, Huijie Zhang et al.2025 · 1 citationarXiv
MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal Control2508.10684v2 · Yuchen Zhu, Wei Guo, Jaemoo Choi et al.2025 · 0 citationsarXiv
Non-equilibrium Annealed Adjoint Sampler2506.18165v3 · Jaemoo Choi, Yongxin Chen, Molei Tao et al.2025 · 0 citationsarXiv
Variational Learning Finds Flatter Solutions at the Edge of Stability2506.12903v3 · Avrajit Ghosh, Bai Cong, Rio Yokota et al.2025 · 0 citationsNeurIPS-2025
Diffuse Everything: Multimodal Diffusion Models on Arbitrary State Spaces2506.07903v2 · Kevin Rojas, Yuchen Zhu, Sichen Zhu et al.2025 · 0 citationsarXiv
Complexity Analysis of Normalizing Constant Estimation: from Jarzynski Equality to Annealed Importance Sampling and beyond2502.04575v3 · Wei Guo, Molei Tao, Yongxin Chen2025 · 0 citationsarXiv
Fast Solvers for Discrete Diffusion Models: Theory and Applications of High-Order Algorithms2502.00234v2 · Yinuo Ren, Haoxuan Chen, Yuchen Zhu et al.2025 · 0 citationsarXiv
SODA: Spectral Orthogonal Decomposition Adaptation for Diffusion Models.40385571 · Zhang, Xinxi, Wen, Song, Han, Ligong et al.2026 · 2 citationsIEEE Winter Conference on Applications of Computer Vision. IEEE Winter Conference on Applications of Computer Vision. 2025;2025:4665-4682
Variational Schrödinger Momentum Diffusion2501.16675v1 · Kevin Rojas, Yixin Tan, Molei Tao et al.2025 · 0 citationsarXiv
Diffusion Generative Modeling for Spatially Resolved Gene Expression Inference from Histology Images2501.15598v1 · Sichen Zhu, Yuchen Zhu, Molei Tao et al.2025 · 1 citationarXiv
Generative modeling assisted simulation of measurement-altered quantum criticality2412.01513v1 · Yuchen Zhu, Molei Tao, Yuebo Jin et al.2024 · 1 citationarXiv
Provable Acceleration of Nesterov's Accelerated Gradient for Rectangular Matrix Factorization and Linear Neural Networks2410.09640v3 · Zhenghao Xu, Yuqing Wang, Tuo Zhao et al.2024 · 0 citationsarXiv
Can specific THz fields induce collective base-flipping in DNA? A stochastic averaging and resonant enhancement investigation based on a new mesoscopic model.39177954 · Koon, Wang Sang, Owhadi, Houman, Tao, Molei et al.2024 · 3 citationsChaos (Woodbury, N.Y.). 2024;34(8)
Robust First and Second-Order Differentiation for Regularized Optimal Transport2407.02015v2 · Xingjie Li, Fei Lu, Molei Tao et al.2024 · 1 citationarXiv
Evaluating the design space of diffusion-based generative models2406.12839v4 · Yuqing Wang, Ye He, Molei Tao2024 · 4 citationsarXiv
Spectrum-Aware Parameter Efficient Fine-Tuning for Diffusion Models2405.21050v1 · Xinxi Zhang, Song Wen, Ligong Han et al.2024 · 3 citationsarXiv
Quantitative Convergences of Lie Group Momentum Optimizers2405.20390v1 · Lingkai Kong, Molei Tao2024 · 0 citationsarXiv
Trivialized Momentum Facilitates Diffusion Generative Modeling on Lie Groups2405.16381v2 · Yuchen Zhu, Tianrong Chen, Lingkai Kong et al.2024 · 0 citationsarXiv
Quantum State Generation with Structure-Preserving Diffusion Model2404.06336v2 · Yuchen Zhu, Tianrong Chen, Evangelos A. Theodorou et al.2024 · 0 citationsarXiv
Convergence of Kinetic Langevin Monte Carlo on Lie groups2403.12012v2 · Lingkai Kong, Molei Tao2024 · 0 citationsarXiv
Zeroth-Order Sampling Methods for Non-Log-Concave Distributions: Alleviating Metastability by Denoising Diffusion2402.17886v4 · Ye He, Kevin Rojas, Molei Tao2024 · 0 citationsarXiv
Automated construction of effective potential via algorithmic implicit bias2401.03511v1 · Xingjie Helen Li, Molei Tao2024 · 0 citationsarXiv
Appropriate State-Dependent Friction Coefficient Accelerates Kinetic Langevin Dynamics2312.07817v3 · Keunwoo Lim, Molei Tao2023 · 1 citationarXiv
DFU: scale-robust diffusion model for zero-shot super-resolution image generation2401.06144v2 · Alex Havrilla, Kevin Rojas, Wenjing Liao et al.2023 · 0 citationsarXiv
Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapult2310.17087v2 · Yuqing Wang, Zhenghao Xu, Tuo Zhao et al.2023 · 0 citationsarXiv
Mirror Diffusion Models for Constrained and Watermarked Generation2310.01236v2 · Guan-Horng Liu, Tianrong Chen, Evangelos A. Theodorou et al.2023 · 5 citationsarXiv
Deep Momentum Multi-Marginal Schrödinger Bridge2303.01751v3 · Tianrong Chen, Guan-Horng Liu, Molei Tao et al.2023 · 1 citationarXiv
Essential barrier height and a probabilistic approach in characterizing potential landscape2301.01447v2 · Yao Li, Molei Tao, Shirou Wang2023 · 0 citationsarXiv
Data-driven discovery of non-Newtonian astronomy via learning non-Euclidean Hamiltonian2210.00090v1 · Oswin So, Gongjie Li, Evangelos A. Theodorou et al.2022 · 1 citationarXiv
Low Spin-Axis Variations of Circumbinary Planets2208.00018v1 · Renyi Chen, Gongjie Li, Molei Tao2022 · 1 citationarXiv
NySALT: Nyström-type inference-based schemes adaptive to large time-stepping2207.06012v1 · Xingjie Li, Fei Lu, Molei Tao et al.2022 · 0 citationsarXiv
gDDIM: Generalized denoising diffusion implicit models2206.05564v2 · Qinsheng Zhang, Molei Tao, Yongxin Chen2022 · 33 citationsarXiv
Momentum Stiefel Optimizer, with Applications to Suitably-Orthogonal Attention, and Optimal Transport2205.14173v3 · Lingkai Kong, Yuqing Wang, Molei Tao2022 · 1 citationICLR 2023
Large Learning Rate Tames Homogeneity: Convergence and Balancing Effect2110.03677v2 · Yuqing Wang, Minshuo Chen, Tuo Zhao et al.2021 · 6 citationsICLR 2022
Parametric resonance for enhancing the rate of metastable transition2109.14385v1 · Ying Chao, Molei Tao2021 · 5 citationsarXiv
The Mirror Langevin Algorithm Converges with Vanishing Bias2109.12077v2 · Ruilin Li, Molei Tao, Santosh S. Vempala et al.2021 · 4 citationsarXiv
Sqrt(d) Dimension Dependence of Langevin Monte Carlo2109.03839v3 · Ruilin Li, Hongyuan Zha, Molei Tao2021 · 5 citationsICLR 2022
Variational Symplectic Accelerated Optimization on Lie Groups2103.14166v1 · Taeyoung Lee, Molei Tao, Melvin Leok2021 · 5 citationsarXiv
Data-driven Prediction of General Hamiltonian Dynamics via Learning Exactly-Symplectic Maps2103.05632v2 · Renyi Chen, Molei Tao2021 · 29 citationsarXiv
Accurate and Efficient Simulations of Hamiltonian Mechanical Systems with Discontinuous Potentials2101.11018v1 · Molei Tao, Shi Jin2021 · 4 citationsarXiv
Hessian-Free High-Resolution Nesterov Acceleration for Sampling2006.09230v4 · Ruilin Li, Hongyuan Zha, Molei Tao2020 · 3 citationsarXiv
Career total: 95 works. 72 are in this corpus.Showing the 50 most recent.
Profile built from the corpus for this byline.
Author records are still filling in while the Hub is in alpha. If this is your page, you'll be able to claim it soon. Spot a mistake? Tell us.