Schizophrenia recognition based on three-dimensional adaptive graph convolutional neural network.

Journal: Scientific reports
PMID:

Abstract

Previous deep learning-based brain network research has made significant progress in understanding the pathophysiology of schizophrenia. However, it ignores the three-dimensional spatial characteristics of EEG signals and cannot dynamically learn the interactions between nodes. To address this issue, a schizophrenia classification model based on a three-dimensional adaptive graph convolutional neural network (3D-AGCN) is proposed. Each subject's EEG data is divided into various segment lengths and frequency bands for the experiment. The attention mechanism is then used to integrate the node features in the spatial, feature, and frequency band dimensions. The resulting adaptive brain functional network features are then constructed and fed into the GAT + GCN model. This adaptive approach eliminates the human-specified criteria for feature selection and brain network construction. The trial results demonstrated that, when using a 6-second segment length and time-domain and frequency-domain features, patients with first-episode schizophrenia achieved the highest classification accuracy of 87.64% This method outperforms other feature selection and brain network modeling approaches, providing new insights and directions for the early diagnosis and recognition of schizophrenia.

Authors

  • Guimei Yin
    College of Computer Science and Technology, Taiyuan Normal University, No. 319 Daxue Street, Jinzhong, 030619, Shanxi, China. yinguimeicn@126.com.
  • Jie Yuan
    Key Laboratory of Engineering Biology for Low-Carbon Manufacturing, Tianjin Institute of Industrial Biotechnology, Chinese Academy of Sciences, National Center of Technology Innovation for Synthetic Biology, No. 32, Xiqi Road, Tianjin Airport Economic Park, Tianjin 300308, China.
  • Yanjun Chen
    Department of Medical Imaging, The Third Affiliated Hospital of Southern Medical University (Academy of Orthopedics Guangdong Province), Guangzhou, PR China.
  • Guangxing Guo
    Institute of Big Data Technology Analysis and Application, Taiyuan Normal University, Jinzhong, 030619, China.
  • Dongli Shi
    College of Computer Science and Technology, Taiyuan Normal University, No. 319 Daxue Street, Jinzhong, 030619, Shanxi, China.
  • Lin Wang
    Department of Engineering Mechanics, Tsinghua University, Beijing 100084, China.
  • Zilong Zhao
    Faculty of Electrical Engineering, Mathematics and Computer Science, Delft University of Technology, Delft, Netherlands.
  • YanLi Zhao
    Medicinal Plants Research Institute, Yunnan Academy of Agricultural Sciences, Kunming 650200, China.
  • Manjie Zhang
    School of Computer Science and Technology, Taiyuan Normal University, Jinzhong, 030619, China.
  • Yuan Dong
    School of Computer Science and Technology, Taiyuan Normal University, Jinzhong, 030619, China.
  • Bin Wang
    State Key Laboratory of Soil Erosion and Dryland Farming on the Loess Plateau, Northwest A&F University, Yangling 712100, China; New South Wales Department of Primary Industries, Wagga Wagga Agricultural Institute, Wagga Wagga 2650, Australia. Electronic address: bin.a.wang@dpi.nsw.gov.au.
  • Shuping Tan
    Psychiatry Research Center, Peking University Huilonguan Clinical Medical School, Beijing Huilongguan Hospital, Beijing, 100096, China.