TY - JOUR
T1 - Custom machine learning algorithm for large-scale disease screening - taking heart disease data as an example
AU - Chen, Leran
AU - Ji, Ping
AU - Ma, Yongsheng
AU - Rong, Yiming
AU - Ren, Jingzheng
N1 - Funding Information:
This work was supported by the Department of Industrial and Systems Engineering , The Hong Kong Polytechnic University, Hong Kong [Grant Number: RK3L ].
Publisher Copyright:
© 2023 Elsevier B.V.
PY - 2023/12
Y1 - 2023/12
N2 - 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.
AB - 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.
KW - Attention
KW - Custom model
KW - Customized machine learning
KW - Data augmentation
KW - Disease diagnosis
KW - Heart disease
KW - Large-scale disease screening
KW - Machine learning
KW - Parameter optimization
UR - http://www.scopus.com/inward/record.url?scp=85175261240&partnerID=8YFLogxK
U2 - 10.1016/j.artmed.2023.102688
DO - 10.1016/j.artmed.2023.102688
M3 - Journal article
C2 - 38042606
AN - SCOPUS:85175261240
SN - 0933-3657
VL - 146
JO - Artificial Intelligence in Medicine
JF - Artificial Intelligence in Medicine
M1 - 102688
ER -