Classification of Alzheimer’s Disease Based on Weakly Supervised Learning and Attention Mechanism

Xiaosheng Wu, Shuangshuang Gao, Junding Sun, Yudong Zhang*, Shuihua Wang*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

5 Citations (Scopus)

Abstract

The brain lesions images of Alzheimer’s disease (AD) patients are slightly different from the Magnetic Resonance Imaging of normal people, and the classification effect of general image recognition technology is not ideal. Alzheimer’s datasets are small, making it difficult to train large-scale neural networks. In this paper, we propose a network model (WS-AMN) that fuses weak supervision and an attention mechanism. The weakly supervised data augmentation network is used as the basic model, the attention map generated by weakly supervised learning is used to guide the data augmentation, and an attention module with channel domain and spatial domain is embedded in the residual network to focus on the distinctive channels and spaces of images respectively. The location information enhances the corresponding features of related features and suppresses the influence of irrelevant features.The results show that the F1-score is 99.63%, the accuracy is 99.61%. Our model provides a high-performance solution for accurate classification of AD.

Original languageEnglish
Article number1601
JournalBrain Sciences
Volume12
Issue number12
DOIs
Publication statusPublished - Dec 2022
Externally publishedYes

Keywords

  • attention module
  • classification
  • data augmentation
  • weakly supervised

Fingerprint

Dive into the research topics of 'Classification of Alzheimer’s Disease Based on Weakly Supervised Learning and Attention Mechanism'. Together they form a unique fingerprint.

Cite this