Background: An abnormal variation in blood electrolytes, such as potassium, contributes to mortality in children admitted to intensive care units. Continuous and real-time monitoring of potassium serum levels can prevent fatal arrhythmias, but this is not currently practical. The study aims to use machine learning to estimate blood potassium levels with accuracy in real time noninvasively.
Methods: Hospitalized patients in the Pediatric Department of the Rajaie Cardiology and Medical Research Center and Tehran Heart Center were recruited from December 2021 to June 2022. The electrocardiographic (ECG) features of patients were evaluated. We defined 16 features for each signal and extracted them automatically. The dimension reduction operation was performed with the assistance of the correlation matrix. Linear regression, polynomials, decision trees, random forests, and support vector machine algorithms have been used to find the relationship between characteristics and serum potassium levels. Finally, we used a scatter plot and mean square error (MSE) to display the results.
Results: Of 463 patients (mean age: 8 ± 1 year; 56% boys) hospitalized, 428 patients met the inclusion criteria, with 35 patients having a high noise of ECG were excluded. After the dimension reduction step, 11 features were selected from each cardiac signal. The random forest regression algorithm showed the best performance with an MSE of 0.3.
Conclusion: The accurate estimation of serum potassium levels based on ECG signals is possible using machine learning algorithms. This can be potentially useful in predicting serum potassium levels in specific clinical scenarios.