Machine learning techniques in cardiac risk assessment

Abstract

Background: The objective of this study was to predict the mortality risk of patients during or shortly after cardiac surgery by using machine learning techniques and their learning abilities from collected data.Methods: The dataset was obtained from Acıbadem Maslak Hospital. Risk factors of the European System for Cardiac Operative Risk Evaluation (EuroSCORE) were used to predict mortality risk. First, Standard EuroSCORE scores of patients were calculated and risk groups were determined, because 30-day follow-up information of patients was not available in the dataset. Models were created with five different machine learning algorithms and two different datasets including age, serum creatinine, left ventricular dysfunction, and pulmonary hypertension were numeric in Dataset 1 and categorical in Dataset 2. Model performance evaluation was performed with 10-fold cross-validation.Results: Data analysis and performance evaluation were performed with R, RStudio and Shiny. C4.5 was selected as the best algorithm for risk prediction (accuracy= 0.989) in Dataset 1. This model indicated that pulmonary hypertension, recent myocardial infarct, surgery on thoracic aorta are the primary three risk factors that affect the mortality risk of patients during or shortly after cardiac surgery. Also, this model is used to develop a dynamic web application which is also accessible from mobile devices (https://elifkartal.shinyapps.io/euSCR/).Conclusion: The C4.5 decision tree model was identified as having the highest performance in Dataset 1 in predicting the mortality risk of patients. Using the numerical values of the risk factors can be useful in increasing the performance of machine learning models. Development of hospital-specific local assessment systems using hospital data, such as the application in this study, would be beneficial for both patients and doctors.

Özet

Amaç: Bu çalışmada amaç; makine öğrenmesi tekniklerini ve bu tekniklerin veriden öğrenme yeteneğini kullanarak kalp ameliyatı sırasında ya da kalp ameliyatı geçirdikten kısa bir süre sonra hastanın mortalite riskini öngörebilmektir. Çalışma planı: Veri seti Acıbadem Maslak Hastanesi'nden temin edildi. European System for Cardiac Operative Risk Evaluation (EuroSCORE) risk faktörleri, mortalite riskini tahmin etmek için kullanıldı. Hastaların 30 günlük takip bilgileri veri setinde mevcut olmadığından çalışmada ilk olarak Standart EuroSCORE puanları hesaplandı ve risk grupları belirlendi. Modeller; beş farklı makine öğrenmesi algoritması ile yaş, serum kreatinin, sol ventrikül disfonksiyonu ve pulmoner hipertansiyonun Dataset 1’de sayısal, Dataset 2’de kategorik olduğu iki farklı veri kümesiyle oluşturuldu. Model performans değerlendirmesi, 10-kat çapraz geçerleme ile yapıldı. Bulgular: Veri analizi ve performans değerlendirmesi R, RStudio ve Shiny ile gerçekleştirildi. C4.5 algoritmasıyla Dataset 1 üzerinde kurulan model risk tahmini için en iyi model olarak seçildi (doğruluk= 0.989). Bu model; pulmoner hipertansiyon, geçirilmiş miyokard enfarktüsü ve torasik aort cerrahisi niteliklerini bir hastanın mortalite riskini etkileyen ilk üç risk faktörü olduğuna işaret etmektedir. Ayrıca, bu model mobil cihazlardan da erişilebilen dinamik bir web uygulaması geliştirmek için kullanıldı (https://elifkartal.shinyapps.io/euSCR/). Sonuç: Bir hastanın mortalite riskini öngörebilmede C4.5 karar ağacı modeli, kullanılan veri seti olan Dataset 1’de en iyi performansa sahip bir model olduğu belirlendi. Risk faktörlerinin sayısal değerlerini kullanmak, makine öğrenmesi modellerinin performansını artırmada yararlı olabilir. Bu çalışmadaki uygulamada olduğu gibi hastaneye özgü veri kullanılarak yerel değerlendirme sistemlerinin geliştirilmesi hem hastalar hem de doktorlar için yararlı olabilecektir.