Integrating Kalman filter noise residue into U-Net for robust image denoising: the KU-Net model

Abstract

In low-level image processing, where the main goal is to reconstruct a clean image from a noise-corrupted version, image denoising continues to be a critical challenge. Although recent developments have led to the introduction of complex architectures to improve denoising performance, these models frequently have more parameters and higher computational demands. Here, we propose a new, simplified architecture called KU-Net, which is intended to achieve better denoising performance while requiring less complexity. KU-Net is an extension of the basic U-Net architecture that incorporates gradient information and noise residue from a Kalman filter. The network’s ability to learn is improved by this deliberate incorporation, which also helps it better preserve minute details in the denoised images. Without using Image augmentation, the proposed model is trained on a limited dataset to show its resilience in restricted training settings. Three essential inputs are processed by the architecture: gradient estimations, the predicted noisy image, and the original noisy grey image. These inputs work together to steer the U-Net’s encoding and decoding stages to generate high-quality denoised outputs. According to our experimental results, KU-Net performs better than traditional models, as demonstrated by its superiority on common metrics like the Structural Similarity Index (SSIM) and Peak Signal-to-Noise Ratio (PSNR). KU-Net notably attains a PSNR of 26.60 dB at a noise level of 50, highlighting its efficacy and potential for more widespread use in image denoising.