Publications
(*) denotes equal contribution.
Selected Publications (* indicates equal contribution or corresponding author):
- EM Generalist: A physics-driven diffusion foundation model for electron microscopy
H Sun, E Ye, C Lu, Z Jiang, W Bai, S Ren, C Wang, J Zhang, R Shi, L Ma, ...
2025 - Microscopic image restoration and uncertainty quantification using physics-informed generative models
W Bai, S Ren, E Ye, W Tang, H Sun
Computational Optical Imaging and Artificial Intelligence in Biomedical …, 2025 - InstantViR: Real-Time Video Inverse Problem Solver with Distilled Diffusion Prior
W Bai, S Xu, Y Ren, J Hao, M Sun, W Chen, H Sun
arXiv preprint arXiv:2511.14208, 2025 - Masked Auto-Regressive Variational Acceleration: Fast Inference Makes Practical Reinforcement Learning
Y Gu, W Bai, Y Wang, W Luo, H Sun
arXiv preprint arXiv:2511.15190, 2025 - Let Language Constrain Geometry: Vision-Language Models as Semantic and Spatial Critics for 3D Generation
W Bai, Y Li, W Luo, Z Lai, Y Wang, W Chen, H Sun
arXiv preprint arXiv:2511.14271, 2025 - Dive3D: Diverse Distillation-based Text-to-3D Generation via Score Implicit Matching
W Bai, Y Li, W Chen, W Luo, H Sun
Transactions on Machine Learning Research, 2025 - Vision-Language Models as Differentiable Semantic and Spatial Rewards for Text-to-3D Generation
W Bai, Y Li, W Luo, W Chen, H Sun
arXiv preprint arXiv:2509.15772, 2025 - Blind inversion using latent diffusion priors
W Bai, S Chen, W Chen, H Sun
SIGGRAPH Asia 2025 Poster (IEEE Transactions on Image Processing), 2025 - Uni-Instruct: One-step Diffusion Model through Unified Diffusion Divergence Instruction
Y Wang, W Bai, C Zhang, D Zhang, W Luo, H Sun
Advances in Neural Information Processing Systems, 2025 - Learning diffusion model from noisy measurement using principled expectation-maximization method
W Bai, W Tang, E Ye, S Chen, W Chen, H Sun
ICASSP 2025-2025 IEEE International Conference on Acoustics, Speech and …, 2025 - An expectation-maximization algorithm for training clean diffusion models from corrupted observations
W Bai, Y Wang, W Chen, H Sun
Advances in Neural Information Processing Systems, 2024 - Integrating amortized inference with diffusion models for learning clean distribution from corrupted images
Y Wang, W Bai, W Luo, W Chen, H Sun
arXiv preprint arXiv:2407.11162, 2024 - A temporal kernel for time series forecasting
W Bai, H Xue
Knowledge-Based Systems, 113623, 2023 - Learning Deeper Non-Monotonic Networks by Softly Transferring Solution Space
ZF Wu, H Xue, W Bai
IJCAI, 3200-3206, 2021