Skip to main content
Have a personal or library account? Click to login
Predicting Sepsis in ICU Patients Using GRU-D and LightGBM for Clinical Decision Support Cover

Predicting Sepsis in ICU Patients Using GRU-D and LightGBM for Clinical Decision Support

By:  and    
Open Access
|Oct 2026

Figures & Tables

Table 1:

Summary of literature reviews

YearReferenceDescriptionMethodology usedResults obtainedDataset usedSignificanceResearch gaps
2025Yang et al. [29]Early sepsis predictionXGBoostEarly prediction using vital signsMulti-center datasetSupports early detectionLimited temporal information
2025Kim et al. [30]Temporal dynamic embeddingTDE neural architecture.TDE performed competitivelyPhysioNet 2019Avoids heavy imputationRequire careful tuning on very sparse variables.
2025Almasoud et al. [31]Hybrid DL modelCNNAUC—0.998PhysioNet 2019Improved decision-makingLimit the generalizability of the results
2024Gao et al. [32]Sepsis mortality predictionMachine learningMortality risk predictionICU patient dataSupports mortality risk assessmentRequires broader validation
2024Boussina et al. [33]Deployment of a DL sepsis alertFeed forward neural networkAUROC—0.945UC San Diego Health SystemSignificant reduction in mortalityConfounding biases
2024Zheng et al. [34]scCaT capsule + TransformerCapsule networks + Transformer decoderAUROC—0.93RNA data usedOffers biologically interpretable featuresMay not directly translate to clinical early warning
2023Solís-García et al. [35]Experimental comparison of multiple ML/AI,Classical ML and DL modelsAUROC—0.93Clinical ICU datasetsPreprocessing model rankingsConclusions dependent on the chosen
2023Saleena et al. [36]Summarizing ML/DL modelsNarrative synthesis of model types, datasetsAUROC—0.93MIMIC, PhysioNetSynthesizes gapsHeterogeneity of primary studies
2023Ramirez-Alcocer et al. [37]DL for predictive processDL sequence models (RNNs/transformers)Precision—0.92
Recall—0.90
Healthcare event-sequence datasets)Demonstrates utility of DL for processFocus is on process monitoring workflow events
2022Apalak et al. [38]Conditional recurrentRecurrent adversarial network architectureAUROC—0.94MIMIC datasetGeneralization in sepsisAdversarial training
2022Chen et al. [39]Comparative study of 6 ML methodsDecision trees, SVM, ensemble methodsXGBoost model accuracy—0.79BupaR packageHighlights model differencesLimited external validation
2021Kuo et al. [40]ANN models using real-world EHR data.Artificial Neural NetworksAUC—0.786Real-world hospital EHRModels can be robust without artificial imputationMay reduce predictive performance
2021Shashikumar et al. [41]DeepAISE: a recurrent neural survival modelRecurrent survival modeling RNN + survival analysisAUC—0.89MIMICTo predict both event risk and timingComplexity and survival loss
2021Aşuroğlu et al. [42]To predict SOFA-based sepsis severityCNNs, RFAUC—0.982MIMIC-III ICU datasetThe approach aligns with the need for real-timeMay not generalize to different ICU settings
2020Burdick et al. [43]Retrospective validation of an ML algorithmML algorithmAUROC 0.93Dataset from 461 US hospitalsDemonstrates model generalizabilityHeterogeneity in labeling
2020Pettinati et al. [44]Practical ML approaches focusing onML pipelines, feature engineeringAUROC—0.56MIMIC 3 datasetPractical considerationsLimited external evaluation details
2020Lauritsen et al. [45]Uses DL on EHR event sequences.Sequence models RNN/transformer-styleAUROC—0.75Multiple Danish hospitalsAdvantages of event-sequence modelingTransferability depends on EHR event logging
2019Scherpf et al. [46]Applies RNNs for modelRNN/LSTM sequence modelingAUROC—0.81MIMIC-III public ICU datasetDemonstrates applicability of RNNsResults sensitive to sepsis labeling
2018Wang et al. [47]ML framework.Ensemble ML methods with feature selectionSensitivity—0.91Clinical datasetsEarly contribution to ensemble approachesLack of large external validation
2017Sawada et al. [48]Transfer learning approach for DNNs.Transfer learning methods for deep networksAccuracy—0.90Actual two dimensional imageShows value of transfer learningRequiring further validation and broader external testing

