Background With the rapid development of artificial intelligence, prediction of warfarin dose via machine learning has received more and more attention. Since the dose prediction involve both linear and nonlinear problems, traditional machine learning algorithms are ineffective to solve such problems at one time.Objective Based on the characteristics of clinical data of Chinese warfarin patients, an improved stacking ensemble learning can achieve higher prediction accuracy.Methods Information of 641 patients from southern China who had reached a steady state on warfarin was collected, including demographic information, medical history, genotype, and co-medication status. The dataset was randomly divided into a training set (90%) and a test set (10%). The predictive capability is evaluated on a new test set generated by stacking ensemble learning. Additional factors associated with warfarin dose were discovered by feature selection methods.Results A newly proposed heuristic-stacking ensemble learning performs better than traditional-stacking ensemble learning in key metrics such as accuracy of ideal dose (73.44%, 71.88%), mean absolute errors (0.11 mg/day, 0.13 mg/day), root mean square errors (0.18 mg/day, 0.20 mg/day) and R2 (0.87, 0.82).Conclusions The developed heuristic-stacking ensemble learning can satisfactorily predict warfarin dose with high accuracy. A relationship between hypertension, a history of severe preoperative embolism, and warfarin dose is found, which provides a useful reference for the warfarin dose administration in the future.
基金:
This work is supported by the Yunnan Provincial Cardiovascular
Disease ClinicalMedical Center Project (Grant Nos. FZX2019-06-01,
2022YFKY010), a prospective cohort study on the reliability
evaluation and clinical application promotion of artificial
intelligence-assisted warfarin anticoagulation management decisionmaking
system, the National Natural Science Foundation of China
(Grant No. 12174210), and the Innovation Team Project of Ji’nan
(Grant No. 2019GXRC035).
第一作者机构:[1]Fuwai Yunnan Cardiovasc Hosp, Dept Pharm, Kunming, Peoples R China[2]Qilu Univ Technol, Shandong Inst Mech Design & Res, Shandong Acad Sci, Sch Mech Engn, Jinan, Shandong, Peoples R China
共同第一作者:
通讯作者:
推荐引用方式(GB/T 7714):
Wang Mingyuan,Qian Yiyi,Yang Yaodong,et al.Improved stacking ensemble learning based on feature selection to accurately predict warfarin dose[J].FRONTIERS IN CARDIOVASCULAR MEDICINE.2024,10:doi:10.3389/fcvm.2023.1320938.
APA:
Wang, Mingyuan,Qian, Yiyi,Yang, Yaodong,Chen, Haobin&Rao, Wei-Feng.(2024).Improved stacking ensemble learning based on feature selection to accurately predict warfarin dose.FRONTIERS IN CARDIOVASCULAR MEDICINE,10,
MLA:
Wang, Mingyuan,et al."Improved stacking ensemble learning based on feature selection to accurately predict warfarin dose".FRONTIERS IN CARDIOVASCULAR MEDICINE 10.(2024)