{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,3,4]],"date-time":"2026-03-04T17:04:48Z","timestamp":1772643888726,"version":"3.50.1"},"reference-count":43,"publisher":"MDPI AG","issue":"5","license":[{"start":{"date-parts":[[2021,5,1]],"date-time":"2021-05-01T00:00:00Z","timestamp":1619827200000},"content-version":"vor","delay-in-days":0,"URL":"https:\/\/creativecommons.org\/licenses\/by\/4.0\/"}],"content-domain":{"domain":[],"crossmark-restriction":false},"short-container-title":["Computers"],"abstract":"<jats:p>Atrial fibrillation (AF) and ventricular arrhythmia (Arr) are among the most common and fatal cardiac arrhythmias in the world. Electrocardiogram (ECG) data, collected as part of the UK Biobank, represents an opportunity for analysis and classification of these two diseases in the UK. The main objective of our study is to investigate a two-stage model for the classification of individuals with AF and Arr in the UK Biobank dataset. The current literature addresses heart arrhythmia classification very extensively. However, the data used by most researchers lack enough instances of these common diseases. Moreover, by proposing the two-stage model and separation of normal and abnormal cases, we have improved the performance of the classifiers in detection of each specific disease. Our approach consists of two stages of classification. In the first stage, features of the ECG input are classified into two main classes: normal and abnormal. At the second stage, the features of the ECG are further categorised as abnormal and further classified into two diseases of AF and Arr. A diverse set of ECG features such as the QRS duration, PR interval and RR interval, as well as covariates such as sex, BMI, age and other factors, are used in the modelling process. For both stages, we use the XGBoost Classifier algorithm. The healthy population present in the data, has been undersampled to tackle the class imbalance present in the data. This technique has been applied and evaluated using an ECG dataset from the UKBioBank ECG taken at rest repository. The main results of our paper are as follows: The classification performance for the proposed approach has been measured using F1 score, Sensitivity (Recall) and Specificity (Precision). The results of the proposed system are 87.22%, 88.55% and 85.95%, for average F1 Score, average sensitivity and average specificity, respectively. Contribution and significance: The performance level indicates that automatic detection of AF and Arr in participants present in the UK Biobank is more precise and efficient if done in a two-stage manner. Automatic detection and classification of AF and Arr individuals this way would mean early diagnosis and prevention of more serious consequences later in their lives.<\/jats:p>","DOI":"10.3390\/computers10050060","type":"journal-article","created":{"date-parts":[[2021,5,1]],"date-time":"2021-05-01T21:35:39Z","timestamp":1619904939000},"page":"60","update-policy":"https:\/\/doi.org\/10.3390\/mdpi_crossmark_policy","source":"Crossref","is-referenced-by-count":17,"title":["A Novel Two-Stage Heart Arrhythmia Ensemble Classifier"],"prefix":"10.3390","volume":"10","author":[{"given":"Mercedeh J.","family":"Rezaei","sequence":"first","affiliation":[{"name":"School of Electronic Engineering Furthermore, Computer Science, Queen Mary University of London, London E1 4NS, UK"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"given":"John R.","family":"Woodward","sequence":"additional","affiliation":[{"name":"School of Electronic Engineering Furthermore, Computer Science, Queen Mary University of London, London E1 4NS, UK"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"ORCID":"https:\/\/orcid.org\/0000-0003-4130-5866","authenticated-orcid":false,"given":"Julia","family":"Ram\u00edrez","sequence":"additional","affiliation":[{"name":"William Harvery Research Institute, Barts, Queen Mary University of London, London EC1M 6BQ, UK"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"ORCID":"https:\/\/orcid.org\/0000-0002-4176-2947","authenticated-orcid":false,"given":"Patricia","family":"Munroe","sequence":"additional","affiliation":[{"name":"William Harvery Research Institute, Barts, Queen Mary University of London, London EC1M 6BQ, UK"}],"role":[{"role":"author","vocabulary":"crossref"}]}],"member":"1968","published-online":{"date-parts":[[2021,5,1]]},"reference":[{"key":"ref_1","unstructured":"(2021, April 22). CVD statistics British Heart Foundation. Available online: https:\/\/www.bhf.org.uk."},{"key":"ref_2","first-page":"427","article-title":"Cardiac Arrhythmia Disease Classification Using LSTM Deep Learning Approach","volume":"67","year":"2021","journal-title":"Comput. Mater. Contin."},{"key":"ref_3","doi-asserted-by":"crossref","first-page":"116","DOI":"10.1007\/s10278-020-00394-2","article-title":"Computer-Aided Diagnosis Scheme for Distinguishing Between Benign and Malignant Masses on Breast DCE-MRI Images Using Deep Convolutional Neural Network with Bayesian Optimization","volume":"34","author":"Hizukuri","year":"2020","journal-title":"J. Digit. Imaging"},{"key":"ref_4","unstructured":"(2021, April 23). UK Biobank ECG at Rest Repository. Available online: https:\/\/biobank.ndph.ox.ac.uk\/ukb\/field.cgi?id=20205."},{"key":"ref_5","doi-asserted-by":"crossref","first-page":"101875","DOI":"10.1016\/j.bspc.2020.101875","article-title":"Cardiac arrhythmia classification using tunable Q-wavelet transform based features and support vector machine classifier","volume":"59","author":"Jha","year":"2020","journal-title":"Biomed. Signal Process. Control"},{"key":"ref_6","doi-asserted-by":"crossref","unstructured":"Ram\u00edrez, E., Melin, P., and Prado-Arechiga, G. (2019). Hybrid Model Based on Neural Networks and Fuzzy Logic for 2-Lead Cardiac Arrhythmia Classification. Hybrid Intelligent Systems in Control, Pattern Recognition and Medicine Studies in Computational Intelligence, Springer.","DOI":"10.1007\/978-3-030-34135-0_14"},{"key":"ref_7","doi-asserted-by":"crossref","first-page":"551","DOI":"10.1007\/s11633-019-1219-2","article-title":"Automatic Classification of Cardiac Arrhythmias Based on Hybrid Features and Decision Tree Algorithm","volume":"17","author":"Sahoo","year":"2020","journal-title":"Int. J. Autom. Comput."},{"key":"ref_8","doi-asserted-by":"crossref","unstructured":"Swetha, R., and Ramakrishnan, S. (2021, January 19\u201320). k-means Clustering Optimized Fuzzy Logic Control Algorithm for Arrhythmia Classification. Proceedings of the 2021 International Conference on Advances in Electrical, Computing, Communication and Sustainable Technologies (ICAECT), Bhilai, India.","DOI":"10.1109\/ICAECT49130.2021.9392494"},{"key":"ref_9","doi-asserted-by":"crossref","first-page":"025001","DOI":"10.1088\/1361-6579\/ab6f9a","article-title":"Identification of patients with atrial fibrillation: A big data exploratory analysis of the UK Biobank","volume":"41","author":"Oster","year":"2020","journal-title":"Physiol. Meas."},{"key":"ref_10","doi-asserted-by":"crossref","first-page":"101819","DOI":"10.1016\/j.bspc.2019.101819","article-title":"Automated arrhythmia classification based on a combination network of CNN and LSTM","volume":"57","author":"Chen","year":"2020","journal-title":"Biomed. Signal Process. Control"},{"key":"ref_11","doi-asserted-by":"crossref","first-page":"231","DOI":"10.1016\/j.ins.2019.02.065","article-title":"Computer Aided detection for fibrillations and flutters using deep convolutional neural network","volume":"486","author":"Fujita","year":"2019","journal-title":"Inf. Sci."},{"key":"ref_12","doi-asserted-by":"crossref","unstructured":"Nishio, M., Sugiyama, O., Yakami, M., Ueno, S., Kubo, T., Kuroda, T., and Togashi, K. (2018). Computer-aided diagnosis of lung nodule classification between benign nodule, primary lung cancer, and metastatic lung cancer at different image size using deep convolutional neural network with transfer learning. PLoS ONE, 13.","DOI":"10.1371\/journal.pone.0200721"},{"key":"ref_13","doi-asserted-by":"crossref","first-page":"105584","DOI":"10.1016\/j.cmpb.2020.105584","article-title":"Evaluation of deep learning detection and classification towards computer-aided diagnosis of breast lesions in digital X-ray mammograms","volume":"196","author":"Han","year":"2020","journal-title":"Comput. Methods Programs Biomed."},{"key":"ref_14","doi-asserted-by":"crossref","unstructured":"Komeda, Y., Handa, H., Matsui, R., Kashida, H., Watanabe, T., Sakurai, T., and Kudo, M. (2019). Computer-Aided Diagnosis (Cad) Based On Convolutional Neural Network (Cnn) System Using Artificial Intelligence (Ai) For Colorectal Polyp Classification. ESGE Days 2019.","DOI":"10.1055\/s-0039-1681181"},{"key":"ref_15","doi-asserted-by":"crossref","first-page":"389","DOI":"10.1016\/j.compbiomed.2017.08.022","article-title":"A deep convolutional neural network model to classify heartbeats","volume":"89","author":"Acharya","year":"2017","journal-title":"Comput. Biol. Med."},{"key":"ref_16","doi-asserted-by":"crossref","first-page":"37","DOI":"10.1016\/j.compbiomed.2010.11.003","article-title":"A multi-stage automatic arrhythmia recognition and classification system","volume":"41","author":"Kutlu","year":"2011","journal-title":"Comput. Biol. Med."},{"key":"ref_17","doi-asserted-by":"crossref","unstructured":"Manju, B.R., and Nair, A.R. (2019, January 13\u201314). Classification of Cardiac Arrhythmia of 12 Lead ECG Using Combination of SMOTEENN, XGBoost and Machine Learning Algorithms. Proceedings of the 2019 9th International Symposium on Embedded Computing and System Design (ISED), Kollam, India.","DOI":"10.1109\/ISED48680.2019.9096244"},{"key":"ref_18","doi-asserted-by":"crossref","first-page":"2898","DOI":"10.1038\/s41598-020-59821-7","article-title":"Optimal Multi-Stage Arrhythmia Classification Approach","volume":"10","author":"Zheng","year":"2020","journal-title":"Sci. Rep."},{"key":"ref_19","doi-asserted-by":"crossref","unstructured":"Rambhia, J., Naik, A., and Nagori, N. (2019). Arrhythmia Detection and Classification by Neural Network Using ECG Features. SSRN Electron. J.","DOI":"10.2139\/ssrn.3367675"},{"key":"ref_20","doi-asserted-by":"crossref","first-page":"e53","DOI":"10.1016\/j.resuscitation.2018.07.096","article-title":"A machine learning approach for detecting ventricular fibrillation during out-of-hospital cardiac arrest","volume":"130","author":"Alonso","year":"2018","journal-title":"Resuscitation"},{"key":"ref_21","doi-asserted-by":"crossref","first-page":"192","DOI":"10.1016\/j.swevo.2017.10.002","article-title":"Novel genetic ensembles of classifiers applied to myocardium dysfunction recognition based on ECG signals","volume":"39","year":"2018","journal-title":"Swarm Evol. Comput."},{"key":"ref_22","doi-asserted-by":"crossref","unstructured":"Trapeznikov, K., Saligrama, V., and Castanon, D. (2012). Multi-Stage Classifier Design. Mach. Learn., 92.","DOI":"10.1007\/s10994-013-5349-4"},{"key":"ref_23","doi-asserted-by":"crossref","unstructured":"Liu, X., Si, Y., and Wang, D. (2019). LSTM Neural Network for Beat Classification in ECG Identity Recognition. Intell. Autom. Soft Comput.","DOI":"10.31209\/2019.100000104"},{"key":"ref_24","first-page":"35","article-title":"Developing Resource Efficient Heart Arrhythmia Classifier","volume":"109","author":"Jan","year":"2015","journal-title":"Int. J. Comput. Appl."},{"key":"ref_25","unstructured":"(2021, April 23). Eoghan Keany, What makes \u201cXGBoost\u201d so Extreme?. Available online: https:\/\/https:\/\/medium.com\/analytics-vidhya\/what-makes-xgboost-so-extreme-e1544a4433bb."},{"key":"ref_26","unstructured":"Fordham, S. (2021, April 23). How to Use the SimpleImputer Class in Machine Learning with Python. Available online: https:\/\/towardsdatascience.com\/how-to-use-the-simpleimputer-class-in-machine-learning-with-python-10b321c94861."},{"key":"ref_27","unstructured":"Kim, H. (2021, April 23). Isolation Forest Step by Step. Available online: https:\/\/hyunsukim-9320.medium.com\/isolation-forest-step-by-step-341b82923168."},{"key":"ref_28","unstructured":"Morde, V. (2021, April 23). XGBoost Algorithm: Long May She Reign!. Available online: https:\/\/towardsdatascience.com\/https-medium-com-vishalmorde-XGBoost-algorithm-long-she-may-rein-edd9f99be63d."},{"key":"ref_29","doi-asserted-by":"crossref","unstructured":"Bao, J. (2020, January 1\u20132). Multi-features Based Arrhythmia Diagnosis Algorithm Using Xgboost. Proceedings of the 2020 International Conference on Computing and Data Science (CDS), Stanford, CA, USA.","DOI":"10.1109\/CDS49703.2020.00095"},{"key":"ref_30","unstructured":"Brownlee, J. (2021, April 22). Hyperparameter Optimization With Random Search and Grid Search. Available online: https:\/\/machinelearningmastery.com\/hyperparameter-optimization-with-random-search-and-grid-search\/."},{"key":"ref_31","unstructured":"Brownlee, J. (2021, April 22). A Gentle Introduction to k-Fold Cross-Validation. Available online: https:\/\/machinelearningmastery.com\/k-fold-cross-validation\/."},{"key":"ref_32","doi-asserted-by":"crossref","unstructured":"Singh, V., Tewary, S., Sardana, V., and Sardana, H.K. (2019, January 29\u201331). Arrhythmia Detection\u2014A Machine Learning based Comparative Analysis with MIT-BIH ECG Data. Proceedings of the 2019 IEEE 5th International Conference for Convergence in Technology (I2CT), Bombay, India.","DOI":"10.1109\/I2CT45611.2019.9033665"},{"key":"ref_33","doi-asserted-by":"crossref","unstructured":"Apandi, Z.F.M., Ikeura, R., and Hayakawa, S. (2018, January 15\u201317). Arrhythmia Detection Using MIT-BIH Dataset: A Review. Proceedings of the 2018 International Conference on Computational Approach in Smart Systems Design and Applications (ICASSDA), Kuching, Malaysia.","DOI":"10.1109\/ICASSDA.2018.8477620"},{"key":"ref_34","doi-asserted-by":"crossref","unstructured":"Bulbul, H.I., Usta, N., and Yildiz, M. (2017, January 18\u201321). Classification of ECG Arrhythmia with Machine Learning Techniques. Proceedings of the 2017 16th IEEE International Conference on Machine Learning and Applications (ICMLA), Cancun, Mexico.","DOI":"10.1109\/ICMLA.2017.0-104"},{"key":"ref_35","first-page":"9159158","article-title":"Automated Classification of Atrial Fibrillation Using Artificial Neural Network for Wearable Devices","volume":"2020","author":"Ma","year":"2020","journal-title":"Math. Probl. Eng."},{"key":"ref_36","doi-asserted-by":"crossref","unstructured":"Islam, S., Ammour, N., and Alajlan, N. (2017, January 21\u201323). Atrial fibrillation detection with multiparametric RR interval feature and machine learning technique. Proceedings of the 2017 International Conference on Informatics, Health & Technology (ICIHT), Riyadh, Saudi Arabia.","DOI":"10.1109\/ICIHT.2017.7899003"},{"key":"ref_37","doi-asserted-by":"crossref","first-page":"76","DOI":"10.1016\/j.protcy.2013.12.339","article-title":"Cardiac Arrhythmia Classification Using Neural Networks with Selected Features","volume":"10","author":"Mitra","year":"2013","journal-title":"Procedia Technol."},{"key":"ref_38","doi-asserted-by":"crossref","unstructured":"Rajagopal, R. (2019). Automated arrhythmia classification for monitoring cardiac patients using machine learning techniques. Classification Techniques for Medical Image Analysis and Computer Aided Diagnosis, Academic Press.","DOI":"10.1016\/B978-0-12-818004-4.00007-8"},{"key":"ref_39","doi-asserted-by":"crossref","first-page":"7310496","DOI":"10.1155\/2018\/7310496","article-title":"Multiclass Classification of Cardiac Arrhythmia Using Improved Feature Selection and SVM Invariants","volume":"2018","author":"Mustaqeem","year":"2018","journal-title":"Comput. Math. Methods Med."},{"key":"ref_40","unstructured":"Deep, K., Nagar, A., Pant, M., and Bansal, J.C. (2012). ECG Arrhythmia Classification Using Spearman Rank Correlation and Support Vector Machine. Proceedings of the International Conference on Soft Computing for Problem Solving (SocProS 2011) December 20\u201322, 2011, Springer."},{"key":"ref_41","doi-asserted-by":"crossref","unstructured":"Alarsan, F.I., and Younes, M. (2019). Analysis and classification of heart diseases using heartbeat features and machine learning algorithms. J. Big Data, 6.","DOI":"10.1186\/s40537-019-0244-x"},{"key":"ref_42","unstructured":"Dua, D., and Karra Taniskidou, E. (2021, April 21). UCI Machine Learning Repository. Available online: http:\/\/archive.ics.uci.edu\/ml."},{"key":"ref_43","doi-asserted-by":"crossref","unstructured":"Hill, N.R., Ayoubkhani, D., Mcewan, P., Sugrue, D.M., Farooqui, U., Lister, S., Lumley, M., Bakhai, A., Cohen, A.T., and O\u2019Neill, M. (2019). Predicting atrial fibrillation in primary care using machine learning. PLoS ONE, 14.","DOI":"10.1371\/journal.pone.0224582"}],"container-title":["Computers"],"original-title":[],"language":"en","link":[{"URL":"https:\/\/www.mdpi.com\/2073-431X\/10\/5\/60\/pdf","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2025,10,11]],"date-time":"2025-10-11T05:56:28Z","timestamp":1760162188000},"score":1,"resource":{"primary":{"URL":"https:\/\/www.mdpi.com\/2073-431X\/10\/5\/60"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2021,5,1]]},"references-count":43,"journal-issue":{"issue":"5","published-online":{"date-parts":[[2021,5]]}},"alternative-id":["computers10050060"],"URL":"https:\/\/doi.org\/10.3390\/computers10050060","relation":{},"ISSN":["2073-431X"],"issn-type":[{"value":"2073-431X","type":"electronic"}],"subject":[],"published":{"date-parts":[[2021,5,1]]}}}