A Learnable Counter-condition Analysis Framework for Functional Connectivity-based Neurological Disorder Diagnosis (2310.03964v1)
Abstract: To understand the biological characteristics of neurological disorders with functional connectivity (FC), recent studies have widely utilized deep learning-based models to identify the disease and conducted post-hoc analyses via explainable models to discover disease-related biomarkers. Most existing frameworks consist of three stages, namely, feature selection, feature extraction for classification, and analysis, where each stage is implemented separately. However, if the results at each stage lack reliability, it can cause misdiagnosis and incorrect analysis in afterward stages. In this study, we propose a novel unified framework that systemically integrates diagnoses (i.e., feature selection and feature extraction) and explanations. Notably, we devised an adaptive attention network as a feature selection approach to identify individual-specific disease-related connections. We also propose a functional network relational encoder that summarizes the global topological properties of FC by learning the inter-network relations without pre-defined edges between functional networks. Last but not least, our framework provides a novel explanatory power for neuroscientific interpretation, also termed counter-condition analysis. We simulated the FC that reverses the diagnostic information (i.e., counter-condition FC): converting a normal brain to be abnormal and vice versa. We validated the effectiveness of our framework by using two large resting-state functional magnetic resonance imaging (fMRI) datasets, Autism Brain Imaging Data Exchange (ABIDE) and REST-meta-MDD, and demonstrated that our framework outperforms other competing methods for disease identification. Furthermore, we analyzed the disease-related neurological patterns based on counter-condition analysis.
- T.-E. Kam, H. Zhang, Z. Jiao, and D. Shen, “Deep learning of static and dynamic brain functional networks for early MCI detection,” IEEE Transactions on Medical Imaging, vol. 39, no. 2, pp. 478–487, 2019.
- Y. Li, J. Liu, Y. Jiang, Y. Liu, and B. Lei, “Virtual adversarial training-based deep feature aggregation network from dynamic effective connectivity for mci identification,” IEEE Transactions on Medical Imaging, vol. 41, no. 1, pp. 237–251, 2021.
- S.-J. Hong, R. Vos de Wael, R. A. Bethlehem, S. Lariviere, C. Paquola, S. L. Valk, M. P. Milham, A. Di Martino, D. S. Margulies, J. Smallwood et al., “Atypical functional connectome hierarchy in autism,” Nature Communications, vol. 10, no. 1, pp. 1–13, 2019.
- J. Ciarrusta, R. Dimitrova, D. Batalle, J. O’Muircheartaigh, L. Cordero-Grande, A. Price, E. Hughes, J. Kangas, E. Perry, A. Javed et al., “Emerging functional connectivity differences in newborn infants vulnerable to autism spectrum disorders,” Translational Psychiatry, vol. 10, no. 1, pp. 1–10, 2020.
- A. Nair, M. Jolliffe, Y. S. S. Lograsso, and C. E. Bearden, “A review of default mode network connectivity and its association with social cognition in adolescents with autism spectrum disorder and early-onset psychosis,” Frontiers in Psychiatry, vol. 11, p. 614, 2020.
- C.-M. Chen, P. Yang, M.-T. Wu, T.-C. Chuang, and T.-Y. Huang, “Deriving and validating biomarkers associated with autism spectrum disorders from a large-scale resting-state database,” Scientific Reports, vol. 9, no. 1, pp. 1–10, 2019.
- X. Xu, W. Li, J. Mei, M. Tao, X. Wang, Q. Zhao, X. Liang, W. Wu, D. Ding, and P. Wang, “Feature selection and combination of information in the functional brain connectome for discrimination of mild cognitive impairment and analyses of altered brain patterns,” Frontiers in Aging Neuroscience, vol. 12, p. 28, 2020.
- H. S. Nogay and H. Adeli, “Machine learning (ML) for the diagnosis of autism spectrum disorder (ASD) using brain imaging,” Reviews in the Neurosciences, vol. 31, no. 8, pp. 825–841, 2020.
- P. Zille, V. D. Calhoun, J. M. Stephen, T. W. Wilson, and Y.-P. Wang, “Fused estimation of sparse connectivity patterns from rest fMRI—application to comparison of children and adult brains,” IEEE Transactions on Medical Imaging, vol. 37, no. 10, pp. 2165–2175, 2017.
- Y. Li, J. Liu, X. Gao, B. Jie, M. Kim, P.-T. Yap, C.-Y. Wee, and D. Shen, “Multimodal hyper-connectivity of functional networks using functionally-weighted LASSO for MCI classification,” Medical Image Analysis, vol. 52, pp. 80–96, 2019.
- Y. Li, J. Liu, Z. Tang, and B. Lei, “Deep spatial-temporal feature fusion from adaptive dynamic functional connectivity for MCI identification,” IEEE Transactions on Medical Imaging, vol. 39, no. 9, pp. 2818–2830, 2020.
- N. Pudjihartono, T. Fadason, A. W. Kempa-Liehr, and J. M. O’Sullivan, “A review of feature selection methods for machine learning-based disease risk prediction,” Frontiers in Bioinformatics, vol. 2, p. 927312, 2022.
- C. Wang, Z. Xiao, and J. Wu, “Functional connectivity-based classification of autism and control using SVM-RFECV on rs-fMRI data,” Physica Medica, vol. 65, pp. 99–105, 2019.
- P. Dai, T. Xiong, X. Zhou, Y. Ou, Y. Li, X. Kui, Z. Chen, B. Zou, W. Li, Z. Huang et al., “The alterations of brain functional connectivity networks in major depressive disorder detected by machine learning through multisite rs-fMRI data,” Behavioural Brain Research, vol. 435, p. 114058, 2022.
- H. Wen, J. Shi, Y. Zhang, K.-H. Lu, J. Cao, and Z. Liu, “Neural encoding and decoding with deep learning for dynamic natural vision,” Cerebral Cortex, vol. 28, no. 12, pp. 4136–4160, 2018.
- G. Qu, L. Xiao, W. Hu, J. Wang, K. Zhang, V. D. Calhoun, and Y.-P. Wang, “Ensemble manifold regularized multi-modal graph convolutional network for cognitive ability prediction,” IEEE Transactions on Biomedical Engineering, vol. 68, no. 12, pp. 3564–3573, 2021.
- S. Mueller, D. Wang, M. D. Fox, B. T. Yeo, J. Sepulcre, M. R. Sabuncu, R. Shafee, J. Lu, and H. Liu, “Individual variability in functional connectivity architecture of the human brain,” Neuron, vol. 77, no. 3, pp. 586–595, 2013.
- E. Feczko, O. Miranda-Dominguez, M. Marr, A. M. Graham, J. T. Nigg, and D. A. Fair, “The heterogeneity problem: approaches to identify psychiatric subtypes,” Trends in cognitive sciences, vol. 23, no. 7, pp. 584–601, 2019.
- J. Lee, W. Ko, E. Kang, H.-I. Suk, A. D. N. Initiative et al., “A unified framework for personalized regions selection and functional relation modeling for early MCI identification,” NeuroImage, vol. 236, p. 118048, 2021.
- X. Li, Y. Zhou, N. Dvornek, M. Zhang, S. Gao, J. Zhuang, D. Scheinost, L. H. Staib, P. Ventola, and J. S. Duncan, “BrainGNN: Interpretable brain graph neural network for fMRI analysis,” Medical Image Analysis, vol. 74, p. 102233, 2021.
- M. P. van den Heuvel and O. Sporns, “A cross-disorder connectome landscape of brain dysconnectivity,” Nature Reviews Neuroscience, vol. 20, no. 7, pp. 435–446, 2019.
- J. Kawahara, C. J. Brown, S. P. Miller, B. G. Booth, V. Chau, R. E. Grunau, J. G. Zwicker, and G. Hamarneh, “BrainNetCNN: Convolutional neural networks for brain networks; towards predicting neurodevelopment,” NeuroImage, vol. 146, pp. 1038–1049, 2017.
- D. Zhi, V. D. Calhoun, C. Wang, X. Li, X. Ma, L. Lv, W. Yan, D. Yao, S. Qi, R. Jiang et al., “BNCPL: Brain-network-based convolutional prototype learning for discriminating depressive disorders,” in 2021 43rd Annual International Conference of the IEEE Engineering in Medicine & Biology Society, 2021, pp. 1622–1626.
- P.-J. Kindermans, S. Hooker, J. Adebayo, M. Alber, K. T. Schütt, S. Dähne, D. Erhan, and B. Kim, “The (un) reliability of saliency methods,” in Explainable AI: Interpreting, Explaining and Visualizing Deep Learning. Springer, 2019, pp. 267–280.
- N. Arun, N. Gaw, P. Singh, K. Chang, M. Aggarwal, B. Chen, K. Hoebel, S. Gupta, J. Patel, M. Gidwani et al., “Assessing the trustworthiness of saliency maps for localizing abnormalities in medical imaging,” Radiology: Artificial Intelligence, vol. 3, no. 6, 2021.
- J. Adebayo, J. Gilmer, M. Muelly, I. Goodfellow, M. Hardt, and B. Kim, “Sanity checks for saliency maps,” Advances in Neural Information Processing Systems, vol. 31, 2018.
- H. M. Van Loo, P. De Jonge, J.-W. Romeijn, R. C. Kessler, and R. A. Schoevers, “Data-driven subtypes of major depressive disorder: a systematic review,” BMC medicine, vol. 10, pp. 1–12, 2012.
- W. K. Lau, M.-K. Leung, and B. W. Lau, “Resting-state abnormalities in autism spectrum disorders: a meta-analysis,” Scientific Reports, vol. 9, no. 1, pp. 1–8, 2019.
- E. Kang, D.-W. Heo, and H.-I. Suk, “Prototype learning of inter-network connectivity for ASD diagnosis and personalized analysis,” in International Conference on Medical Image Computing and Computer-Assisted Intervention. Springer, 2022, pp. 334–343.
- X. Chen, B. Lu, H.-X. Li, X.-Y. Li, Y.-W. Wang, F. X. Castellanos, L.-P. Cao, N.-X. Chen, W. Chen, Y.-Q. Cheng et al., “The DIRECT consortium and the REST-meta-MDD project: towards neuroimaging biomarkers of major depressive disorder,” Psychoradiology, vol. 2, no. 1, pp. 32–42, 2022.
- C.-G. Yan, X. Chen, L. Li, F. X. Castellanos, T.-J. Bai, Q.-J. Bo, J. Cao, G.-M. Chen, N.-X. Chen, W. Chen et al., “Reduced default mode network functional connectivity in patients with recurrent major depressive disorder,” Proceedings of the National Academy of Sciences, vol. 116, no. 18, pp. 9078–9083, 2019.
- J.-P. Fortin, D. Parker, B. Tunç, T. Watanabe, M. A. Elliott, K. Ruparel, D. R. Roalf, T. D. Satterthwaite, R. C. Gur, R. E. Gur et al., “Harmonization of multi-site diffusion tensor imaging data,” NeuroImage, vol. 161, pp. 149–170, 2017.
- R. C. Craddock, G. A. James, P. E. Holtzheimer III, X. P. Hu, and H. S. Mayberg, “A whole brain fMRI atlas generated via spatially constrained spectral clustering,” Human Brain Mapping, vol. 33, no. 8, pp. 1914–1928, 2012.
- H. Banville, S. U. Wood, C. Aimone, D.-A. Engemann, and A. Gramfort, “Robust learning from corrupted EEG with dynamic spatial filtering,” NeuroImage, vol. 251, p. 118994, 2022.
- X. Geng, L. Wang, X. Wang, B. Qin, T. Liu, and Z. Tu, “How does selective mechanism improve self-attention networks?” in Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics, 2020, pp. 2986–2995.
- J. Zhang, F. Feng, T. Han, X. Gong, and F. Duan, “Detection of autism spectrum disorder using fMRI functional connectivity with feature selection and deep learning,” Cognitive Computation, pp. 1–12, 2022.
- A. Abrol, Z. Fu, M. Salman, R. Silva, Y. Du, S. Plis, and V. Calhoun, “Deep learning encodes robust discriminative neuroimaging representations to outperform standard machine learning,” Nature Communications, vol. 12, no. 1, pp. 1–17, 2021.
- Y. Du, Z. Fu, and V. D. Calhoun, “Classification and prediction of brain disorders using functional connectivity: Promising but challenging,” Frontiers in Neuroscience, vol. 12, p. 525, 2018.
- A. Dosovitskiy, L. Beyer, A. Kolesnikov, D. Weissenborn, X. Zhai, T. Unterthiner, M. Dehghani, M. Minderer, G. Heigold, S. Gelly, J. Uszkoreit, and N. Houlsby, “An image is worth 16x16 words: Transformers for image recognition at scale,” in International Conference on Learning Representations, 2021.
- J. L. Ba, J. R. Kiros, and G. E. Hinton, “Layer normalization,” arXiv preprint arXiv:1607.06450, 2016.
- A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, Ł. Kaiser, and I. Polosukhin, “Attention is all you need,” in Advances in Neural Information Processing Systems, vol. 30, 2017.
- H.-M. Yang, X.-Y. Zhang, F. Yin, and C.-L. Liu, “Robust classification with convolutional prototype learning,” in Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2018, pp. 3474–3482.
- T. N. Kipf and M. Welling, “Semi-supervised classification with graph convolutional networks,” arXiv preprint arXiv:1609.02907, 2016.
- P. Veličković, G. Cucurull, A. Casanova, A. Romero, P. Liò, and Y. Bengio, “Graph attention networks,” in International Conference on Learning Representations, 2018.
- W. Hamilton, Z. Ying, and J. Leskovec, “Inductive representation learning on large graphs,” in Advances in Neural Information Processing Systems, vol. 30, 2017.
- R. Liu, Z.-A. Huang, Y. Hu, Z. Zhu, K.-C. Wong, and K. C. Tan, “Spatial–temporal co-attention learning for diagnosis of mental disorders from resting-state fMRI data,” IEEE Transactions on Neural Networks and Learning Systems, 2023.
- B.-H. Kim, J. C. Ye, and J.-J. Kim, “Learning dynamic graph representation of brain connectome with spatio-temporal attention,” Advances in Neural Information Processing Systems, vol. 34, pp. 4314–4327, 2021.
- R. Tibshirani, “Regression shrinkage and selection via the LASSO,” Journal of the Royal Statistical Society: Series B (Methodological), vol. 58, no. 1, pp. 267–288, 1996.
- I. Guyon, J. Weston, S. Barnhill, and V. Vapnik, “Gene selection for cancer classification using support vector machines,” Machine Learning, vol. 46, pp. 389–422, 2002.
- I. Loshchilov and F. Hutter, “Decoupled weight decay regularization,” in International Conference on Learning Representations, 2019.
- J. Wang, X. Zuo, and Y. He, “Graph-based network analysis of resting-state functional MRI,” Frontiers in Systems Neuroscience, vol. 4, p. 1419, 2010.
- J. S. Nomi and L. Q. Uddin, “Developmental changes in large-scale network connectivity in autism,” NeuroImage: Clinical, vol. 7, pp. 732–741, 2015.
- S. Liang, W. Deng, X. Li, A. J. Greenshaw, Q. Wang, M. Li, X. Ma, T.-J. Bai, Q.-J. Bo, J. Cao et al., “Biotypes of major depressive disorder: Neuroimaging evidence from resting-state default mode network patterns,” NeuroImage: Clinical, vol. 28, p. 102514, 2020.
- A. P. Spencer and M. Goodfellow, “Using deep clustering to improve fMRI dynamic functional connectivity analysis,” NeuroImage, vol. 257, p. 119288, 2022.
- H. Lin, X. Xiang, J. Huang, S. Xiong, H. Ren, and Y. Gao, “Abnormal degree centrality values as a potential imaging biomarker for major depressive disorder: a resting-state functional magnetic resonance imaging study and support vector machine analysis,” Frontiers in Psychiatry, vol. 13, p. 960294, 2022.
- Eunsong Kang (4 papers)
- Jiwon Lee (16 papers)
- Heung-Il Suk (28 papers)
- Da-Woon Heo (5 papers)