A novel graph neural network method for Alzheimer's disease classification.

Journal: Computers in biology and medicine
Published Date:

Abstract

Alzheimer's disease (AD) is a chronic neurodegenerative disease. Early diagnosis are very important to timely treatment and delay the progression of the disease. In the past decade, many computer-aided diagnostic (CAD) algorithms have been proposed for classification of AD. In this paper, we propose a novel graph neural network method, termed Brain Graph Attention Network (BGAN) for classification of AD. First, brain graph data are used to model classification of AD as a graph classification task. Second, a local attention layer is designed to capture and aggregate messages of interactions between node neighbors. And, a global attention layer is introduced to obtain the contribution of each node for graph representation. Finally, using the BGAN to implement AD classification. We train and test on two open public databases for AD classification task. Compared to classic models, the experimental results show that our model is superior to six classic models. We demonstrate that BGAN is a powerful classification model for AD. In addition, our model can provide an analysis of brain regions in order to judge which regions are related to AD disease and which regions are related to AD progression.

Authors

  • Zhiheng Zhou
    Academy of Mathematics and Systems Science, Chinese Academy of Sciences, Beijing, China; School of Mathematical Sciences, University of Chinese Academy of Sciences, Beijing, China.
  • Qi Wang
    Biotherapeutics Discovery Research Center, Shanghai Institute of Materia Medica, Chinese Academy of Sciences, Shanghai, 201203, China.
  • Xiaoyu An
    Research Center for Mathematics and Interdisciplinary Sciences, Shandong University, Qingdao, China.
  • Siwei Chen
    School of Pharmacy, Southwest Medical University, Luzhou 646000, China.
  • Yongan Sun
    Department of Neurology, Peking University First Hospital, Beijing, China.
  • Guanghui Wang
    School of Engineering, University of Kansas, Lawrence, KS, United States of America.
  • Guiying Yan
    Academy of Mathematics and Systems Science, Chinese Academy of Sciences, Beijing, China; School of Mathematical Sciences, University of Chinese Academy of Sciences, Beijing, China. Electronic address: yangy@amss.ac.cn.