English

Sepsis Prediction Using Graph Convolutional Networks over Patient-Feature-Value Triplets

Machine Learning 2025-12-08 v1

Abstract

In the intensive care setting, sepsis continues to be a major contributor to patient illness and death; however, its timely detection is hindered by the complex, sparse, and heterogeneous nature of electronic health record (EHR) data. We propose Triplet-GCN, a single-branch graph convolutional model that represents each encounter as patient-feature-value triplets, constructs a bipartite EHR graph, and learns patient embeddings via a Graph Convolutional Network (GCN) followed by a lightweight multilayer perceptron (MLP). The pipeline applies type-specific preprocessing -- median imputation and standardization for numeric variables, effect coding for binary features, and mode imputation with low-dimensional embeddings for rare categorical attributes -- and initializes patient nodes with summary statistics, while retaining measurement values on edges to preserve "who measured what and by how much". In a retrospective, multi-center Chinese cohort (N = 648; 70/30 train-test split) drawn from three tertiary hospitals, Triplet-GCN consistently outperforms strong tabular baselines (KNN, SVM, XGBoost, Random Forest) across discrimination and balanced error metrics, yielding a more favorable sensitivity-specificity trade-off and improved overall utility for early warning. These findings indicate that encoding EHR as triplets and propagating information over a patient-feature graph produce more informative patient representations than feature-independent models, offering a simple, end-to-end blueprint for deployable sepsis risk stratification.

Keywords

Cite

@article{arxiv.2512.05416,
  title  = {Sepsis Prediction Using Graph Convolutional Networks over Patient-Feature-Value Triplets},
  author = {Bozhi Dan and Di Wu and Ji Xu and Xiang Liu and Yiziting Zhu and Xin Shu and Yujie Li and Bin Yi},
  journal= {arXiv preprint arXiv:2512.05416},
  year   = {2025}
}
R2 v1 2026-07-01T08:10:41.990Z