Uncertainty-Propelled Physics-MAE Fusion for Self-Supervised Diffusion-Weighted Image Denoising
Zeyu Deng, Lihui Wang, Xi Tao, Qijian Chen, Ying Cao, Xulin Hu, Yingfeng Ou
Abstract
The inherently low signal-to-noise ratio (SNR) in diffusionweighted (DW) imaging fundamentally impedes precise tissue microstructure characterization, rendering effective noise suppression a persistent challenge. Existing denoising methods frequently suffer from over-smoothing or distortion of microstructure information when handling spatially correlated or severe noise. To address these limitations, we propose UP 2 -MAE fusion model, a self-supervised DWI denoising method based on Uncertainty-Propelled Physics and Masked Auto-Encoder (MAE) fusion. This framework integrates two complementary branches: one leverages MAE to suppress noise through robust feature learning, while the other constructs uncorrelated noisy pairs using diffusion tensor imaging (DTI) physics and denoises them via a Noise2Noise approach, which can preserve texture details by exploiting angular relationships between DW images along diffusion encoding directions. To fully integrate the strengths of both branches, an uncertainty-propelled fusion strategy based on maximum likelihood estimation is proposed to derive the final denoised output. In addition, to further promote the performance, uncertainty-guided reconstruction and consistency loss are presented. Evaluations against state-of-the-art denoising methods on both simulated and acquired DW datasets confirm the efficacy of our approach.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 7abce62e-02b2-4a2c-b355-6fa4fbed3dbbBuilds on9
- Blind2Unblind: Self-Supervised Image Denoising with Visible Blind SpotsZejin Wang, Jiazheng Liu, Guoqing Li, Hua HanCVPR 2022 · 174 citations
- Patch2Self: Denoising Diffusion MRI with Self-Supervised LearningShreyas Fadnavis, Joshua Batson, Eleftherios GaryfallidisNeurIPS 2020 · 151 citations
- AP-BSN: Self-Supervised Denoising for Real-World Images via Asymmetric PD and Blind-Spot NetworkWooseok Lee, Sanghyun Son, Kyoung Mu LeeCVPR 2022 · 148 citations
- PUCA: Patch-Unshuffle and Channel Attention for Enhanced Self-Supervised Image DenoisingHyemi Jang, Junsung Park, Dahuin Jung, Jaihyun Lew et al.NeurIPS 2023 · 42 citations
- Neighbor2Neighbor: Self-Supervised Denoising From Single Noisy ImagesTao Huang, Songjiang Li, Xu Jia, Huchuan Lu et al.CVPR 2021
Related papers
- Self-Supervised Diffusion MRI Denoising via Iterative and Stable RefinementChenxu Wu, Qingpeng Kong, Zihang Jiang, S. Kevin ZhouICLR 2025
- DDM2: Self-Supervised Diffusion MRI Denoising with Generative Diffusion ModelsTiange Xiang, Mahmut Yurt, Ali B. Syed, Kawin Setsompop et al.ICLR 2023
- Masked-Diffusion Autoencoders for 3D Medical Vision Representation LearningJiachen Tu, Guanghui Qin, Theodore Zhengde Zhao, Jeya Maria Jose Valanarasu et al.CVPR 2026
- Multi-view Self-supervised Disentanglement for General Image DenoisingHao Chen, Chenyuan Qu, Yu Zhang, Chen Chen et al.ICCV 2023 · 15 citations
- Score-based Self-supervised MRI DenoisingJiachen Tu, Yaokun Shi, Fan LamICLR 2025
