SINET: Sparsity-driven Interpretable Neural Network for Underwater Image Enhancement

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Panda, Gargi, Kundu, Soumitra, Bhattacharya, Saumik, Routray, Aurobinda
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909538079211520
author Panda, Gargi
Kundu, Soumitra
Bhattacharya, Saumik
Routray, Aurobinda
author_facet Panda, Gargi
Kundu, Soumitra
Bhattacharya, Saumik
Routray, Aurobinda
contents Improving the quality of underwater images is essential for advancing marine research and technology. This work introduces a sparsity-driven interpretable neural network (SINET) for the underwater image enhancement (UIE) task. Unlike pure deep learning methods, our network architecture is based on a novel channel-specific convolutional sparse coding (CCSC) model, ensuring good interpretability of the underlying image enhancement process. The key feature of SINET is that it estimates the salient features from the three color channels using three sparse feature estimation blocks (SFEBs). The architecture of SFEB is designed by unrolling an iterative algorithm for solving the $\ell_1$ regularized convolutional sparse coding (CSC) problem. Our experiments show that SINET surpasses state-of-the-art PSNR value by $1.05$ dB with $3873$ times lower computational complexity. Code can be found at: https://github.com/gargi884/SINET-UIE/tree/main.
format Preprint
id arxiv_https___arxiv_org_abs_2409_01022
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SINET: Sparsity-driven Interpretable Neural Network for Underwater Image Enhancement
Panda, Gargi
Kundu, Soumitra
Bhattacharya, Saumik
Routray, Aurobinda
Computer Vision and Pattern Recognition
Image and Video Processing
Improving the quality of underwater images is essential for advancing marine research and technology. This work introduces a sparsity-driven interpretable neural network (SINET) for the underwater image enhancement (UIE) task. Unlike pure deep learning methods, our network architecture is based on a novel channel-specific convolutional sparse coding (CCSC) model, ensuring good interpretability of the underlying image enhancement process. The key feature of SINET is that it estimates the salient features from the three color channels using three sparse feature estimation blocks (SFEBs). The architecture of SFEB is designed by unrolling an iterative algorithm for solving the $\ell_1$ regularized convolutional sparse coding (CSC) problem. Our experiments show that SINET surpasses state-of-the-art PSNR value by $1.05$ dB with $3873$ times lower computational complexity. Code can be found at: https://github.com/gargi884/SINET-UIE/tree/main.
title SINET: Sparsity-driven Interpretable Neural Network for Underwater Image Enhancement
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2409.01022