PuSH - Publication Server of Helmholtz Zentrum München

Mueller, T.T.* ; Starck, S.* ; Feiner, L.F.* ; Bintsi, K.M.* ; Rueckert, D.* ; Kaissis, G.

Extended graph assessment metrics for regression and weighted graphs.

In:. Berlin [u.a.]: Springer, 2024. 14-26 (Lect. Notes Comput. Sc. ; 14373 LNCS)
DOI
Open Access Green as soon as Postprint is submitted to ZB.
When re-structuring patient cohorts into so-called population graphs, initially independent patients can be incorporated into one interconnected graph structure. This population graph can then be used for medical downstream tasks using graph neural networks (GNNs). The construction of a suitable graph structure is a challenging step in the learning pipeline that can have a severe impact on model performance. To this end, different graph assessment metrics have been introduced to evaluate graph structures. However, these metrics are limited to classification tasks and discrete adjacency matrices, only covering a small subset of real-world applications. In this work, we introduce extended graph assessment metrics (GAMs) for regression tasks and weighted graphs. We focus on two GAMs in particular: homophily and cross-class neighbourhood similarity (CCNS). We extend the notion of GAMs to more than one hop, define homophily for regression tasks, as well as continuous adjacency matrices, and propose a lightweight CCNS distance for discrete and continuous adjacency matrices. We show the correlation of these metrics with model performance on different medical population graphs and under different learning settings, using the TADPOLE and UKBB datasets1(The source code can be found at https://github.com/tamaramueller/ExtendedGAMs).
Altmetric
Additional Metrics?
[➜Log in]
Edit extra informations Login
Publication type Article: Conference contribution
Corresponding Author
Keywords Graph Assessment Metrics ; Graph Neural Networks ; Medical Population Graphs
ISSN (print) / ISBN 0302-9743
e-ISSN 1611-3349
Journal Lecture Notes in Computer Science
Quellenangaben Volume: 14373 LNCS, Issue: , Pages: 14-26 Article Number: , Supplement: ,
Publisher Springer
Publishing Place Berlin [u.a.]
Non-patent literature Publications
Institute(s) Institute for Machine Learning in Biomed Imaging (IML)
Grants NextGenerationEU of the European Union
DOD ADNI (Department of Defense)
Alzheimers Disease Neuroimaging Initiative (ADNI) (National Institutes of Health)
BMBF
ERC