Heart Disease Prediction Using Hybrid Machine Learning Model Based on Decision Tree and Neural Network

被引:0
|
作者
Bakhshi, Mostafa [1 ]
Mirtaheri, Seyedeh Leili [2 ]
Greco, Sergio [3 ]
机构
[1] Kharazmi Univ, Tehran, Iran
[2] Kharazmi Univ, Dept Elect & Comp Engn, Fac Engn, Tehran, Iran
[3] Univ Calabria, Dept Comp Engn Modeling Elect & Syst DIMES, Arcavacata Di Rende, Italy
关键词
heart disease diagnosis; feature selection; pearson correlation coefficient; basic component analysis; classification;
D O I
10.1109/ISCMI56532.2022.10068473
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Cardiovascular disease is the leading cause of death in the world. Nowadays, tremendous amount of data is collected on heart disease. Investigating the data and obtaining insight using data mining can improve the detection and prevention rate, especially in early stages. So far, many researches are performed on data mining models for diagnoses. In this paper, we intend to present a model for the diagnosis of heart disease using a feature-based approach as a preprocessing step. The proposed solution include four main steps as preprocessing the data, selecting effective features, clustering by using the K-Means algorithm and proposing a hybrid model of decision tree and neural network to determine the disease. In selecting the effective features, we use three methods as Pearson correlation coefficient, information gain, and component analysis. The evaluation results confirm that the proposed hybrid model outperforms the existing methods by 0.97 accuracy.
引用
收藏
页码:36 / 41
页数:6
相关论文
共 50 条
  • [1] Heart Disease Prediction using Hybrid machine Learning Model
    Kavitha, M.
    Gnaneswar, G.
    Dinesh, R.
    Sai, Y. Rohith
    Suraj, R. Sai
    [J]. PROCEEDINGS OF THE 6TH INTERNATIONAL CONFERENCE ON INVENTIVE COMPUTATION TECHNOLOGIES (ICICT 2021), 2021, : 1329 - 1333
  • [2] Using neural network and decision tree for machine reliability prediction
    Kuo, Yiyo
    Lin, Kuo-Ping
    [J]. INTERNATIONAL JOURNAL OF ADVANCED MANUFACTURING TECHNOLOGY, 2010, 50 (9-12): : 1243 - 1251
  • [3] Using neural network and decision tree for machine reliability prediction
    Yiyo Kuo
    Kuo-Ping Lin
    [J]. The International Journal of Advanced Manufacturing Technology, 2010, 50 : 1243 - 1251
  • [4] A hybrid disease prediction model based on decision tree and extreme learning machine for predicting dialysis of diabetic nephropathy
    Chen, I-Fei
    Lee, Tian-Shyug
    Jhou, Mao-Jhen
    Lu, Chi-Jie
    [J]. Journal of Quality, 2020, 27 (04): : 214 - 230
  • [5] MOOC Dropout Prediction Using a Hybrid Algorithm Based on Decision Tree and Extreme Learning Machine
    Chen, Jing
    Feng, Jun
    Sun, Xia
    Wu, Nannan
    Yang, Zhengzheng
    Chen, Sushing
    [J]. MATHEMATICAL PROBLEMS IN ENGINEERING, 2019, 2019
  • [6] Prediction of the coronary heart disease using neural network based model
    Mital, DP
    Haque, S
    Srinivasan, S
    Namboodiri, S
    [J]. METMBS'03: PROCEEDINGS OF THE INTERNATIONAL CONFERENCE ON MATHEMATICS AND ENGINEERING TECHNIQUES IN MEDICINE AND BIOLOGICAL SCIENCES, 2003, : 269 - 275
  • [7] Optimized machine learning model using Decision Tree for cancer prediction
    Chandrasegar, T.
    Vutukuri, Sai Brahma Nikhilesh
    [J]. 2019 INNOVATIONS IN POWER AND ADVANCED COMPUTING TECHNOLOGIES (I-PACT), 2019,
  • [8] Prediction of Heart Disease using Forest Algorithm over Decision Tree using Machine Learning with Improved Accuracy
    Raj, K. N. S. Shanmukha
    Thinakaran, K.
    [J]. CARDIOMETRY, 2022, (25): : 1520 - 1525
  • [9] Prediction of Heart Disease using Decision Tree over Logistic Regression using Machine Learning with Improved Accuracy
    Raj, K. N. S. Shanmukha
    Thinakaran, K.
    [J]. CARDIOMETRY, 2022, (25): : 1514 - 1519
  • [10] Supplier selection: A hybrid model using DEA, decision tree and neural network
    Wu, Desheng
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2009, 36 (05) : 9105 - 9112