Analyzing the Sample Complexity of Self-Supervised Image Reconstruction Methods
Tobit Klug, Dogukan Atik, Reinhard Heckel
Abstract
Supervised training of deep neural networks on pairs of clean image and noisy measurement achieves state-of-the-art performance for many image reconstruction tasks, but such training pairs are difficult to collect. Self-supervised methods enable training based on noisy measurements only, without clean images. In this work, we investigate the cost of self-supervised training in terms of sample complexity for a class of self-supervised methods that enable the computation of unbiased estimates of gradients of the supervised loss, including noise2noise methods. We analytically show that a model trained with such self-supervised training is as good as the same model trained in a supervised fashion, but self-supervised training requires more examples than supervised training. We then study self-supervised denoising and accelerated MRI empirically and characterize the cost of self-supervised training in terms of the number of additional samples required, and find that the performance gap between self-supervised and supervised training vanishes as a function of the training examples, at a problem-dependent rate, as predicted by our theory.
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 2cdc63b8-4276-428d-80af-b35c042b9b24Cited by top-tier papers2
- MotionTTT: 2D Test-Time-Training Motion Estimation for 3D Motion Corrected MRITobit Klug, Kun Wang, Stefan Ruschke, Reinhard HeckelNeurIPS 2024 · 4 citations
- Language models scale reliably with over-training and on downstream tasksSamir Yitzhak Gadre, Georgios Smyrnis, Vaishaal Shankar, Suchin Gururangan et al.ICLR 2025 · 3 citations
Builds on10
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- Scaling Vision TransformersXiaohua Zhai, Alexander Kolesnikov, Neil Houlsby, Lucas BeyerCVPR 2022 · 767 citations
- Zero-Shot Self-Supervised Learning for MRI ReconstructionBurhaneddin Yaman, Seyed Amir Hossein Hosseini, Mehmet AkçakayaICLR 2022 · 109 citations
- HUMUS-Net: Hybrid Unrolled Multi-scale Network Architecture for Accelerated MRI ReconstructionZalan Fabian, Berk Tinaz, Mahdi SoltanolkotabiNeurIPS 2022 · 83 citations
- Data augmentation for deep learning based accelerated MRI reconstruction with limited dataZalan Fabian, Reinhard Heckel, Mahdi SoltanolkotabiICML 2021 · 60 citations
Related papers
- Self2Self With Dropout: Learning Self-Supervised Denoising From Single ImageYuhui Quan, Mingqin Chen, Tongyao Pang, Hui JiCVPR 2020
- Convexity-Aware Noise Calibration: A Self-Supervised Framework for Noise-Level-Unknown Image DenoisingZhan Wang, Leiquan Wang, Chunlei Wu, Yu MengCVPR 2026
- Zero-Shot Noise2Noise: Efficient Image Denoising without any DataYoussef Mansour, Reinhard HeckelCVPR 2023
- Neighbor2Neighbor: Self-Supervised Denoising From Single Noisy ImagesTao Huang, Songjiang Li, Xu Jia, Huchuan Lu et al.CVPR 2021
- DDM2: Self-Supervised Diffusion MRI Denoising with Generative Diffusion ModelsTiange Xiang, Mahmut Yurt, Ali B. Syed, Kawin Setsompop et al.ICLR 2023
