A diagnostic unified classification model for classifying multi-sized and multi-modal brain graphs using graph alignment

• Published in: Journal of neuroscience methods 2021-01, Vol.348, p.109014-109014, Article 109014
• Main Authors: Yalçin, Abdullah, Rekik, Islem
• Format: Article
• Language: English
• Subjects: Graph node embedding; Linear discriminant analysis; Multi-modal brain connectome classification; Multi-sized brain graph alignment; Neurological disorder diagnosis
• ISSN: 0165-0270; 1872-678X
• DOI: 10.1016/j.jneumeth.2020.109014

[Display omitted] •We propose a unified brain graph classification model that can classify multi-modal multisized brain graphs.•We design a graph alignment strategy to a fixed graph template.•Our unified classification model is diagnostic while being agnostic to the connectomic data source and size.•Our proposed method is scalable and generalizable to different diagnostic clinical frameworks. Presence of multimodal brain graphs derived from different neuroimaging modalities is inarguably one of the most critical challenges in building unified classification models that can be trained and tested on any brain graph regardless of its size and the modality it was derived from. One solution is to learn a model for each modality independently, which is cumbersome and becomes more time-consuming as the number of modalities increases. Another traditional solution is to build a model inputting multimodal brain graphs for the target prediction task; however, this is only applicable to datasets where all samples have joint neuro-modalities. In this paper, we propose to build a unified brain graph classification model trained on unpaired multimodal brain graphs, which can classify any brain graph of any size. This is enabled by incorporating a graph alignment step where all multi-modal graphs of different sizes and heterogeneous distributions are mapped to a common template graph. Next, we design a graph alignment strategy to the target fixed-size template and further apply linear discriminant analysis (LDA) to the aligned graphs as a supervised dimensionality reduction technique for the target classification task. We tested our method on unpaired autistic and healthy brain connectomes derived from functional and morphological MRI datasets (two modalities). Our results showed that our unified model method not only has great promise in solving such a challenging problem but achieves comparable performance to models trained on each modality independently.