Custom machine learning algorithm for large-scale disease screening - taking heart disease data as an example.
Attention
Custom model
Customized machine learning
Data augmentation
Disease diagnosis
Heart disease
Large-scale disease screening
Machine learning
Parameter optimization
Journal
Artificial intelligence in medicine
ISSN: 1873-2860
Titre abrégé: Artif Intell Med
Pays: Netherlands
ID NLM: 8915031
Informations de publication
Date de publication:
Dec 2023
Dec 2023
Historique:
received:
20
03
2023
revised:
09
10
2023
accepted:
13
10
2023
medline:
4
12
2023
pubmed:
3
12
2023
entrez:
2
12
2023
Statut:
ppublish
Résumé
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.
Identifiants
pubmed: 38042606
pii: S0933-3657(23)00202-6
doi: 10.1016/j.artmed.2023.102688
pii:
doi:
Types de publication
Journal Article
Langues
eng
Sous-ensembles de citation
IM
Pagination
102688Informations de copyright
Copyright © 2023 Elsevier B.V. All rights reserved.
Déclaration de conflit d'intérêts
Declaration of competing interest The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.