Proceedings chapter
English

On Graph Construction for Classification of Clinical Trials Protocols Using Graph Neural Networks

Published inMichalowski, M., Abidi, S.S.R., Abidi, S. (Ed.), International Conference on Artificial Intelligence in Medicine. AIME 2022., p. 249-259
PublisherCham : Springer
Collection
  • Lecture Notes in Computer Science; 13263
Publication date2022-07-09
First online date2022-07-09
Abstract

A recent trend in health-related machine learning proposes the use of Graph Neural Networks (GNN’s) to model biomedical data. This is justified due to the complexity of healthcare data and the modelling power of graph abstractions. Thus, GNN’s emerge as the natural choice to learn from increasing amounts of healthcare data. While formulating the problem, however, there are usually multiple design choices and decisions that can affect the final performance. In this work, we focus on Clinical Trial (CT) protocols consisting of hierarchical documents, containing free text as well as medical codes and terms, and design a classifier to predict each CT protocol termination risk as “low” or “high”. We show that while using GNN’s to solve this classification task is very successful, the way the graph is constructed is also of importance and one can benefit from making a priori useful information more explicit. While a natural choice is to consider each CT protocol as an independent graph and pose the problem as a graph classification, consistent performance improvements can be achieved by considering them as super-nodes in one unified graph and connecting them according to some metadata, like similar medical condition or intervention, and finally approaching the problem as a node classification task rather than graph classification. We validate this hypothesis experimentally on a large-scale manually labeled CT database. This provides useful insights on the flexibility of graph-based modeling for machine learning in the healthcare domain.

Keywords
  • Graph Neural Networks
  • Machine learning
  • Natural language processing
  • Clinical Trials
  • Healthcare informatics
Citation (ISO format)
FERDOWSI, Sohrab et al. On Graph Construction for Classification of Clinical Trials Protocols Using Graph Neural Networks. In: International Conference on Artificial Intelligence in Medicine. AIME 2022. Michalowski, M., Abidi, S.S.R., Abidi, S. (Ed.). Cham : Springer, 2022. p. 249–259. (Lecture Notes in Computer Science) doi: 10.1007/978-3-031-09342-5_24
Main files (1)
Proceedings chapter (Submitted version)
accessLevelPublic
Identifiers
ISBN978-3-031-09342-5
300views
249downloads

Technical informations

Creation13/07/2022 12:59:00
First validation13/07/2022 12:59:00
Update13/10/2025 15:10:29
Status update16/03/2023 07:11:21
Last indexation27/08/2026 07:19:35
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack