Dermoscopic image segmentation based on Pyramid Residual Attention Module

PLoS One. 2022 Sep 16;17(9):e0267380. doi: 10.1371/journal.pone.0267380. eCollection 2022.

Abstract

We propose a stacked convolutional neural network incorporating a novel and efficient pyramid residual attention (PRA) module for the task of automatic segmentation of dermoscopic images. Precise segmentation is a significant and challenging step for computer-aided diagnosis technology in skin lesion diagnosis and treatment. The proposed PRA has the following characteristics: First, we concentrate on three widely used modules in the PRA. The purpose of the pyramid structure is to extract the feature information of the lesion area at different scales, the residual means is aimed to ensure the efficiency of model training, and the attention mechanism is used to screen effective features maps. Thanks to the PRA, our network can still obtain precise boundary information that distinguishes healthy skin from diseased areas for the blurred lesion areas. Secondly, the proposed PRA can increase the segmentation ability of a single module for lesion regions through efficient stacking. The third, we incorporate the idea of encoder-decoder into the architecture of the overall network. Compared with the traditional networks, we divide the segmentation procedure into three levels and construct the pyramid residual attention network (PRAN). The shallow layer mainly processes spatial information, the middle layer refines both spatial and semantic information, and the deep layer intensively learns semantic information. The basic module of PRAN is PRA, which is enough to ensure the efficiency of the three-layer architecture network. We extensively evaluate our method on ISIC2017 and ISIC2018 datasets. The experimental results demonstrate that PRAN can obtain better segmentation performance comparable to state-of-the-art deep learning models under the same experiment environment conditions.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Diagnosis, Computer-Assisted
  • Disease Progression
  • Humans
  • Neural Networks, Computer*
  • Pyramidal Tracts
  • Skin Diseases*

Grants and funding

This work was supported by National Natural Science Foundation of China in the form of a grant (61962054) to YJ. This work was also supported by The Cultivation Plan of Major Scientific Research Projects of Northwest Normal in the form of a grant (NWNU-LKZD2021-06) to YJ. The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.