Publications

2025

STeP: A General and Scalable Framework for Solving Video Inverse Problems with Spatiotemporal Diffusion Priors

Bingliang Zhang*, Zihui Wu*, Berthy T. Feng, Yisong Yue, Katherine L. Bouman

A Unified Model for Compressed Sensing MRI Across Undersampling Patterns

Armeet Singh Jatyani, Jiayun Wang, Aditi Chandrashekar, Zihui Wu, Miguel Liu-Schiaffini, Bahareh Tolooshams, Anima Anandkumar

CVPR 2025 / Paper

InverseBench: Benchmarking Plug-and-Play Diffusion Priors for Inverse Problems in Physical Sciences

Hongkai Zheng*, Wenda Chu*, Bingliang Zhang*, Zihui Wu*, Austin Wang, Berthy T. Feng, Caifeng Zou, Yu Sun, Nikola Borislavov Kovachki, Zachary E. Ross, Katherine L. Bouman, Yisong Yue

ICLR 2025 / Paper / Code / Project Page

Spotlight

2024

Principled Probabilistic Imaging using Diffusion Models as Plug-and-Play Priors

Zihui Wu, Yu Sun, Yifan Chen, Bingliang Zhang, Yisong Yue, Katherine L. Bouman

NeurIPS 2024 / Paper / Project Page

2023

Provable Probabilistic Imaging using Score-Based Generative Priors

Yu Sun, Zihui Wu, Yifan Chen, Berthy Feng, Katherine L. Bouman

IEEE TCI (under review) / Paper

Demystifying Oversmoothing in Attention-Based Graph Neural Networks

Xinyi Wu, Amir Ajorlou, Zihui Wu, Ali Jadbabaie

NeurIPS 2023 / Paper

Spotlight

Learning Task-Specific Strategies for Accelerated MRI

Zihui Wu, Tianwei Yin, Yu Sun, Robert Frost, Andre van der Kouwe, Adrian V. Dalca, Katherine L. Bouman

IEEE TCI (in press) / Paper / Code / Project Page

2022

Region-of-Interest Adaptive Acquisition for Accelerated MRI

Zihui Wu*, Tianwei Yin*, Adrian V. Dalca, Katherine L. Bouman

Medical Imaging Meets NeurIPS 2022 / Paper

ICCP 2022 Poster, CVPR 2022 CCD Workshop Poster

2021

End-to-End Sequential Sampling and Reconstruction for MR Imaging

Tianwei Yin*, Zihui Wu*, He Sun, Adrian V. Dalca, Yisong Yue, Katherine L. Bouman

ML4H 2021 / Paper / Code / Project Page

Best Paper Award

ICCP 2021 Poster, CVPR 2021 CCD Workshop Poster

2020

Scalable Plug-and-Play ADMM with Convergence Guarantees

Yu Sun, Zihui Wu, Xiaojian Xu, Brendt Wohlberg, Ulugbek S. Kamilov

IEEE TCI / Paper

SIMBA: Scalable Inversion in Optical Tomography using Deep Denoising Priors

Zihui Wu, Yu Sun, Alex Matlock, Jiaming Liu, Lei Tian, U. S. Kamilov

IEEE JSTSP / Paper

2019

Online Regularization by Denoising with Applications to Phase Retrieval

Zihui Wu, Yu Sun, Jiaming Liu, U. S. Kamilov

ICCVW 2019 / Paper

Oral Presentation