Abstract
The difference between the chronological and biological brain age of a subject can be an important biomarker for neurodegenerative diseases, thus brain age estimation can be crucial in clinical settings. One way to incorporate multimodal information into this estimation is through population graphs, which combine various types of imaging data and capture the associations among individuals within a population. In medical imaging, population graphs have demonstrated promising results, mostly for classification tasks. In most cases, the graph structure is pre-defined and remains static during training. However, extracting population graphs is a non-trivial task and can significantly impact the performance of Graph Neural Networks (GNNs), which are sensitive to the graph structure. In this work, we highlight the importance of a meaningful graph construction and experiment with different population-graph construction methods and their effect on GNN performance on brain age estimation. We use the homophily metric and graph visualizations to gain valuable quantitative and qualitative insights on the extracted graph structures. For the experimental evaluation, we leverage the UK Biobank dataset, which offers many imaging and non-imaging phenotypes. Our results indicate that architectures highly sensitive to the graph structure, such as Graph Convolutional Network (GCN) and Graph Attention Network (GAT), struggle with low homophily graphs, while other architectures, such as GraphSage and Chebyshev, are more robust across different homophily ratios. We conclude that static graph construction approaches are potentially insufficient for the task of brain age estimation and make recommendations for alternative research directions.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Ahmedt-Aristizabal, D., Armin, M.A., Denman, S., Fookes, C., Petersson, L.: Graph-based deep learning for medical diagnosis and analysis: past, present and future. Sensors 21(14), 4758 (2021)
Alam, S.B., Nakano, R., Kamiura, N., Kobashi, S.: Morphological changes of aging brain structure in MRI analysis. In: 2014 Joint 7th International Conference on Soft Computing and Intelligent Systems (SCIS) and 15th International Symposium on Advanced Intelligent Systems (ISIS), pp. 683–687. IEEE (2014)
Alfaro-Almagro, F., et al.: Image processing and quality control for the first 10,000 brain imaging datasets from UK Biobank. NeuroImage 166, 400–424 (2018)
Bi, W., Du, L., Fu, Q., Wang, Y., Han, S., Zhang, D.: Make heterophily graphs better fit GNN: a graph rewiring approach. arXiv preprint arXiv:2209.08264 (2022)
Bintsi, K.M., Baltatzis, V., Potamias, R.A., Hammers, A., Rueckert, D.: Multimodal brain age estimation using interpretable adaptive population-graph learning. In: Greenspan, H., et al. (eds.) MICCAI 2023. LNCS, vol. 14227, pp. 195–204. Springer, Cham (2023). https://doi.org/10.1007/978-3-031-43993-3_19
Cole, J.H.: Multimodality neuroimaging brain-age in UK biobank: relationship to biomedical, lifestyle, and cognitive factors. Neurobiol. Aging 92, 34–42 (2020)
Cole, J.H., et al.: Predicting brain age with deep learning from raw imaging data results in a reliable and heritable biomarker. Neuroimage 163, 115–124 (2017)
Cosmo, L., Kazi, A., Ahmadi, S.-A., Navab, N., Bronstein, M.: Latent-graph learning for disease prediction. In: Martel, A.L., et al. (eds.) MICCAI 2020. LNCS, vol. 12262, pp. 643–653. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-59713-9_62
Davatzikos, C., Bhatt, P., Shaw, L.M., Batmanghelich, K.N., Trojanowski, J.Q.: Neurobiol. AgingPrediction of mci to ad conversion, via MRI, CSF biomarkers, and pattern classification 32(12), 2322-e19 (2011)
Defferrard, M., Bresson, X., Vandergheynst, P.: Convolutional neural networks on graphs with fast localized spectral filtering. In: Advances in Neural Information Processing Systems, vol. 29 (2016)
Erdős, P., Rényi, A., et al.: On the evolution of random graphs. Publ. Math. Inst. Hung. Acad. Sci. 5(1), 17–60 (1960)
Fey, M., Lenssen, J.E.: Fast graph representation learning with PyTorch geometric. In: ICLR Workshop on Representation Learning on Graphs and Manifolds (2019)
Franke, K., Gaser, C.: Ten years of brainage as a neuroimaging biomarker of brain aging: what insights have we gained? Front. Neurol. 789 (2019)
Hamilton, W., Ying, Z., Leskovec, J.: Inductive representation learning on large graphs. In: Advances in Neural Information Processing Systems, vol. 30 (2017)
Kazi, A., Cosmo, L., Ahmadi, S.A., Navab, N., Bronstein, M.: Differentiable graph module (DGM) for graph convolutional networks. IEEE Trans. Pattern Anal. Mach. Intell. (2022)
Kazi, A., et al.: InceptionGCN: receptive field aware graph convolutional network for disease prediction. In: Chung, A.C.S., Gee, J.C., Yushkevich, P.A., Bao, S. (eds.) IPMI 2019. LNCS, vol. 11492, pp. 73–85. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-20351-1_6
Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907 (2016)
Koutsouleris, N., et al.: Accelerated brain aging in schizophrenia and beyond: a neuroanatomical marker of psychiatric disorders. Schizophr. Bull. 40(5), 1140–1153 (2014)
Loshchilov, I., Hutter, F.: Decoupled weight decay regularization. arXiv preprint arXiv:1711.05101 (2017)
Luan, S., Hua, C., Lu, Q., Zhu, J., Chang, X.W., Precup, D.: When do we need GNN for node classification? arXiv preprint arXiv:2210.16979 (2022)
Ma, Y., Liu, X., Shah, N., Tang, J.: Is homophily a necessity for graph neural networks? arXiv preprint arXiv:2106.06134 (2021)
Mueller, T., Starck, S., Feiner, L.F., Bintsi, K.M., Rueckert, D., Kaissis, G.: Extended graph assessment metrics for regression and weighted graphs. arXiv preprint (2023)
Parisot, S., et al.: Disease prediction using graph convolutional networks: application to autism spectrum disorder and Alzheimer’s disease. Med. Image Anal. 48, 117–130 (2018)
Reeve, A., Simcox, E., Turnbull, D.: Ageing and Parkinson’s disease: why is advancing age the biggest risk factor? Ageing Res. Rev. 14, 19–30 (2014)
Stankeviciute, K., Azevedo, T., Campbell, A., Bethlehem, R., Lio, P.: Population graph GNNs for brain age prediction. In: ICML Workshop on Graph Representation Learning and Beyond (GRL+), pp. 17–83 (2020)
Veličković, P., Cucurull, G., Casanova, A., Romero, A., Lio, P., Bengio, Y.: Graph attention networks. arXiv preprint arXiv:1710.10903 (2017)
Wei, S., Zhao, Y.: Graph learning: a comprehensive survey and future directions. arXiv preprint arXiv:2212.08966 (2022)
Zhao, X., Zhou, F., Ou-Yang, L., Wang, T., Lei, B.: Graph convolutional network analysis for mild cognitive impairment prediction. In: 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019), pp. 1598–1601. IEEE (2019)
Zheng, X., Liu, Y., Pan, S., Zhang, M., Jin, D., Yu, P.S.: Graph neural networks for graphs with heterophily: a survey. arXiv preprint arXiv:2202.07082 (2022)
Zhu, J., Yan, Y., Zhao, L., Heimann, M., Akoglu, L., Koutra, D.: Beyond homophily in graph neural networks: current limitations and effective designs. In: Advances in Neural Information Processing Systems, vol. 33, pp. 7793–7804 (2020)
Acknowledgements
KMB would like to acknowledge funding from the EPSRC Centre for Doctoral Training in Medical Imaging (EP/L015226/1).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Bintsi, KM., Mueller, T.T., Starck, S., Baltatzis, V., Hammers, A., Rueckert, D. (2024). A Comparative Study of Population-Graph Construction Methods and Graph Neural Networks for Brain Age Regression. In: Ahmadi, SA., Pereira, S. (eds) Graphs in Biomedical Image Analysis, and Overlapped Cell on Tissue Dataset for Histopathology. MICCAI 2023. Lecture Notes in Computer Science, vol 14373. Springer, Cham. https://doi.org/10.1007/978-3-031-55088-1_6
Download citation
DOI: https://doi.org/10.1007/978-3-031-55088-1_6
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-55087-4
Online ISBN: 978-3-031-55088-1
eBook Packages: Computer ScienceComputer Science (R0)