{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,10,8]],"date-time":"2026-10-08T04:33:19Z","timestamp":1791433999963,"version":"4.3.3"},"reference-count":45,"publisher":"MDPI AG","issue":"11","license":[{"start":{"date-parts":[[2025,5,25]],"date-time":"2025-05-25T00:00:00Z","timestamp":1748131200000},"content-version":"vor","delay-in-days":0,"URL":"https:\/\/creativecommons.org\/licenses\/by\/4.0\/"}],"content-domain":{"domain":[],"crossmark-restriction":false},"short-container-title":["JCM"],"abstract":"<jats:p>Background\/Objectives: Atrial fibrillation (AF) is a common arrhythmia linked to increased mortality and significant healthcare burden, especially in the elderly. Direct Oral Anticoagulants (DOACs) are crucial for stroke prevention in AF, offering benefits over traditional vitamin K antagonists. Despite scoring systems like HATCH and CHA2DS2-VASc, their predictive ability for mortality in AF patients is limited. This study aims to use machine learning to predict mortality within six months of hospital discharge in AF patients treated with DOACs. Methods: Using the MIMIC-IV database, data from 6431 AF patients were analyzed. Feature selection was done with the LASSO algorithm. Five machine learning models were built: Logistic Regression, Random Forest, XGBoost, LightGBM, and AdaBoost, using 27 features. The top two models were tested on a separate dataset. SHAP values explained model predictions and feature importance. Results: The best model, LightGBM, achieved an AUC of 0.886, accuracy of 0.862, sensitivity of 0.913, and specificity of 0.859. SHAP values highlighted the importance of length of hospital stay, ICU duration, and comorbidities. The model\u2019s interpretability allows for identifying individual patient risk factors, applicable in clinical practice. Conclusions: This study demonstrates that machine learning models effectively predict mortality in AF patients treated with DOACs, potentially enhancing personalized patient care.<\/jats:p>","DOI":"10.3390\/jcm14113697","type":"journal-article","created":{"date-parts":[[2025,5,25]],"date-time":"2025-05-25T20:26:50Z","timestamp":1748204810000},"page":"3697","update-policy":"https:\/\/doi.org\/10.3390\/mdpi_crossmark_policy","source":"Crossref","is-referenced-by-count":3,"title":["Predicting Mortality in Atrial Fibrillation Patients Treated with Direct Oral Anticoagulants: A Machine Learning Study Based on the MIMIC-IV Database"],"prefix":"10.3390","volume":"14","author":[{"ORCID":"https:\/\/orcid.org\/0000-0001-9812-4607","authenticated-orcid":false,"given":"\u0141ukasz","family":"Ledzi\u0144ski","sequence":"first","affiliation":[{"name":"Department of Cardiology and Clinical Pharmacology, Collegium Medicum in Bydgoszcz, Nicolaus Copernicus University in Toru\u0144, 85-168 Bydgoszcz, Poland"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"ORCID":"https:\/\/orcid.org\/0000-0002-6685-4952","authenticated-orcid":false,"given":"El\u017cbieta","family":"Grze\u015bk","sequence":"additional","affiliation":[{"name":"Department of Pediatrics, Hematology and Oncology, Collegium Medicum in Bydgoszcz, Nicolaus Copernicus University in Toru\u0144, 85-094 Bydgoszcz, Poland"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"ORCID":"https:\/\/orcid.org\/0009-0001-3598-8376","authenticated-orcid":false,"given":"Ma\u0142gorzata","family":"Ledzi\u0144ska","sequence":"additional","affiliation":[{"name":"Department of Rheumatology and Connective Tissue Diseases, Collegium Medicum in Bydgoszcz, Nicolaus Copernicus University in Toru\u0144, 85-168 Bydgoszcz, Poland"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"ORCID":"https:\/\/orcid.org\/0000-0001-6669-5931","authenticated-orcid":false,"given":"Grzegorz","family":"Grze\u015bk","sequence":"additional","affiliation":[{"name":"Department of Cardiology and Clinical Pharmacology, Collegium Medicum in Bydgoszcz, Nicolaus Copernicus University in Toru\u0144, 85-168 Bydgoszcz, Poland"}],"role":[{"vocabulary":"crossref","role":"author"}]}],"member":"1968","published-online":{"date-parts":[[2025,5,25]]},"reference":[{"key":"ref_1","first-page":"929","article-title":"Increased Risk of Stroke and Mortality Following New-Onset Atrial Fibrillation during Hospitalization","volume":"19","author":"Massera","year":"2017","journal-title":"Europace"},{"key":"ref_2","doi-asserted-by":"crossref","first-page":"487","DOI":"10.1016\/j.hrthm.2017.11.025","article-title":"Race and Stroke in an Atrial Fibrillation Inception Cohort: Findings from the Penn Atrial Fibrillation Free Study","volume":"15","author":"Patel","year":"2018","journal-title":"Heart. Rhythm."},{"key":"ref_3","doi-asserted-by":"crossref","first-page":"448.E1","DOI":"10.1016\/j.amjmed.2005.10.057","article-title":"Quality of Life in Patients with Atrial Fibrillation: A Systematic Review","volume":"119","author":"Thrall","year":"2006","journal-title":"Am. J. Med."},{"key":"ref_4","doi-asserted-by":"crossref","first-page":"2370","DOI":"10.1001\/jama.285.18.2370","article-title":"Prevalence of Diagnosed Atrial Fibrillation in Adults","volume":"285","author":"Go","year":"2001","journal-title":"JAMA"},{"key":"ref_5","doi-asserted-by":"crossref","first-page":"1807","DOI":"10.1016\/j.ijcard.2012.12.093","article-title":"Atrial Fibrillation: Profile and Burden of an Evolving Epidemic in the 21st Century","volume":"167","author":"Ball","year":"2013","journal-title":"Int. J. Cardiol."},{"key":"ref_6","doi-asserted-by":"crossref","first-page":"14","DOI":"10.33963\/KP.a2022.0202","article-title":"Prevalence of Atrial Fibrillation in the 65 or over Polish Population. Report of Cross-Sectional NOMED-AF Study","volume":"81","author":"Kalarus","year":"2023","journal-title":"Kardiol. Pol."},{"key":"ref_7","first-page":"1612","article-title":"2021 European Heart Rhythm Association Practical Guide on the Use of Non-Vitamin K Antagonist Oral Anticoagulants in Patients with Atrial Fibrillation","volume":"23","author":"Steffel","year":"2021","journal-title":"EP Eur."},{"key":"ref_8","doi-asserted-by":"crossref","unstructured":"Grze\u015bk, G., Rogowicz, D., Wo\u0142owiec, \u0141., Ratajczak, A., Gilewski, W., Chudzi\u0144ska, M., Sinkiewicz, A., and Banach, J. (2021). The Clinical Significance of Drug\u2013Food Interactions of Direct Oral Anticoagulants. Int. J. Mol. Sci., 22.","DOI":"10.3390\/ijms22168531"},{"key":"ref_9","doi-asserted-by":"crossref","first-page":"446","DOI":"10.5603\/AHP.a2021.0039","article-title":"Therapeutic Monitoring of Direct Oral Anticoagulants\u2014an 8-Year Observational Study","volume":"52","year":"2021","journal-title":"Acta Haematol. Pol."},{"key":"ref_10","doi-asserted-by":"crossref","unstructured":"Ledzi\u0144ski, \u0141., and Grze\u015bk, G. (2023). Artificial Intelligence Technologies in Cardiology. J. Cardiovasc. Dev. Dis., 10.","DOI":"10.3390\/jcdd10050202"},{"key":"ref_11","doi-asserted-by":"crossref","first-page":"842104","DOI":"10.3389\/fpubh.2022.842104","article-title":"Prediction of Atrial Fibrillation in Hospitalized Elderly Patients With Coronary Heart Disease and Type 2 Diabetes Mellitus Using Machine Learning: A Multicenter Retrospective Study","volume":"10","author":"Xu","year":"2022","journal-title":"Front. Public Health"},{"key":"ref_12","doi-asserted-by":"crossref","first-page":"837","DOI":"10.1136\/postgradmedj-2021-140754","article-title":"Real-World Observational Study of Assessment of CHA2DS2-VASc, C2HEST and HAVOC Scores for Atrial Fibrillation among Patients with Rheumatological Disorders: A Nationwide Analysis","volume":"98","author":"Hu","year":"2022","journal-title":"Postgrad. Med. J."},{"key":"ref_13","doi-asserted-by":"crossref","first-page":"214","DOI":"10.1093\/ehjqcco\/qcab028","article-title":"GARFIELD-AF Risk Score for Mortality, Stroke, and Bleeding within 2 Years in Patients with Atrial Fibrillation","volume":"8","author":"Fox","year":"2022","journal-title":"Eur. Heart J. Qual. Care Clin. Outcomes"},{"key":"ref_14","doi-asserted-by":"crossref","first-page":"326","DOI":"10.1136\/postgradmedj-2021-141147","article-title":"Comparisons of HATCH, HAVOC and CHA2DS2-VASc Scores for All-Cause Mortality Prediction in Atrial Fibrillation: A Real-World Evidence Study","volume":"99","author":"Hu","year":"2023","journal-title":"Postgrad. Med. J."},{"key":"ref_15","doi-asserted-by":"crossref","first-page":"941","DOI":"10.33963\/v.phj.101842","article-title":"Development of a Risk Score for Predicting One-Year Mortality in Patients with Atrial Fibrillation Using XGBoost-Assisted Feature Selection","volume":"82","author":"Wang","year":"2024","journal-title":"Pol. Heart J."},{"key":"ref_16","doi-asserted-by":"crossref","first-page":"499","DOI":"10.1186\/s12872-021-02314-w","article-title":"Prediction of All-Cause Mortality in Coronary Artery Disease Patients with Atrial Fibrillation Based on Machine Learning Models","volume":"21","author":"Liu","year":"2021","journal-title":"BMC Cardiovasc. Disord."},{"key":"ref_17","doi-asserted-by":"crossref","first-page":"18925","DOI":"10.1038\/s41598-021-97218-2","article-title":"Risk Prediction of Clinical Adverse Outcomes with Machine Learning in a Cohort of Critically Ill Patients with Atrial Fibrillation","volume":"11","author":"Falsetti","year":"2021","journal-title":"Sci. Rep."},{"key":"ref_18","unstructured":"Alistair, J., Lucas, B., Tom, P., Brian, G., Benjamin, M., Steven, H., Leo, A.C., and Roger, M. (2024). MIMIC-IV, version 3.1, PhysioNet."},{"key":"ref_19","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1038\/s41597-022-01899-x","article-title":"MIMIC-IV, a Freely Accessible Electronic Health Record Dataset","volume":"10","author":"Johnson","year":"2023","journal-title":"Sci. Data"},{"key":"ref_20","doi-asserted-by":"crossref","first-page":"E215","DOI":"10.1161\/01.CIR.101.23.e215","article-title":"PhysioBank, PhysioToolkit, and PhysioNet","volume":"101","author":"Goldberger","year":"2000","journal-title":"Circulation"},{"key":"ref_21","first-page":"115","article-title":"Making a Science of Model Search: Hyperparameter Optimization in Hundreds of Dimensions for Vision Architectures","volume":"Volume 28","author":"Bergstra","year":"2013","journal-title":"Proceedings of the 30th International Conference on Machine Learning"},{"key":"ref_22","doi-asserted-by":"crossref","unstructured":"Chen, T., and Guestrin, C. (2016, January 13\u201317). XGBoost. Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, San Francisco, CA, USA.","DOI":"10.1145\/2939672.2939785"},{"key":"ref_23","unstructured":"Ke, G., Meng, Q., Finley, T., Wang, T., Chen, W., Ma, W., Ye, Q., and Liu, T.-Y. (2017, January 4\u20139). LightGBM: A Highly Efficient Gradient Boosting Decision Tree. Proceedings of the Advances in Neural Information Processing Systems, Long Beach, CA, USA."},{"key":"ref_24","doi-asserted-by":"crossref","first-page":"344","DOI":"10.1186\/s40001-025-02621-4","article-title":"Risk Analysis of the Association between EASIX and All-Cause Mortality in Critical Ill Patients with Atrial Fibrillation: A Retrospective Study from MIMIC-IV Database","volume":"30","author":"Xia","year":"2025","journal-title":"Eur. J. Med. Res"},{"key":"ref_25","doi-asserted-by":"crossref","unstructured":"Morrone, D., Kroep, S., Ricci, F., Renda, G., Patti, G., Kirchhof, P., Chuang, L.-H., van Hout, B., and De Caterina, R. (2020). Mortality Prediction of the CHA2DS2-VASc Score, the HAS-BLED Score, and Their Combination in Anticoagulated Patients with Atrial Fibrillation. J. Clin. Med., 9.","DOI":"10.3390\/jcm9123987"},{"key":"ref_26","unstructured":"Lundberg, S.M., and Lee, S.-I. (2017, January 4\u20139). A Unified Approach to Interpreting Model Predictions. Proceedings of the Advances in Neural Information Processing Systems, Long Beach, CA, USA."},{"key":"ref_27","doi-asserted-by":"crossref","first-page":"56","DOI":"10.1038\/s42256-019-0138-9","article-title":"From Local Explanations to Global Understanding with Explainable AI for Trees","volume":"2","author":"Lundberg","year":"2020","journal-title":"Nat. Mach. Intell."},{"key":"ref_28","doi-asserted-by":"crossref","first-page":"3314","DOI":"10.1093\/eurheartj\/ehae176","article-title":"2024 ESC Guidelines for the Management of Atrial Fibrillation Developed in Collaboration with the European Association for Cardio-Thoracic Surgery (EACTS)","volume":"45","author":"Rienstra","year":"2024","journal-title":"Eur. Heart J."},{"key":"ref_29","doi-asserted-by":"crossref","first-page":"1145","DOI":"10.1093\/eurheartj\/ehv466","article-title":"XANTUS: A Real-World, Prospective, Observational Study of Patients Treated with Rivaroxaban for Stroke Prevention in Atrial Fibrillation","volume":"37","author":"Camm","year":"2016","journal-title":"Eur. Heart J."},{"key":"ref_30","doi-asserted-by":"crossref","first-page":"625","DOI":"10.23736\/S2724-5683.24.06546-3","article-title":"Italian Registry in the Setting of Atrial Fibrillation Ablation with Rivaroxaban-IRIS","volume":"72","author":"Lavalle","year":"2024","journal-title":"Minerva Cardiol. Angiol."},{"key":"ref_31","doi-asserted-by":"crossref","first-page":"i3189","DOI":"10.1136\/bmj.i3189","article-title":"Comparative Effectiveness and Safety of Non-Vitamin K Antagonist Oral Anticoagulants and Warfarin in Patients with Atrial Fibrillation: Propensity Weighted Nationwide Cohort Study","volume":"353","author":"Larsen","year":"2016","journal-title":"BMJ"},{"key":"ref_32","doi-asserted-by":"crossref","first-page":"105585","DOI":"10.1016\/j.ijmedinf.2024.105585","article-title":"A Machine Learning-Based Predictive Model for the in-Hospital Mortality of Critically Ill Patients with Atrial Fibrillation","volume":"191","author":"Luo","year":"2024","journal-title":"Int. J. Med. Inform."},{"key":"ref_33","doi-asserted-by":"crossref","first-page":"101372","DOI":"10.1016\/j.ijcha.2024.101372","article-title":"Association between Inflammation Markers and All-Cause Mortality in Critical Ill Patients with Atrial Fibrillation: Analysis of the Multi-Parameter Intelligent Monitoring in Intensive Care (MIMIC-IV) Database","volume":"51","author":"Li","year":"2024","journal-title":"IJC Heart Vasc."},{"key":"ref_34","doi-asserted-by":"crossref","first-page":"64","DOI":"10.1186\/s12933-024-02153-x","article-title":"The Association between Triglyceride Glucose-Body Mass Index and All-Cause Mortality in Critically Ill Patients with Atrial Fibrillation: A Retrospective Study from MIMIC-IV Database","volume":"23","author":"Hu","year":"2024","journal-title":"Cardiovasc. Diabetol."},{"key":"ref_35","doi-asserted-by":"crossref","first-page":"160035","DOI":"10.1038\/sdata.2016.35","article-title":"MIMIC-III, a Freely Accessible Critical Care Database","volume":"3","author":"Johnson","year":"2016","journal-title":"Sci. Data"},{"key":"ref_36","doi-asserted-by":"crossref","first-page":"1562","DOI":"10.5603\/KP.a2018.0194","article-title":"Adherence in Patients with Atrial Fibrillation Treated with Dabigatran","volume":"76","author":"Janiszewska","year":"2018","journal-title":"Kardiol. Pol."},{"key":"ref_37","doi-asserted-by":"crossref","unstructured":"Wo\u0142owiec, \u0141., Kusiak, M., Budzy\u0144ski, J., Wo\u0142owiec, A., Ja\u015bniak, A., Wici\u0144ski, M., Pedrycz-Wieczorska, A., Rogowicz, D., and Grze\u015bk, G. (2023). Therapeutic Drug Monitoring of Direct Oral Anticoagulants in Patients with Extremely Low and High Body Weight\u2014Pilot Study. J. Clin. Med., 12.","DOI":"10.3390\/jcm12154969"},{"key":"ref_38","doi-asserted-by":"crossref","first-page":"1309","DOI":"10.1177\/1060028021995201","article-title":"The Impact of Body Weight and Renal Function on the Risk of Bleeding with Direct Oral Anticoagulants in Atrial Fibrillation","volume":"55","author":"Whittemore","year":"2021","journal-title":"Ann. Pharmacother."},{"key":"ref_39","doi-asserted-by":"crossref","first-page":"150","DOI":"10.1186\/s12872-023-03179-x","article-title":"Association between Obesity and Short- and Medium-Term Mortality in Critically Ill Patients with Atrial Fibrillation: A Retrospective Cohort Study","volume":"23","author":"Yang","year":"2023","journal-title":"BMC Cardiovasc. Disord."},{"key":"ref_40","doi-asserted-by":"crossref","first-page":"15031","DOI":"10.1038\/s41598-023-41544-0","article-title":"Improved Patient Mortality Predictions in Emergency Departments with Deep Learning Data-Synthesis and Ensemble Models","volume":"13","author":"Son","year":"2023","journal-title":"Sci. Rep."},{"key":"ref_41","doi-asserted-by":"crossref","unstructured":"Hu, T., Chao, C., Wu, C., Chien, T., and Li, C. (2024). Machine Learning-Based Predictions of Mortality and Readmission in Type 2 Diabetes Patients in the ICU. Appl. Sci., 14.","DOI":"10.3390\/app14188443"},{"key":"ref_42","doi-asserted-by":"crossref","first-page":"5223","DOI":"10.1038\/s41598-023-32160-z","article-title":"Machine Learning Algorithm to Predict Mortality in Critically Ill Patients with Sepsis-Associated Acute Kidney Injury","volume":"13","author":"Li","year":"2023","journal-title":"Sci. Rep."},{"key":"ref_43","doi-asserted-by":"crossref","first-page":"129","DOI":"10.1186\/s12911-023-02237-w","article-title":"Machine Learning-Based Mortality Prediction Models for Smoker COVID-19 Patients","volume":"23","author":"Nahvijou","year":"2023","journal-title":"BMC Med. Inform. Decis. Mak."},{"key":"ref_44","doi-asserted-by":"crossref","first-page":"228","DOI":"10.1186\/s12911-024-02630-z","article-title":"Prediction of Sepsis Mortality in ICU Patients Using Machine Learning Methods","volume":"24","author":"Gao","year":"2024","journal-title":"BMC Med. Inform. Decis. Mak."},{"key":"ref_45","doi-asserted-by":"crossref","first-page":"400","DOI":"10.1097\/SHK.0000000000002078","article-title":"DEVELOPMENT OF A MACHINE LEARNING MODEL FOR PREDICTING 28-DAY MORTALITY OF SEPTIC PATIENTS WITH ATRIAL FIBRILLATION","volume":"59","author":"Wang","year":"2023","journal-title":"Shock"}],"container-title":["Journal of Clinical Medicine"],"original-title":[],"language":"en","link":[{"URL":"https:\/\/www.mdpi.com\/2077-0383\/14\/11\/3697\/pdf","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2025,10,9]],"date-time":"2025-10-09T17:40:03Z","timestamp":1760031603000},"score":1,"resource":{"primary":{"URL":"https:\/\/www.mdpi.com\/2077-0383\/14\/11\/3697"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2025,5,25]]},"references-count":45,"journal-issue":{"issue":"11","published-online":{"date-parts":[[2025,6]]}},"alternative-id":["jcm14113697"],"URL":"https:\/\/doi.org\/10.3390\/jcm14113697","relation":{},"ISSN":["2077-0383"],"issn-type":[{"value":"2077-0383","type":"electronic"}],"subject":[],"published":{"date-parts":[[2025,5,25]]}}}