Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Preventing Overfitting in Deep Image Prior for Hyperspectral Image Denoising

About

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.

Panagiotis Gkotsis, Athanasios A. Rontogiannis• 2026

Related benchmarks

TaskDatasetResultRank
Hyperspectral Image DenoisingWashington DC Mall HSI Gaussian + Stripe noise
MPSNR34.28
24
HSI DenoisingSalinas HSI
MPSNR32.57
24
HSI Denoising (Gaussian + Sparse Noise)Salinas HSI
MPSNR31.02
12
Hyperspectral Image DenoisingWashington DC Mall HSI Gaussian + Sparse Noise
MPSNR34.98
12
Hyperspectral Image DenoisingSalinas HSI Gaussian + Stripe noise
MPSNR31.51
12
Hyperspectral Image DenoisingWashington DC Mall HSI Gaussian Noise
MPSNR35.7
12
Showing 6 of 6 rows

Other info

Follow for update