Custom machine learning algorithm for large-scale disease screening - taking heart disease data as an example

Chen，Leran1,2; Ji，Ping3; Ma，Yongsheng1; Rong，Yiming1; Ren，Jingzheng2

2023-12-01

10.1016/j.artmed.2023.102688

Artificial Intelligence in Medicine   影响因子和分区

0933-3657

1873-2860

Heart disease accounts for millions of deaths worldwide annually, representing a major public health concern. Large-scale heart disease screening can yield significant benefits both in terms of lives saved and economic costs. In this study, we introduce a novel algorithm that trains a patient-specific machine learning model, aligning with the real-world demands of extensive disease screening. Customization is achieved by concentrating on three key aspects: data processing, neural network architecture, and loss function formulation. Our approach integrates individual patient data to bolster model accuracy, ensuring dependable disease detection. We assessed our models using two prominent heart disease datasets: the Cleveland dataset and the UC Irvine (UCI) combination dataset. Our models showcased notable results, achieving accuracy and recall rates beyond 95 % for the Cleveland dataset and surpassing 97 % accuracy for the UCI dataset. Moreover, in terms of medical ethics and operability, our approach outperformed traditional, general-purpose machine learning algorithms. Our algorithm provides a powerful tool for large-scale disease screening and has the potential to save lives and reduce the economic burden of heart disease.

Attention Custom model Customized machine learning Data augmentation Disease diagnosis Heart disease Large-scale disease screening Machine learning Parameter optimization

SCI ; EI

英语

第一

WOS:001108626700001

20234414997689

Cardiology ; Data handling ; Diagnosis ; Diseases ; Heart ; Hospital data processing ; Learning algorithms ; Network architecture

Biological Materials and Tissue Engineering:461.2 ; Medicine and Pharmacology:461.6 ; Hospitals, Equipment and Supplies:462.2 ; Data Processing and Image Processing:723.2 ; Artificial Intelligence:723.4 ; Machine Learning:723.4.2

CLINICAL MEDICINE

2-s2.0-85175261240

Scopus

1.Southern University of Science and Technology,Department of Mechanical and Energy Engineering,Shenzhen,China
2.The Hong Kong Polytechnic University,Department of Industrial and Systems Engineering,Hong Kong,Hong Kong
3.Khalifa University,Department of Management Science And Engineering,Abu Dhabi,United Arab Emirates

Chen，Leran,Ji，Ping,Ma，Yongsheng,et al. Custom machine learning algorithm for large-scale disease screening - taking heart disease data as an example[J]. Artificial Intelligence in Medicine,2023,146.

Chen，Leran,Ji，Ping,Ma，Yongsheng,Rong，Yiming,&Ren，Jingzheng.(2023).Custom machine learning algorithm for large-scale disease screening - taking heart disease data as an example.Artificial Intelligence in Medicine,146.

Chen，Leran,et al."Custom machine learning algorithm for large-scale disease screening - taking heart disease data as an example".Artificial Intelligence in Medicine 146(2023).