DOI QR코드

DOI QR Code

Development and application of prediction model of hyperlipidemia using SVM and meta-learning algorithm

SVM과 meta-learning algorithm을 이용한 고지혈증 유병 예측모형 개발과 활용

  • Lee, Seulki (National Evidence-based Healthcare Collaborating Agency) ;
  • Shin, Taeksoo (Division of Business Administration, College of Government and Business, Yonsei University)
  • 이슬기 (한국보건의료연구원 신의료기술평가사업본부) ;
  • 신택수 (연세대학교 정경대학 경영학부)
  • Received : 2018.04.12
  • Accepted : 2018.06.23
  • Published : 2018.06.30

Abstract

This study aims to develop a classification model for predicting the occurrence of hyperlipidemia, one of the chronic diseases. Prior studies applying data mining techniques for predicting disease can be classified into a model design study for predicting cardiovascular disease and a study comparing disease prediction research results. In the case of foreign literatures, studies predicting cardiovascular disease were predominant in predicting disease using data mining techniques. Although domestic studies were not much different from those of foreign countries, studies focusing on hypertension and diabetes were mainly conducted. Since hypertension and diabetes as well as chronic diseases, hyperlipidemia, are also of high importance, this study selected hyperlipidemia as the disease to be analyzed. We also developed a model for predicting hyperlipidemia using SVM and meta learning algorithms, which are already known to have excellent predictive power. In order to achieve the purpose of this study, we used data set from Korea Health Panel 2012. The Korean Health Panel produces basic data on the level of health expenditure, health level and health behavior, and has conducted an annual survey since 2008. In this study, 1,088 patients with hyperlipidemia were randomly selected from the hospitalized, outpatient, emergency, and chronic disease data of the Korean Health Panel in 2012, and 1,088 nonpatients were also randomly extracted. A total of 2,176 people were selected for the study. Three methods were used to select input variables for predicting hyperlipidemia. First, stepwise method was performed using logistic regression. Among the 17 variables, the categorical variables(except for length of smoking) are expressed as dummy variables, which are assumed to be separate variables on the basis of the reference group, and these variables were analyzed. Six variables (age, BMI, education level, marital status, smoking status, gender) excluding income level and smoking period were selected based on significance level 0.1. Second, C4.5 as a decision tree algorithm is used. The significant input variables were age, smoking status, and education level. Finally, C4.5 as a decision tree algorithm is used. In SVM, the input variables selected by genetic algorithms consisted of 6 variables such as age, marital status, education level, economic activity, smoking period, and physical activity status, and the input variables selected by genetic algorithms in artificial neural network consist of 3 variables such as age, marital status, and education level. Based on the selected parameters, we compared SVM, meta learning algorithm and other prediction models for hyperlipidemia patients, and compared the classification performances using TP rate and precision. The main results of the analysis are as follows. First, the accuracy of the SVM was 88.4% and the accuracy of the artificial neural network was 86.7%. Second, the accuracy of classification models using the selected input variables through stepwise method was slightly higher than that of classification models using the whole variables. Third, the precision of artificial neural network was higher than that of SVM when only three variables as input variables were selected by decision trees. As a result of classification models based on the input variables selected through the genetic algorithm, classification accuracy of SVM was 88.5% and that of artificial neural network was 87.9%. Finally, this study indicated that stacking as the meta learning algorithm proposed in this study, has the best performance when it uses the predicted outputs of SVM and MLP as input variables of SVM, which is a meta classifier. The purpose of this study was to predict hyperlipidemia, one of the representative chronic diseases. To do this, we used SVM and meta-learning algorithms, which is known to have high accuracy. As a result, the accuracy of classification of hyperlipidemia in the stacking as a meta learner was higher than other meta-learning algorithms. However, the predictive performance of the meta-learning algorithm proposed in this study is the same as that of SVM with the best performance (88.6%) among the single models. The limitations of this study are as follows. First, various variable selection methods were tried, but most variables used in the study were categorical dummy variables. In the case with a large number of categorical variables, the results may be different if continuous variables are used because the model can be better suited to categorical variables such as decision trees than general models such as neural networks. Despite these limitations, this study has significance in predicting hyperlipidemia with hybrid models such as met learning algorithms which have not been studied previously. It can be said that the result of improving the model accuracy by applying various variable selection techniques is meaningful. In addition, it is expected that our proposed model will be effective for the prevention and management of hyperlipidemia.