[i] AUC, area under the ROC curve; CNN, convolutional neural networks; DL, deep learning; DNNs, Deep Neural Networks; EHR, electronic health records; ICUs, intensive care units; LSTM, long short-term memory; ML, machine learning; RF, random forest; RNA, Ribonucleic Acid; RNN, Recurrent Neural Network; SOFA, Sequential Organ Failure Assessment; SVM, support vector machine; TDE, Temporal Dynamic Embedding.

Figure 1:

GRU-D based on time decay factor [49]. GRU-D, gated recurrent unit with decay.

Figure 2:

Distribution of sepsis onset hours.

Figure 3:

Distribution of records of time-series per patient.

Table 2:

Pseudo-ablation analysis of VAE latent dimension

Latent dimensionObserved effect on representationDownstream impact (GRU-D + LightGBM)
2Over-compressed, loss of clinical variabilityReduced recall and unstable temporal learning
4 (Selected)Balanced compression preserving key physiological patternsStable convergence and best precision–recall balance
8Increased redundancy in latent featuresNo improvement in AUROC/AUPRC; higher complexity
≥16Over-parameterizationIncreased overfitting risk without performance gains

[i] AUPRC, area under the precision-recall curve; GRU-D, gated recurrent unit with decay; LightGBM, Light gradient boosting machine; VAE, variational autoencoder.

Table 3:

VAE reconstruction effects on features

FeatureOriginal valueAfter VAE reconstructionDifferenceInterpretation
HR8885.7−2.3Slight smoothing of variation
RR2221.2−0.8Noise reduction
MAP7677.1+1.1Maintains overall trend
Oxygen saturation (SpO2)9493.6−0.4Small value adjustment
Body temperature (°C)37.637.4−0.2Retains clinical normality
SBP122119.8−2.2Reduced measurement noise
DBP7473.2−0.8Preserved pattern with smoothing
Glucose level (mg/dL)148140.5−7.5Reduced variability/stabilization
WBC12.412.1−0.3Maintains infection trend
Lactate level (mmol/L)2.62.4−0.2Slight artifact reduction
Creatinine (mg/dL)1.91.8−0.1Kidney function pattern retained
Platelet count (× 103/µL)182179.6−2.4Preserves clinical direction

[i] DBP, diastolic blood pressure; HR, heart rate; MAP, mean arterial pressure; RR, respiratory rate; SBP, systolic blood pressure; VAE, variational autoencoder; WBC, white blood cell count.

Figure 4:

Workflow of GRU-D and LightGBM-based sepsis prediction model. GRU-D, gated recurrent unit with decay; ICUs, intensive care units; LightGBM, Light gradient boosting machine.

Table 4:

Hyperparameter settings for GRU-D and LightGBM models

ModelHyperparameterDescriptionTuned range/valueOptimal value
GRU-DHidden sizeNumber of hidden units in GRU-D32, 64, 128, 256128
GRU-DNumber of layersDepth of GRU-D model1, 2, 32
GRU-DDropoutDropout rate for regularization0.1, 0.3, 0.50.3
GRU-DLearning rateStep size for optimizer1e.4, 5e.4, 1e.35e.4
GRU-DBatch sizeNumber of samples per batch32, 64, 12864
GRU-DOptimizerOptimization algorithmAdam, RMsprop, SGDAdam
LightGBMn_estimatorsNumber of boosting iterations100, 200, 500, 1000500
LightGBMmax_depthMaximum depth of tree3, 5, 7, −15
LightGBMlearning_rateShrinkage rate applied to each tree0.01, 0.05, 0.10.05
LightGBMsubsampleFractions of samples per boosting iteration0.6, 0.8, 1.00.8
LightGBMcolsample_bytreeFraction of features used per iteration0.6, 0.8, 1.00.8
LightGBMred_lambdaL2 regularization term0, 0.1, 0.5, 1.00.1

