Proceedings chapter
English

STM-GNN : Space-Time-and-Memory Graph Neural Networks for Predicting Multi-Drug Resistance Risks in Dynamic Patient Networks

Presented atPavia (Italy), June 23–26, 2025
Published inBellazzi, R., Juarez Herrero, J.M., Sacchi, L. & Zupan, B. (Ed.), Artificial Intelligence in Medicine : 23rd International Conference, AIME 2025, Pavia, Italy, June 23–26, 2025, Proceedings, Part I, p. 160-169
PublisherCham : Springer
Collection
  • Lecture Notes in Computer Science; 15734
Publication date2025
First online date2025-06-23
Abstract

Hospital-acquired infections (HAIs), particularly those caused by multidrug-resistant (MDR) bacteria, pose significant risks to vulnerable patients. Accurate predictive models are important for assessing infection dynamics and informing infection prediction and control (IPC) programmes. Graph-based methods, including graph neural networks (GNNs), offer a powerful approach to model complex relationships between patients and environments but often struggle with data sparsity, irregularity, and heterogeneity. We propose the space-time-and-memory (STM)-GNN, a temporal GNN enhanced with recurrent connectivity designed to capture spatiotemporal infection dynamics. STM-GNN effectively integrates sparse, heterogeneous data combining network information from patient-environment interactions and internal memory from historical colonization and contact patterns. Using a unique IPC dataset containing clinical and environmental colonization information collected from a long-term healthcare unit, we show that STM-GNN effectively addresses the challenges of limited and irregular data in an MDR prediction task. Our model reaches 0.84 AUROC, and achieves the most balanced performance overall compared to classic machine learning algorithms, as well as temporal GNN approaches.

Keywords
  • Hospital acquired infection
  • Temporal graph neural network
Citation (ISO format)
GEISSBUHLER, Damien et al. STM-GNN : Space-Time-and-Memory Graph Neural Networks for Predicting Multi-Drug Resistance Risks in Dynamic Patient Networks. In: Artificial Intelligence in Medicine : 23rd International Conference, AIME 2025, Pavia, Italy, June 23–26, 2025, Proceedings, Part I. Bellazzi, R., Juarez Herrero, J.M., Sacchi, L. & Zupan, B. (Ed.). Pavia (Italy). Cham : Springer, 2025. p. 160–169. (Lecture Notes in Computer Science) doi: 10.1007/978-3-031-95838-0_16
Main files (2)
Proceedings chapter (Published version)
accessLevelRestricted
Proceedings chapter (Accepted version)
accessLevelRestrictedaccessLevelPublic 01/01/2027
Identifiers
ISBN978-3-031-95837-3
20views
0downloads

Technical informations

Creation29/06/2025 21:03:59
First validation02/09/2025 08:26:57
Update02/09/2025 08:26:57
Status update02/09/2025 08:26:57
Last indexation02/09/2025 08:26:58
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack