Adaptive Extensions of Unbiased Risk Estimators for Unsupervised Magnetic Resonance Image Denoising (2407.15799v2)
Abstract: The application of Deep Neural Networks (DNNs) to image denoising has notably challenged traditional denoising methods, particularly within complex noise scenarios prevalent in medical imaging. Despite the effectiveness of traditional and some DNN-based methods, their reliance on high-quality, noiseless ground truth images limits their practical utility. In response to this, our work introduces and benchmarks innovative unsupervised learning strategies, notably Stein's Unbiased Risk Estimator (SURE), its extension (eSURE), and our novel implementation, the Extended Poisson Unbiased Risk Estimator (ePURE), within medical imaging frameworks. This paper presents a comprehensive evaluation of these methods on MRI data afflicted with Gaussian and Poisson noise types, a scenario typical in medical imaging but challenging for most denoising algorithms. Our main contribution lies in the effective adaptation and implementation of the SURE, eSURE, and particularly the ePURE frameworks for medical images, showcasing their robustness and efficacy in environments where traditional noiseless ground truth cannot be obtained.
- José V. Manjón1, Pierrick Coupe, “MRI denoising using Deep Learning and Non-local averaging, arXiv, 2019
- Magauiya Zhussip, Shakarim Soltanayev, Se Young Chun, “Extending Stein’s unbiased risk estimator to train deep denoisers with correlated pairs of noisy images,” in Proceedings of the 33rd Conference on Neural Information Processing Systems (NeurIPS), Vancouver, Canada, 2019.
- Jaakko Lehtinen, Jacob Munkberg, Jon Hasselgren, Samuli Laine, Tero Karras, Miika Aittala, and Timo Aila, “Noise2Noise: Learning image restoration without clean data,” in Proc. of the 35th Int. Conf. on Machine Learning (ICML), pp. 2965–2974, 2018.
- J. Lehtinen, J. Munkberg, J. Hasselgren, S. Laine, T. Karras, M. Aittala, T. Aila, “Noise2Noise: Learning Image Restoration without Clean Data,” in Proc. of the 35th Int. Conf. on Machine Learning, Stockholm, Sweden, 2018.
- Alexander Krull, Tim-Oliver Buchholz, Florian Jug, “Noise2Void - Learning Denoising from Single Noisy Images,” in Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 2019.
- Joshua Batson, Loic Royer, “Noise2Self: Blind Denoising by Self-Supervision,” in Proceedings of the 36th International Conference on Machine Learning, Long Beach, CA, USA, 2019.