본 연구는 만성질환 중의 하나인 고지혈증 유병을 예측하는 분류모형을 개발하고자 한다. 이를 위해 SVM과 meta-learning 알고리즘을 이용하여 성과를 비교하였다. 또한 각 알고리즘에서 성과를 향상시키기 위해 변수선정 방법을 통해 유의한 변수만을 선정하여 투입하여 분석하였고 이 결과 역시 각각 성과를 비교하였다. 본 연구목적을 달성하기 위해 한국의료패널 2012년 자료를 이용하였고, 변수 선정을 위해 세 가지 방법을 사용하였다. 먼저 단계적 회귀분석(stepwise regression)을 실시하였다. 둘째, 의사결정나무(decision tree) 알고리즘을 사용하였다. 마지막으로 유전자 알고리즘을 사용하여 변수를 선정하였다. 한편, 이렇게 선정된 변수를 기준으로 SVM, meta-learning 알고리즘 등을 이용하여 고지혈증 환자분류 예측모형을 비교하였고, TP rate, precision 등을 사용하여 분류 성과를 비교분석하였다. 이에 대한 분석결과는 다음과 같다. 첫째, 모든 변수를 투입하여 분류한 결과 SVM의 정확도는 88.4%, 인공신경망의 정확도는 86.7%로 SVM의 정확도가 좀 더 높았다. 둘째, stepwise를 통해 선정된 변수만을 투입하여 분류한 결과 전체 변수를 투입하였을 때보다 각각 정확도가 약간 높았다. 셋째, 의사결정나무에 의해 선정된 변수 3개만을 투입하였을 때 인공신경망의 정확도가 SVM보다 높았다. 유전자 알고리즘을 통해 선정된 변수를 투입하여 분류한 결과 SVM은 88.5%, 인공신경망은 87.9%의 분류 정확도를 보여 주었다. 마지막으로, 본 연구에서 제안하는 meta-learning 알고리즘인 스태킹(stacking)을 적용한 결과로서, SVM과 MLP의 예측결과를 메타 분류기인 SVM의 입력변수로 사용하여 예측한 결과, 고지혈증 분류 정확도가 meta-learning 알고리즘 중에서는 가장 높은 것으로 나타났다.

Keywords

References

  1. Chitra, R., & Seenivasagam, V., "Heart disease prediction system using supervised learning classifier," Bonfring International Journal of Software Engineering and Soft Computing. Vol.3, No.1(2013), 1-7. https://doi.org/10.9756/BIJSESC.4336
  2. Dogan, S., Turkoglu, I., "Diagnosing hyperlipidemia using association rules," Mathematical and Computational Applicationas. Vol.13, No.3(2008), 193-202. https://doi.org/10.3390/mca13030193
  3. Huang, M. J., CHen, M. Y., Lee, S. C. "Integrating data mining with case-based reasoning for chronic diseases prognosis and diagnosis," Expert Systems with Application. Vol.32,(2007), 856-867. https://doi.org/10.1016/j.eswa.2006.01.038
  4. Jabbar, M. A., Deekshatulu, B. L., & Chandra, P., "Heart disease prediction system using associative classification and genetic algorithm," arXiv preprint arXiv:1303.5919. (2013)
  5. Jackson, N., "Developing the concept of metalearning," Innovations in Education and Teaching Internal. Vol.41, No.4(2004), 391-403. https://doi.org/10.1080/1470329042000276995
  6. Kohavi, R., John, G.H., "Wrappers for feature subset selection," Artificial Intelligence. Vol.97, No.1(1997), 273-324. https://doi.org/10.1016/S0004-3702(97)00043-X
  7. Kim, M. J., "Ensemble Learning with Support Vector Machines for Bond Rating," Journal of Intelligence and Information Systems, Vol.18, No.2(2012), 29-45. https://doi.org/10.13088/JIIS.2012.18.2.029
  8. Min, S. H, "Bankruptcy prediction using an improved bagging ensemble," Journal of Intelligence and Information Systems, Vol.20, No.4(2014), 121-139. https://doi.org/10.13088/JIIS.2014.20.4.121
  9. OECD(2012). Health Care Quality Review: Korea. OECD.
  10. Palaniappan, S., & Awang, R. "Intelligent Heart Disease Prediction system using data mining techniques," International Journal of Computer Science and Network Security. Vol. 8, No.8(2008), 343-350.
  11. Santhanam, T., Padmavathi, M. S., "Application of K-means and Genetic Algorithms for dimension reduction by intergrating SVM for diabetes diagnosis," Procedia Computer Science, Vol.47, (2015), 76-83. https://doi.org/10.1016/j.procs.2015.03.185
  12. Shin, T. S., Hong, T. H., "Corporate Credit Rating based on Bankruptcy Probability Using AdaBoost Algorithm-based Support Vector Machine," Journal of Intelligence and Information Systems, Vol.17, No.3(2011), 25-41.
  13. Soni, J., Ansari, U., Sharma, D., "Predictive data mining for medical diagnosis: An overview of heart disease prediction," International Journal of Computer Applications. Vol.17, No.8(2011), 43-48. https://doi.org/10.5120/2237-2860
  14. Sudha, A., Gayathri, P., Jaisankar, N., "Utilization of data mining approaches for prediction of life threatening diseases survivability," International Journal of Computer Applications. Vol.41, No.17(2012), 51-55. https://doi.org/10.5120/5637-8023
  15. Tan, P. N., Steinbach, M., & Kumar, V. , Introduction to data mining. Boston: Pearson Addison Wesley.(2005).