Adversarial Distortion Learning for Medical Image Denoising

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ghahremani, Morteza, Khateri, Mohammad, Sierra, Alejandra, Tohka, Jussi
Formato: Preprint
Publicado: 2022
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917610872897536
author Ghahremani, Morteza
Khateri, Mohammad
Sierra, Alejandra
Tohka, Jussi
author_facet Ghahremani, Morteza
Khateri, Mohammad
Sierra, Alejandra
Tohka, Jussi
contents We present a novel adversarial distortion learning (ADL) for denoising two- and three-dimensional (2D/3D) biomedical image data. The proposed ADL consists of two auto-encoders: a denoiser and a discriminator. The denoiser removes noise from input data and the discriminator compares the denoised result to its noise-free counterpart. This process is repeated until the discriminator cannot differentiate the denoised data from the reference. Both the denoiser and the discriminator are built upon a proposed auto-encoder called Efficient-Unet. Efficient-Unet has a light architecture that uses the residual blocks and a novel pyramidal approach in the backbone to efficiently extract and re-use feature maps. During training, the textural information and contrast are controlled by two novel loss functions. The architecture of Efficient-Unet allows generalizing the proposed method to any sort of biomedical data. The 2D version of our network was trained on ImageNet and tested on biomedical datasets whose distribution is completely different from ImageNet; so, there is no need for re-training. Experimental results carried out on magnetic resonance imaging (MRI), dermatoscopy, electron microscopy and X-ray datasets show that the proposed method achieved the best on each benchmark. Our implementation and pre-trained models are available at https://github.com/mogvision/ADL.
format Preprint
id arxiv_https___arxiv_org_abs_2204_14100
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Adversarial Distortion Learning for Medical Image Denoising
Ghahremani, Morteza
Khateri, Mohammad
Sierra, Alejandra
Tohka, Jussi
Image and Video Processing
Computer Vision and Pattern Recognition
We present a novel adversarial distortion learning (ADL) for denoising two- and three-dimensional (2D/3D) biomedical image data. The proposed ADL consists of two auto-encoders: a denoiser and a discriminator. The denoiser removes noise from input data and the discriminator compares the denoised result to its noise-free counterpart. This process is repeated until the discriminator cannot differentiate the denoised data from the reference. Both the denoiser and the discriminator are built upon a proposed auto-encoder called Efficient-Unet. Efficient-Unet has a light architecture that uses the residual blocks and a novel pyramidal approach in the backbone to efficiently extract and re-use feature maps. During training, the textural information and contrast are controlled by two novel loss functions. The architecture of Efficient-Unet allows generalizing the proposed method to any sort of biomedical data. The 2D version of our network was trained on ImageNet and tested on biomedical datasets whose distribution is completely different from ImageNet; so, there is no need for re-training. Experimental results carried out on magnetic resonance imaging (MRI), dermatoscopy, electron microscopy and X-ray datasets show that the proposed method achieved the best on each benchmark. Our implementation and pre-trained models are available at https://github.com/mogvision/ADL.
title Adversarial Distortion Learning for Medical Image Denoising
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2204.14100