Title
An ensemble model with cluster assumption for warfarin dose prediction in Chinese patients.
Abstract
The prediction of daily stable warfarin dosage for a specific patient is difficult. To improve the predictive accuracy and build a highly accurate predictive model, we developed an ensemble learning method, called evolutionary fuzzy c-mean (EFCM) clustering algorithm with support vector regression (SVR). A dataset of 517 Han Chinese patients was collected from the data of The First Affiliated Hospital of Soochow University and dataset of International Warfarin Pharmacogenetics Consortium for training and testing. In EFCM+SVR, we adopted SVR to build a generalized base model (SVR model). To achieve an accurate prediction on patients with large dosage, we proposed an EFCM clustering algorithm that can be used to cluster the training set and designed a clustering model on clusters and centroids. The SVR and clustering models were integrated into an ensemble model by stepwise functions. In the experiment, three artificial neural networks, SVR, two ensemble models and three regression models were used as comparators to the EFCM+SVR model, which obtained the smallest mean absolute error (0.67 mg/day) in warfarin dose prediction and the largest R-squared (43.9%). The model achieved satisfactory prediction in terms of the percentage of patients whose predicted dose of warfarin was within 15% and 20% of the actual stable therapeutic dose (15%-p of 36% and 20%-p of 46.6%).
Year
DOI
Venue
2019
10.1109/JBHI.2019.2891164
IEEE journal of biomedical and health informatics
Keywords
Field
DocType
Warfarin dose prediction,predictive model,ensemble model,cluster assumption,fuzzy clustering,support vector regression
Fuzzy clustering,Pattern recognition,Ensemble forecasting,Computer science,Regression analysis,Fuzzy logic,Support vector machine,Artificial intelligence,Cluster analysis,Artificial neural network,Ensemble learning
Journal
Volume
Issue
ISSN
23
6
2168-2208
Citations 
PageRank 
References 
0
0.34
0
Authors
6
Name
Order
Citations
PageRank
Yanyun Tao101.69
Yenming J. Chen2106.33
Ling Xue311.02
Cheng Xie410.69
Bin Jiang511.02
Yuzhen Zhang658.86