A novel method for disease prediction: Hybrid of random forest and multivariate adaptive regression splines

作者：Yao, Dengju^*; Yang, Jing; Zhan, Xiaojuan

来源：Journal of Computers, 2013, 8(1): 170-177.

DOI：10.4304/jcp.8.1.170-177

• 10-fold cross-validation
• Clustering analysis
• Data mining technology
• Medical data
• Multivariate adaptive regression splines
• Performance measurements
• Random forest algorithm
• Random forests

Using data mining technology for disease prediction and diagnosis has become the focus of attention. Data mining technology provides an important means for extracting valuable medical rules hidden in medical data and acts as an important role in disease prediction and clinical diagnosis. This paper surveys some kind of popular data mining techniques for disease prediction and diagnosis, such as decision tree, associated rule analysis and clustering analysis. Then, a novel hybrid method of random forest and multivariate adaptive regression splines is proposed for building disease prediction model. Firstly, random forest algorithm is used to perform a preliminary screening of variables and to gain an importance ranks. Then, the new dataset selected by top-k important predictors is input into the MARS procedure, which is responsible for building interpretable models for predicting disease survivability. The capability of this combination method is evaluated using basic performance measurements (e.g., accuracy, sensitivity, and specificity) along with a 10-fold cross validation. Experimental results show that the proposed method provides a higher accuracy and a relatively simple model.

• 出版日期2013
• 单位哈尔滨工程大学