Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper addresses overfitting in Deep Image Prior (DIP) for hyperspectral image (HSI) denoising by introducing a combined approach of robust data fidelity using a Smooth $\ell_1$ loss and explicit sensitivity regularization via divergence-based regularization and input optimization. This mitigates the need for early stopping, a common issue in DIP. Experiments on real HSIs with various noise types demonstrate that the proposed method outperforms existing DIP-based HSI denoising techniques by effectively preventing overfitting.
Stop early stopping: a new regularization strategy lets Deep Image Prior models train longer and achieve state-of-the-art denoising on hyperspectral images.
Deep image prior (DIP) is an unsupervised deep learning framework that has been successfully applied to a variety of inverse imaging problems. However, DIP-based methods are inherently prone to overfitting, which leads to performance degradation and necessitates early stopping. In this paper, we propose a method to mitigate overfitting in DIP-based hyperspectral image (HSI) denoising by jointly combining robust data fidelity and explicit sensitivity regularization. The proposed approach employs a Smooth $\ell_1$ data term together with a divergence-based regularization and input optimization during training. Experimental results on real HSIs corrupted by Gaussian, sparse, and stripe noise demonstrate that the proposed method effectively prevents overfitting and achieves superior denoising performance compared to state-of-the-art DIP-based HSI denoising methods.