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

Geissbuhler, Damien; Bornet, Alban; Marques, Catarina; Anjos, André; Pereira, Sónia; Teodoro, Douglas

http://hdl.handle.net/10400.8/14203

Use this identifier to reference this record.

Name:	Description:	Size:	Format:
AIME2025_-_Conf_Proc_-_STM_GNN_Predict.pdf		4.48 MB	Adobe PDF	Download

Send Feedback

Authors

Abstract(s)

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.

Description

Book Series Part of the book series: Lecture Notes in Computer Science (LNAI,volume 15734)

Keywords

Hospital acquired infection Temporal graph neural network

URI

http://hdl.handle.net/10400.8/14203

Citation

Geissbuhler, D., Bornet, A., Marques, C., Anjos, A., Pereira, S., Teodoro, D. (2025). STM-GNN: Space-Time-and-Memory Graph Neural Networks for Predicting Multi-Drug Resistance Risks in Dynamic Patient Networks. In: Bellazzi, R., Juarez Herrero, J.M., Sacchi, L., Zupan, B. (eds) Artificial Intelligence in Medicine. AIME 2025. Lecture Notes in Computer Science(), vol 15734. Springer, Cham. https://doi.org/10.1007/978-3-031-95838-0_16