Zero-shot Denoising via Neural Compression: Theoretical and algorithmic framework
Ali Zafari, Xi Chen, Shirin Jalali
Abstract
Zero-shot denoising aims to denoise observations without access to training samples or clean reference images. This setting is particularly relevant in practical imaging scenarios involving specialized domains such as medical imaging or biology. In this work, we propose the Zero-Shot Neural Compression Denoiser (ZS-NCD), a novel denoising framework based on neural compression. ZS-NCD treats a neural compression network as an untrained model, optimized directly on patches extracted from a single noisy image. The final reconstruction is then obtained by aggregating the outputs of the trained model over overlapping patches. Thanks to the built-in entropy constraints of compression architectures, our method naturally avoids overfitting and does not require manual regularization or early stopping. Through extensive experiments, we show that ZS-NCD achieves state-of-the-art performance among zero-shot denoisers for both Gaussian and Poisson noise, and generalizes well to both natural and non-natural images. Additionally, we provide new finite-sample theoretical results that characterize upper bounds on the achievable reconstruction error of general maximum-likelihood compression-based denoisers. These results further establish the theoretical foundations of compression-based denoising. Our code is available at: https://github.com/Computational-Imaging-RU/ZS-NCDenoiser .
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 31e36821-8c55-4a14-913b-e524bf387e11Cited by top-tier papers2
- Lottery Prior: Randomized Neural Compression for Zero-Shot Inverse ProblemsHaotian Wu, Di You, Pier Luigi Dragotti, Deniz GunduzICML 2026
- Cross-Domain Lossy Compression via Rate- and Classification-Constrained Optimal TransportNam Nguyen, Thinh Nguyen, Bella BoseICLR 2026
Builds on13
- Restormer: Efficient Transformer for High-Resolution Image RestorationSyed Waqas Zamir, Aditya Arora, Salman Khan, Munawar Hayat et al.CVPR 2022 · 3,348 citations
- Transformer-based Transform CodingYinhao Zhu, Yang Yang, Taco CohenICLR 2022 · 218 citations
- Noise2Score: Tweedie's Approach to Self-Supervised Image Denoising without Clean ImagesKwanyoung Kim, Jong Chul YeNeurIPS 2021 · 176 citations
- VCT: A Video Compression TransformerFabian Mentzer, George Toderici, David Minnen, Sergi Caelles et al.NeurIPS 2022 · 155 citations
- Noise2Same: Optimizing A Self-Supervised Bound for Image DenoisingYaochen Xie, Zhengyang Wang, Shuiwang JiNeurIPS 2020 · 135 citations
Related papers
- Noisier2Noise: Learning to Denoise From Unpaired Noisy DataNick Moran, Dan Schmidt, Yu Zhong, Patrick CoadyCVPR 2020
- Masked Pre-training Enables Universal Zero-shot DenoiserXiaoxiao Ma, Zhixiang Wei, Yi Jin, Pengyang Ling et al.NeurIPS 2024 · 11 citations
- DiffSCI: Zero-Shot Snapshot Compressive Imaging via Iterative Spectral Diffusion ModelZhenghao Pan, Haijin Zeng, Jiezhang Cao, Kai Zhang et al.CVPR 2024 · 8 citations
- An Unsupervised Deep Learning Approach for Real-World Image DenoisingDihan Zheng, Sia Huat Tan, Xiaowen Zhang, Zuoqiang Shi et al.ICLR 2021 · 29 citations
- Invertible generative models for inverse problems: mitigating representation error and dataset biasMuhammad Asim, Max Daniels, Oscar Leong, Ali Ahmed et al.ICML 2020 · 172 citations
