Back to Search Start Over

MHANet: A hybrid attention mechanism for retinal diseases classification

Authors :
Lianghui Xu
Liejun Wang
Shuli Cheng
Yongming Li
Source :
PLoS ONE, Vol 16, Iss 12, p e0261285 (2021), PLoS ONE
Publication Year :
2021
Publisher :
Public Library of Science (PLoS), 2021.

Abstract

With the increase of patients with retinopathy, retinopathy recognition has become a research hotspot. In this article, we describe the etiology and symptoms of three kinds of retinal diseases, including drusen(DRUSEN), choroidal neovascularization(CNV) and diabetic macular edema(DME). In addition, we also propose a hybrid attention mechanism to classify and recognize different types of retinopathy images. In particular, the hybrid attention mechanism proposed in this paper includes parallel spatial attention mechanism and channel attention mechanism. It can extract the key features in the channel dimension and spatial dimension of retinopathy images, and reduce the negative impact of background information on classification results. The experimental results show that the hybrid attention mechanism proposed in this paper can better assist the network to focus on extracting thr fetures of the retinopathy area and enhance the adaptability to the differences of different data sets. Finally, the hybrid attention mechanism achieved 96.5% and 99.76% classification accuracy on two public OCT data sets of retinopathy, respectively.

Details

ISSN :
19326203
Volume :
16
Database :
OpenAIRE
Journal :
PLOS ONE
Accession number :
edsair.doi.dedup.....a61703cd382bb162a7d57978561cf1fe