AMR-GNN: A multi-representation graph neural network framework to enable genomic antimicrobial resistance prediction

Abstract

Whole-genome sequencing (WGS) data are an invaluable resource for understanding antimicrobial resistance (AMR) mechanisms. However, WGS data are high-dimensional and the lack of standardized genomic representations is a key barrier to AMR prediction. To fully explore these high-resolution data, we propose AMR-GNN, a graph deep learning-based framework that integrates multiple genomic representations with graph neural networks (GNN) to enable AMR prediction from genomic sequence data. We tested AMR-GNN with Pseudomonas aeruginosa, a clinically relevant Gram-negative bacterial pathogen known for its complex AMR mechanisms. We demonstrate that AMR-GNN addresses several key problems in AMR prediction with data-driven machine learning (ML) approaches, including using multiple genomic representations to enhance performance, mitigate the influence of clonal relationships, and identify informative biomarkers to provide explainability and generate novel hypotheses. Follow-up validation on the largest publicly available dataset spanning both Gram-negative and Gram-positive pathogens highlights AMR-GNNs broad applicability in detecting AMR in diverse and clinically relevant pathogen-drug combinations.