Ia-gcn: Interpretable attention based graph convolutional network for disease prediction

A Kazi, S Farghadani, I Aganj, N Navab - International Workshop on …, 2023 - Springer
International Workshop on Machine Learning in Medical Imaging, 2023Springer
Abstract Interpretability in Graph Convolutional Networks (GCNs) has been explored to
some extent in general in computer vision; yet, in the medical domain, it requires further
examination. Most of the interpretability approaches for GCNs, especially in the medical
domain, focus on interpreting the output of the model in a post-hoc fashion. In this paper, we
propose an interpretable attention module (IAM) that explains the relevance of the input
features to the classification task on a GNN Model. The model uses these interpretations to …
Abstract
Interpretability in Graph Convolutional Networks (GCNs) has been explored to some extent in general in computer vision; yet, in the medical domain, it requires further examination. Most of the interpretability approaches for GCNs, especially in the medical domain, focus on interpreting the output of the model in a post-hoc fashion. In this paper, we propose an interpretable attention module (IAM) that explains the relevance of the input features to the classification task on a GNN Model. The model uses these interpretations to improve its performance. In a clinical scenario, such a model can assist the clinical experts in better decision-making for diagnosis and treatment planning. The main novelty lies in the IAM, which directly operates on input features. IAM learns the attention for each feature based on the unique interpretability-specific losses. We show the application of our model on two publicly available datasets, Tadpole and the UK Biobank (UKBB). For Tadpole we choose the task of disease classification, and for UKBB, age, and sex prediction. The proposed model achieves an increase in an average accuracy of 3.2% for Tadpole and 1.6% for UKBB sex and 2% for the UKBB age prediction task compared to the state-of-the-art. Further, we show exhaustive validation and clinical interpretation of our results.
Springer
以上显示的是最相近的搜索结果。 查看全部搜索结果