The impact of Negative to Positive Training Dataset Ratio on Atrial Fibrillation Classification Machine Learning Algorithms Performance

Abstract

With the few numbers of cardiologists in Indonesia who not evenly distributed, especially in rural areas, there has been a lot of smart telehealth specifically developed for heart monitoring using ECG. Many techniques have been developed to improve the accuracy of this device by using datasets that are mostly imbalanced, more positive data than negative. This paper presents the comparison of negative to positive training dataset ratio on atrial fibrillation classification machine learning algorithms performance. An AliveCor ECG recording dataset is train with deep neural networks, support vector machine and logistic regression as classifier with three different ratios, 1:1, 1:5 to 1:All. Results show an increase in classifier performance along with the increasing number of negative data.