HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

SwinIA: Self-Supervised Blind-Spot Image Denoising without Convolutions

Papkov Mikhail ; Chizhov Pavel ; Parts Leopold

SwinIA: Self-Supervised Blind-Spot Image Denoising without Convolutions

Abstract

Self-supervised image denoising implies restoring the signal from a noisyimage without access to the ground truth. State-of-the-art solutions for thistask rely on predicting masked pixels with a fully-convolutional neuralnetwork. This most often requires multiple forward passes, information aboutthe noise model, or intricate regularization functions. In this paper, wepropose a Swin Transformer-based Image Autoencoder (SwinIA), the firstfully-transformer architecture for self-supervised denoising. The flexibilityof the attention mechanism helps to fulfill the blind-spot property thatconvolutional counterparts normally approximate. SwinIA can be trainedend-to-end with a simple mean squared error loss without masking and does notrequire any prior knowledge about clean data or noise distribution. Simple touse, SwinIA establishes the state of the art on several common benchmarks.

Benchmarks

BenchmarkMethodologyMetrics
color-image-denoising-on-bsd300-lambda30SwinIA
PSNR: 27.92
SSIM: 0.775
color-image-denoising-on-bsd300-lambda5-50SwinIA
PSNR: 27.74
SSIM: 0.764
color-image-denoising-on-bsd300-sigma25SwinIA
PSNR: 28.4
SSIM: 0.789
color-image-denoising-on-bsd300-sigma5-50SwinIA
PSNR: 28.4
SSIM: 0.785
color-image-denoising-on-kodak24-lambda30SwinIA
PSNR: 29.51
SSIM: 0.805
color-image-denoising-on-kodak24-lambda5-50SwinIA
PSNR: 29.06
SSIM: 0.788
color-image-denoising-on-kodak24-sigma25SwinIA
PSNR: 30.12
SSIM: 0.819
color-image-denoising-on-kodak24-sigma5-50SwinIA
PSNR: 30.3
SSIM: 0.820
color-image-denoising-on-set14-lambda30SwinIA
PSNR: 28.74
SSIM: 0.799
color-image-denoising-on-set14-lambda5-50SwinIA
PSNR: 28.27
SSIM: 0.780
color-image-denoising-on-set14-sigma25SwinIA
PSNR: 29.54
SSIM: 0.814
color-image-denoising-on-set14-sigma5-50SwinIA
PSNR: 29.49
SSIM: 0.809
grayscale-image-denoising-on-bsd68-sigma15SwinIA
PSNR: 31.07
SSIM: 0.856
grayscale-image-denoising-on-bsd68-sigma25SwinIA
PSNR: 29.17
SSIM: 0.801
grayscale-image-denoising-on-bsd68-sigma50SwinIA
PSNR: 26.61
SSIM: 0.706
grayscale-image-denoising-on-hanziSwinIA
PSNR: 14.35
SSIM: 0.556
grayscale-image-denoising-on-set12-sigma15SwinIA
PSNR: 30.37
SSIM: 0.857
grayscale-image-denoising-on-set12-sigma25SwinIA
PSNR: 28.72
SSIM: 0.817
grayscale-image-denoising-on-set12-sigma50SwinIA
PSNR: 26.03
SSIM: 0.736
medical-image-denoising-on-fmd-confocal-fishSwinIA
PSNR: 31.79
SSIM: 0.871
medical-image-denoising-on-fmd-confocal-miceSwinIA
PSNR: 37.65
SSIM: 0.960
medical-image-denoising-on-fmd-two-photonSwinIA
PSNR: 33.25
SSIM: 0.915

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp