MHANet: A hybrid attention mechanism for retinal diseases classification

Xu, Lianghui and Wang, Liejun and Cheng, Shuli and Li, Yongming and Zhang, Jie (2021) MHANet: A hybrid attention mechanism for retinal diseases classification. PLOS ONE, 16 (12). e0261285. ISSN 1932-6203

[thumbnail of journal.pone.0261285.pdf] Text
journal.pone.0261285.pdf - Published Version

Download (3MB)

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.

Item Type: Article
Subjects: STM Repository > Social Sciences and Humanities
Depositing User: Managing Editor
Date Deposited: 08 Feb 2023 06:29
Last Modified: 06 Jul 2024 06:33
URI: http://classical.goforpromo.com/id/eprint/2302

Actions (login required)

View Item
View Item