Background: We aimed to develop and validate an automated machine learning (autoML) prediction model for cardiac surgery-associated acute kidney injury (CSA-AKI). Methods: Using 69 preoperative variables, we developed several models to predict post-operative AKI in adult patients undergoing cardiac surgery. Models included autoML and non-autoML types, including decision tree (DT), random forest (RF), extreme gradient boosting (XGBoost), and artificial neural network (ANN), as well as a logistic regression prediction model. We then compared model performance using area under the receiver operating characteristic curve (AUROC) and assessed model calibration using Brier score on the independent testing dataset. Results: The incidence of CSA-AKI was 36%. Stacked ensemble autoML had the highest predictive performance among autoML models, and was chosen for comparison with other non-autoML and multivariable logistic regression models. The autoML had the highest AUROC (0.79), followed by RF (0.78), XGBoost (0.77), multivariable logistic regression (0.77), ANN (0.75), and DT (0.64). The autoML had comparable AUROC with RF and outperformed the other models. The autoML was well-calibrated. The Brier score for autoML, RF, DT, XGBoost, ANN, and multivariable logistic regression was 0.18, 0.18, 0.21, 0.19, 0.19, and 0.18, respectively. We applied SHAP and LIME algorithms to our autoML prediction model to extract an explanation of the variables that drive patient-specific predictions of CSA-AKI. Conclusion: We were able to present a preoperative autoML prediction model for CSA-AKI that provided high predictive performance that was comparable to RF and superior to other ML and multivariable logistic regression models. The novel approaches of the proposed explainable preoperative autoML prediction model for CSA-AKI may guide clinicians in advancing individualized medicine plans for patients under cardiac surgery.
Machine Learning Consensus Clustering of Morbidly Obese Kidney Transplant Recipients in the United States
Background: This study aimed to better characterize morbidly obese kidney transplant recipients, their clinical characteristics, and outcomes by using an unsupervised machine learning approach. Methods: Consensus cluster analysis was applied to OPTN/UNOS data from 2010 to 2019 based on recipient, donor, and transplant characteristics in kidney transplant recipients with a pre-transplant BMI ≥ 40 kg/m2. Key cluster characteristics were identified using the standardized mean difference. Post-transplant outcomes, including death-censored graft failure, patient death, and acute allograft rejection, were compared among the clusters. Results: Consensus clustering analysis identified 3204 kidney transplant recipients with a BMI ≥ 40 kg/m2. In this cohort, five clinically distinct clusters were identified. Cluster 1 recipients were predominantly white and non-sensitized, had a short dialysis time or were preemptive, and were more likely to receive living donor kidney transplants. Cluster 2 recipients were older and diabetic. They were likely to have been on dialysis >3 years and receive a standard KDPI deceased donor kidney. Cluster 3 recipients were young, black, and had kidney disease secondary to hypertension or glomerular disease. Cluster 3 recipients had >3 years of dialysis and received non-ECD, young, deceased donor kidney transplants with a KDPI < 85%. Cluster 4 recipients were diabetic with variable dialysis duration who either received non-ECD standard KDPI kidneys or living donor kidney transplants. Cluster 5 recipients were young retransplants that were sensitized. One-year patient survival in clusters 1, 2, 3, 4, and 5 was 98.0%, 94.4%, 98.5%, 98.7%, and 97%, and one-year death-censored graft survival was 98.1%, 93.0%, 96.1%, 98.8%, and 93.0%, respectively. Cluster 2 had the worst one-year patient survival. Clusters 2 and 5 had the worst one-year death-censored graft survival. Conclusions: With the application of unsupervised machine learning, variable post-transplant outcomes are observed among morbidly obese kidney transplant recipients. Recipients with earlier access to transplant and living donation show superior outcomes. Unexpectedly, reduced graft survival in cluster 3 recipients perhaps underscores socioeconomic access to post-transplant support and minorities being disadvantaged in access to preemptive and living donor transplants. Despite obesity-related concerns, one-year patient and graft survival were favorable in all clusters, and obesity itself should be reconsidered as a hard barrier to kidney transplantation.
Distinct Phenotypes of Kidney Transplant Recipients in the United States with Limited Functional Status as Identified through Machine Learning Consensus Clustering
Background: There have been concerns regarding increased perioperative mortality, length of hospital stay, and rates of graft loss in kidney transplant recipients with functional limitations. The application of machine learning consensus clustering approach may provide a novel understanding of unique phenotypes of functionally limited kidney transplant recipients with distinct outcomes in order to identify strategies to improve outcomes. Methods: Consensus cluster analysis was performed based on recipient-, donor-, and transplant-related characteristics in 3205 functionally limited kidney transplant recipients (Karnofsky Performance Scale (KPS) < 40% at transplant) in the OPTN/UNOS database from 2010 to 2019. Each cluster’s key characteristics were identified using the standardized mean difference. Posttransplant outcomes, including death-censored graft failure, patient death, and acute allograft rejection were compared among the clusters Results: Consensus cluster analysis identified two distinct clusters that best represented the clinical characteristics of kidney transplant recipients with limited functional status prior to transplant. Cluster 1 patients were older in age and were more likely to receive deceased donor kidney transplant with a higher number of HLA mismatches. In contrast, cluster 2 patients were younger, had shorter dialysis duration, were more likely to be retransplants, and were more likely to receive living donor kidney transplants from HLA mismatched donors. As such, cluster 2 recipients had a higher PRA, less cold ischemia time, and lower proportion of machine-perfused kidneys. Despite having a low KPS, 5-year patient survival was 79.1 and 83.9% for clusters 1 and 2; 5-year death-censored graft survival was 86.9 and 91.9%. Cluster 1 had lower death-censored graft survival and patient survival but higher acute rejection, compared to cluster 2. Conclusion: Our study used an unsupervised machine learning approach to characterize kidney transplant recipients with limited functional status into two clinically distinct clusters with differing posttransplant outcomes.
Machine Learning Consensus Clustering Approach for Hospitalized Patients with Dysmagnesemia
Background: The objectives of this study were to classify patients with serum magnesium derangement on hospital admission into clusters using unsupervised machine learning approach and to evaluate the mortality risks among these distinct clusters. Methods: Consensus cluster analysis was performed based on demographic information, principal diagnoses, comorbidities, and laboratory data in hypomagnesemia (serum magnesium ≤ 1.6 mg/dL) and hypermagnesemia cohorts (serum magnesium ≥ 2.4 mg/dL). Each cluster’s key features were determined using the standardized mean difference. The associations of the clusters with hospital mortality and one-year mortality were assessed. Results: In hypomagnesemia cohort (n = 13,320), consensus cluster analysis identified three clusters. Cluster 1 patients had the highest comorbidity burden and lowest serum magnesium. Cluster 2 patients had the youngest age, lowest comorbidity burden, and highest kidney function. Cluster 3 patients had the oldest age and lowest kidney function. Cluster 1 and cluster 3 were associated with higher hospital and one-year mortality compared to cluster 2. In hypermagnesemia cohort (n = 4671), the analysis identified two clusters. Compared to cluster 1, the key features of cluster 2 included older age, higher comorbidity burden, more hospital admissions primarily due to kidney disease, more acute kidney injury, and lower kidney function. Compared to cluster 1, cluster 2 was associated with higher hospital mortality and one-year mortality. Conclusion: Our cluster analysis identified clinically distinct phenotypes with differing mortality risks in hospitalized patients with dysmagnesemia. Future studies are required to assess the application of this ML consensus clustering approach to care for hospitalized patients with dysmagnesemia.
Machine Learning Consensus Clustering Approach for Patients with Lactic Acidosis in Intensive Care Units
Background: Lactic acidosis is a heterogeneous condition with multiple underlying causes and associated outcomes. The use of multi-dimensional patient data to subtype lactic acidosis can personalize patient care. Machine learning consensus clustering may identify lactic acidosis subgroups with unique clinical profiles and outcomes. Methods: We used the Medical Information Mart for Intensive Care III database to abstract electronic medical record data from patients admitted to intensive care units (ICU) in a tertiary care hospital in the United States. We included patients who developed lactic acidosis (defined as serum lactate ≥ 4 mmol/L) within 48 h of ICU admission. We performed consensus clustering analysis based on patient characteristics, comorbidities, vital signs, organ supports, and laboratory data to identify clinically distinct lactic acidosis subgroups. We calculated standardized mean differences to show key subgroup features. We compared outcomes among subgroups. Results: We identified 1919 patients with lactic acidosis. The algorithm revealed three best unique lactic acidosis subgroups based on patient variables. Cluster 1 (n = 554) was characterized by old age, elective admission to cardiac surgery ICU, vasopressor use, mechanical ventilation use, and higher pH and serum bicarbonate. Cluster 2 (n = 815) was characterized by young age, admission to trauma/surgical ICU with higher blood pressure, lower comorbidity burden, lower severity index, and less vasopressor use. Cluster 3 (n = 550) was characterized by admission to medical ICU, history of liver disease and coagulopathy, acute kidney injury, lower blood pressure, higher comorbidity burden, higher severity index, higher serum lactate, and lower pH and serum bicarbonate. Cluster 3 had the worst outcomes, while cluster 1 had the most favorable outcomes in terms of persistent lactic acidosis and mortality. Conclusions: Consensus clustering analysis synthesized the pattern of clinical and laboratory data to reveal clinically distinct lactic acidosis subgroups with different outcomes.
Machine Learning Prediction Models for Mortality in Intensive Care Unit Patients with Lactic Acidosis
Background: Lactic acidosis is the most common cause of anion gap metabolic acidosis in the intensive care unit (ICU), associated with poor outcomes including mortality. We sought to compare machine learning (ML) approaches versus logistic regression analysis for prediction of mortality in lactic acidosis patients admitted to the ICU. Methods: We used the Medical Information Mart for Intensive Care (MIMIC-III) database to identify ICU adult patients with lactic acidosis (serum lactate ≥4 mmol/L). The outcome of interest was hospital mortality. We developed prediction models using four ML approaches consisting of random forest (RF), decision tree (DT), extreme gradient boosting (XGBoost), artificial neural network (ANN), and statistical modeling with forward stepwise logistic regression using the testing dataset. We then assessed model performance using area under the receiver operating characteristic curve (AUROC), accuracy, precision, error rate, Matthews correlation coefficient (MCC), F1 score, and assessed model calibration using the Brier score, in the independent testing dataset. Results: Of 1919 lactic acidosis ICU patients, 1535 and 384 were included in the training and testing dataset, respectively. Hospital mortality was 30%. RF had the highest AUROC at 0.83, followed by logistic regression 0.81, XGBoost 0.81, ANN 0.79, and DT 0.71. In addition, RF also had the highest accuracy (0.79), MCC (0.45), F1 score (0.56), and lowest error rate (21.4%). The RF model was the most well-calibrated. The Brier score for RF, DT, XGBoost, ANN, and multivariable logistic regression was 0.15, 0.19, 0.18, 0.19, and 0.16, respectively. The RF model outperformed multivariable logistic regression model, SOFA score (AUROC 0.74), SAP II score (AUROC 0.77), and Charlson score (AUROC 0.69). Conclusion: The ML prediction model using RF algorithm provided the highest predictive performance for hospital mortality among ICU patient with lactic acidosis.
Clinically Distinct Subtypes of Acute Kidney Injury on Hospital Admission Identified by Machine Learning Consensus Clustering
Background: We aimed to cluster patients with acute kidney injury at hospital admission into clinically distinct subtypes using an unsupervised machine learning approach and assess the mortality risk among the distinct clusters. Methods: We performed consensus clustering analysis based on demographic information, principal diagnoses, comorbidities, and laboratory data among 4289 hospitalized adult patients with acute kidney injury at admission. The standardized difference of each variable was calculated to identify each cluster’s key features. We assessed the association of each acute kidney injury cluster with hospital and one-year mortality. Results: Consensus clustering analysis identified four distinct clusters. There were 1201 (28%) patients in cluster 1, 1396 (33%) patients in cluster 2, 1191 (28%) patients in cluster 3, and 501 (12%) patients in cluster 4. Cluster 1 patients were the youngest and had the least comorbidities. Cluster 2 and cluster 3 patients were older and had lower baseline kidney function. Cluster 2 patients had lower serum bicarbonate, strong ion difference, and hemoglobin, but higher serum chloride, whereas cluster 3 patients had lower serum chloride but higher serum bicarbonate and strong ion difference. Cluster 4 patients were younger and more likely to be admitted for genitourinary disease and infectious disease but less likely to be admitted for cardiovascular disease. Cluster 4 patients also had more severe acute kidney injury, lower serum sodium, serum chloride, and serum bicarbonate, but higher serum potassium and anion gap. Cluster 2, 3, and 4 patients had significantly higher hospital and one-year mortality than cluster 1 patients (p < 0.001). Conclusion: Our study demonstrated using machine learning consensus clustering analysis to characterize a heterogeneous cohort of patients with acute kidney injury on hospital admission into four clinically distinct clusters with different associated mortality risks.
Machine Learning Consensus Clustering Approach for Hospitalized Patients with Phosphate Derangements
Background: The goal of this study was to categorize patients with abnormal serum phosphate upon hospital admission into distinct clusters utilizing an unsupervised machine learning approach, and to assess the mortality risk associated with these clusters. Methods: We utilized the consensus clustering approach on demographic information, comorbidities, principal diagnoses, and laboratory data of hypophosphatemia (serum phosphate ≤ 2.4 mg/dL) and hyperphosphatemia cohorts (serum phosphate ≥ 4.6 mg/dL). The standardized mean difference was applied to determine each cluster’s key features. We assessed the association of the clusters with mortality. Results: In the hypophosphatemia cohort (n = 3113), the consensus cluster analysis identified two clusters. The key features of patients in Cluster 2, compared with Cluster 1, included: older age; a higher comorbidity burden, particularly hypertension; diabetes mellitus; coronary artery disease; lower eGFR; and more acute kidney injury (AKI) at admission. Cluster 2 had a comparable hospital mortality (3.7% vs. 2.9%; p = 0.17), but a higher one-year mortality (26.8% vs. 14.0%; p < 0.001), and five-year mortality (20.2% vs. 44.3%; p < 0.001), compared to Cluster 1. In the hyperphosphatemia cohort (n = 7252), the analysis identified two clusters. The key features of patients in Cluster 2, compared with Cluster 1, included: older age; more primary admission for kidney disease; more history of hypertension; more end-stage kidney disease; more AKI at admission; and higher admission potassium, magnesium, and phosphate. Cluster 2 had a higher hospital (8.9% vs. 2.4%; p < 0.001) one-year mortality (32.9% vs. 14.8%; p < 0.001), and five-year mortality (24.5% vs. 51.1%; p < 0.001), compared with Cluster 1. Conclusion: Our cluster analysis classified clinically distinct phenotypes with different mortality risks among hospitalized patients with serum phosphate derangements. Age, comorbidities, and kidney function were the key features that differentiated the phenotypes.
Subtyping Hyperchloremia among Hospitalized Patients by Machine Learning Consensus Clustering
Background and Objectives: Despite the association between hyperchloremia and adverse outcomes, mortality risks among patients with hyperchloremia have not consistently been observed among all studies with different patient populations with hyperchloremia. The objective of this study was to characterize hyperchloremic patients at hospital admission into clusters using an unsupervised machine learning approach and to evaluate the mortality risk among these distinct clusters. Materials and Methods: We performed consensus cluster analysis based on demographic information, principal diagnoses, comorbidities, and laboratory data among 11,394 hospitalized adult patients with admission serum chloride of >108 mEq/L. We calculated the standardized mean difference of each variable to identify each cluster’s key features. We assessed the association of each hyperchloremia cluster with hospital and one-year mortality. Results: There were three distinct clusters of patients with admission hyperchloremia: 3237 (28%), 4059 (36%), and 4098 (36%) patients in clusters 1 through 3, respectively. Cluster 1 was characterized by higher serum chloride but lower serum sodium, bicarbonate, hemoglobin, and albumin. Cluster 2 was characterized by younger age, lower comorbidity score, lower serum chloride, and higher estimated glomerular filtration (eGFR), hemoglobin, and albumin. Cluster 3 was characterized by older age, higher comorbidity score, higher serum sodium, potassium, and lower eGFR. Compared with cluster 2, odds ratios for hospital mortality were 3.60 (95% CI 2.33–5.56) for cluster 1, and 4.83 (95% CI 3.21–7.28) for cluster 3, whereas hazard ratios for one-year mortality were 4.49 (95% CI 3.53–5.70) for cluster 1 and 6.96 (95% CI 5.56–8.72) for cluster 3. Conclusions: Our cluster analysis identified three clinically distinct phenotypes with differing mortality risks in hospitalized patients with admission hyperchloremia.
Machine Learning Consensus Clustering of Hospitalized Patients with Admission Hyponatremia
Background: The objective of this study was to characterize patients with hyponatremia at hospital admission into clusters using an unsupervised machine learning approach, and to evaluate the short- and long-term mortality risk among these distinct clusters. Methods: We performed consensus cluster analysis based on demographic information, principal diagnoses, comorbidities, and laboratory data among 11,099 hospitalized adult hyponatremia patients with an admission serum sodium below 135 mEq/L. The standardized mean difference was utilized to identify each cluster’s key features. We assessed the association of each hyponatremia cluster with hospital and one-year mortality using logistic and Cox proportional hazard analysis, respectively. Results: There were three distinct clusters of hyponatremia patients: 2033 (18%) in cluster 1, 3064 (28%) in cluster 2, and 6002 (54%) in cluster 3. Among these three distinct clusters, clusters 3 patients were the youngest, had lowest comorbidity burden, and highest kidney function. Cluster 1 patients were more likely to be admitted for genitourinary disease, and have diabetes and end-stage kidney disease. Cluster 1 patients had the lowest kidney function, serum bicarbonate, and hemoglobin, but highest serum potassium and prevalence of acute kidney injury. In contrast, cluster 2 patients were the oldest and were more likely to be admitted for respiratory disease, have coronary artery disease, congestive heart failure, stroke, and chronic obstructive pulmonary disease. Cluster 2 patients had lowest serum sodium and serum chloride, but highest serum bicarbonate. Cluster 1 patients had the highest hospital mortality and one-year mortality, followed by cluster 2 and cluster 3, respectively. Conclusion: We identified three clinically distinct phenotypes with differing mortality risks in a heterogeneous cohort of hospitalized hyponatremic patients using an unsupervised machine learning approach.