Revolutionizing Cardiac Risk Assessment: AI-Powered Patient Segmentation Using Advanced Machine Learning Techniques

Joan D. Gonzalez-Franco, Alejandro Galaviz-Mosqueda, Salvador Villarreal-Reyes, Jose E. Lozano-Rizk, Raul Rivera-Rodriguez, Jose E. Gonzalez-Trejo, Alexei-Fedorovish Licea-Navarro, Jorge Lozoya-Arandia, Edgar A. Ibarra-Flores

Cardiovascular diseases stand as the leading cause of mortality worldwide, underscoring the urgent need for effective tools that enable early detection and monitoring of at-risk patients. This study combines Artificial Intelligence (AI) techniques—specifically the k-means clustering algorithm—alongside dimensionality reduction methods like Principal Component Analysis (PCA) and Uniform Manifold Approximation and Projection (UMAP) to identify patient groups with varying levels of heart attack risk. We used a publicly available clinical dataset with 1319 patient records, which included variables such as age, gender, blood pressure, glucose levels, CK-MB Creatine Kinase MB (KCM), and troponin levels. We normalized and prepared the data, then we employed PCA and UMAP to reduce dimensionality and facilitate visualization. Using the k-means algorithm, we segmented the patients into distinct groups based on their clinical features. Our analysis revealed two distinct patient groups. Group 2 exhibited significantly higher levels of troponin (mean 0.4761 ng/mL), KCM (18.65 ng/mL), and glucose (mean 148.19 mg/dL) and was predominantly composed of men (97%). These factors indicate an increased risk of cardiac events compared to Group 1, which had lower levels of these biomarkers and a slightly higher average age. Interestingly, no significant differences in blood pressure were observed between the groups. This study demonstrates the effectiveness of combining Machine Learning (ML) techniques with dimensionality reduction methods to enhance risk stratification accuracy in cardiology. By enabling more targeted interventions for high-risk patients, our unsupervised segmentation approach focuses on intrinsic data patterns rather than predefined diagnostic labels, serves as a powerful complement to traditional risk assessment tools.