Self2Self+: Single-Image Denoising with Self-Supervised Learning and Image Quality Assessment Loss

Jaekyun Ko, Sanghwan Lee
7/20/2023
cs.CVcs.LGeess.IV

Official code: JK-the-Ko/Self2SelfPlus (20 stars, BSD-3-Clause, last commit 2023-09-07).

Abstract

Recently, denoising methods based on supervised learning have exhibited promising performance. However, their reliance on external datasets containing noisy-clean image pairs restricts their applicability. To address this limitation, researchers have focused on training denoising networks using solely a set of noisy inputs. To improve the feasibility of denoising procedures, in this study, we proposed a single-image self-supervised learning method in which only the noisy input image is used for network training. Gated convolution was used for feature extraction and no-reference image quality assessment was used for guiding the training process. Moreover, the proposed method sampled instances from the input image dataset using Bernoulli sampling with a certain dropout rate for training. The corresponding result was produced by averaging the generated predictions from various instances of the trained network with dropouts. The experimental results indicated that the proposed method achieved state-of-the-art denoising performance on both synthetic and real-world datasets. This highlights the effectiveness and practicality of our method as a potential solution for various noise removal tasks.

View on arXivView PDF

Code Implementations(1)

BSD-3-Clause

[arXiv Preprint] Official PyTorch Implementation of "Self2Self+: Single-Image Denoising with Self-Supervised Learning and Image Quality Assessment Loss"

201Shell, PythonMay 8, 20233 years agoBSD-3-Clause
computer-visionconvolutional-neural-networksdeep-learningimage-denoisingself-supervised-learning

Similar Papers

Cite this paper

@article{ko2023selfself,
  title  = {Self2Self+: Single-Image Denoising with Self-Supervised Learning and Image Quality Assessment Loss},
  author = {Jaekyun Ko and Sanghwan Lee},
  year   = {2023},
  eprint = {2307.10695},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2307.10695v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2307.10695.svg)](https://codeofpaper.com/paper/2307.10695)

Discussion