Learning Medical Image Denoising with Deep Dynamic Residual Attention Network

This is the official implementation of a state-of-the-art medical image denoising method titled as "Dynamic Residual Attention Network (DRAN)". [Click Here] to download the full paper (in PDF).

Please consider citing this paper as follows:

@article{sharif2020learning,
  title={Learning Medical Image Denoising with Deep Dynamic Residual Attention Network},
  author={Sharif, SMA and Naqvi, Rizwan Ali and Biswas, Mithun},
  journal={Mathematics},
  volume={8},
  number={12},
  pages={2192},
  year={2020},
  publisher={Multidisciplinary Digital Publishing Institute}
}

Explore our work on medical image deblurring [Here].

Consider citing our deblurring paper as follows:

@article{sharif2022meddeblur,
  title={MedDeblur: Medical Image Deblurring with Residual Dense Spatial-Asymmetric Attention},
  author={Sharif, SMA and Naqvi, Rizwan Ali and Mehmood, Zahid and Hussain, Jamil and Ali, Ahsan and Lee, Seung-Won},
  journal={Mathematics},
  volume={11},
  number={1},
  pages={115},
  year={2022},
  publisher={MDPI}
}

Overview

Figure: Overview of proposed medical image denoising method.

Network Architecture

Figure: The overview of proposed network architecture. The proposed network incorporates novel dynamic residual attention blocks, which utilizes dynamic convolution and a noise gate. Also, the network leverage the residual learning along with the learning feature correlation.

Medical Image Denoising with DRAN

Qualitative Comparison

Figure: Performance of existing medical image denoising methods in removing image noise at sigma = 50. The existing denoising methods immensely failed in addressing a substantial amount of noise removal and susceptible to produce artefacts. (a) Noisy input. (b) Result obtained by BM3D. (c) Result obtained by DnCNN. (d) Result obtained by Residual MID. (e) Result obtained by DRAN. (f) Reference Image.

Quantitative Comparison

Table: Quantitative comparison between different medical image denoising methods. Results are obtained by calculating the mean on two evaluation metrics (PSNR and SSIM). In all comparing categories, the proposed method illustrates the consistency and outperforms the existing denoising methods.

Prerequisites

Python 3.8
CUDA 10.1 + CuDNN
pip
Virtual environment (optional)

Installation

Please consider using a virtual environment to continue the installation process.

git clone https://github.com/sharif-apu/MID-DRAN.git
cd MID-DRAN
pip install -r requirement.txt

Testing

DRAN can be inferenced with pretrained weights and default setting as follows:
python main.py -i

A few testing images are provided in a sub-directory under testingImages (i.e., testingImages/sampleImages/)
In such occasion, denoised image(s) will be available in modelOutput/sampleImages/.

To inference with custom setting execute the following command:
python main.py -i -s path/to/inputImages -d path/to/outputImages -ns=sigma(s)
Here,-ns specifies the standard deviation of a Gaussian distribution (i.e., -ns=15, 25, 50),-s specifies the root directory of the source images (i.e., testingImages/), and -d specifies the destination root (i.e., modelOutput/).

Training

To train with your own dataset execute:
python main.py -ts -e X -b Y To specify your trining images path, go to mainModule/config.json and update "trainingImagePath" entity.
You can specify the number of epoch with -e flag (i.e., -e 5) and number of images per batch with -b flag (i.e., -b 24).

For transfer learning execute:
python main.py -tr -e -b

Others

Check model configuration:
python main.py -ms
Create new configuration file:
python main.py -c
Update configuration file:
python main.py -u
Overfitting testing
python main.py -to

Contact

For any further query, feel free to contact us through the following emails: [email protected], [email protected], or [email protected]