[i] GRU-D, gated recurrent unit with decay; LightGBM, Light gradient boosting machine.

Table 5:

GRD-D and LightGBM performance metrics

ModelAccuracy (%)Precision (%)Recall (%)F1-Score (%)ROC AUCPR AUC
GRU-D94.3981.6734.6248.630.900.61
LightGBM94.9877.09949.2959.940.920.68

[i] AUC, area under the ROC curve; GRU-D, gated recurrent unit with decay; LightGBM, Light gradient boosting machine; PR, Precision–Recall.

Figure 5:

LightGBM precision-recall curve. LightGBM, Light gradient boosting machine.

Figure 6:

LightGBM ROC curve. LightGBM, Light gradient boosting machine.

Figure 7:

LightGBM confusion matrix. LightGBM, Light gradient boosting machine.

Figure 8:

GRU-D precision-recall curve. AUC, area under the ROC curve; GRU-D, gated recurrent unit with decay.

Figure 9:

GRU-D ROC curve. AUC, area under the ROC curve; GRU-D, gated recurrent unit with decay.

Figure 10:

GRU-D ROC confusion matrix. GRU-D, gated recurrent unit with decay.

Figure 11:

ROC curve comparison. AUC, area under the ROC curve; GRU-D, gated recurrent unit with decay; LightGBM, Light gradient boosting machine.

Figure 12:

Precision recall curve comparison. AUC, area under the ROC curve; GRU-D, gated recurrent unit with decay; LightGBM, Light gradient boosting machine.

Table 6:

Cross-validation performance of GRU-D, LightGBM, and hybrid models

ModelAccuracy (mean ± std)Precision (mean ± std)Recall (mean ± std)F1-score (mean ± std)AUROC (mean ± std)
GRU-D0.827 ± 0.0140.842 ± 0.0210.815 ± 0.0240.828 ± 0.0190.912 ± 0.010
LightGBM0.861 ± 0.0180.835 ± 0.0200.808 ± 0.0260.821 ± 0.0220.905 ± 0.012
Hybrid (GRU-D + LGBM)0.888 ± 0.0110.857 ± 0.0170.845 ± 0.0200.851 ± 0.0180.931 ± 0.009

[i] GRU-D, gated recurrent unit with decay; LightGBM, Light gradient boosting machine.

Figure 13:

Top 20 important features from LightGBM model. GRU-D, gated recurrent unit with decay; LightGBM, Light gradient boosting machine.

Figure 14:

Feature correlation heatmap of clinical variables. DBP, diastolic blood pressure; HR, heart rate; MAP, mean arterial pressure; SBP, systolic blood pressure; WBC, white blood cell count.

Figure 15:

GRU-D embedding heatmap representing learned latent features. GRU-D, gated recurrent unit with decay.

Table 7:

Performance of the models

ModelAccuracy (%)Precision (%)Recall (%)F1-score (%)AUC (%)
Best Single LSTM0.940.500.600.540.60
Best Single GRU0.940.500.620.600.62
Baseline ensemble (LSTM)0.940.500.600.550.60
Baseline ensemble (GRU)0.940.500.590.540.59
Ensemble GRU & LSTM0.940.500.600.540.60

[i] AUC, area under the ROC curve; GRU, gated recurrent unit; LSTM, long short-term memory.

Language: English
Submitted on: Dec 4, 2025
Published on: Oct 8, 2026
Published by: International Journal on Smart Sensing and Intelligent Systems
In partnership with: Paradigm Publishing Services
Publication frequency: Volume open

© 2026 Manju Goel, Renu Popli, published by International Journal on Smart Sensing and Intelligent Systems
This work is licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 License.