%0 Journal Article %T 基于改进DeepFM的心脏病预测应用研究
Applied Research on Heart Disease Prediction Based on Improved DeepFM %A 张笑 %A 李宁 %J Computer Science and Application %P 2117-2125 %@ 2161-881X %D 2021 %I Hans Publishing %R 10.12677/CSA.2021.118217 %X
近年来,心脏病在全球已严重威胁到人类的身体和生命健康安全,通过利用人工智能等技术手段来辅助医疗诊断的科学技术日益普遍,为提高心脏病诊断的准确性,本文提出了一种在DeepFM模型的基础上改进后的较为新颖的模型——RDF模型。RDF模型由三个组件共同构成,其中因子分解机对低阶特征交互进行建模,BP神经网络对高阶特征交互进行建模,集成树则进一步提高模型的准确性和稳健性。本文在UCI数据集中的303个心脏病样本上进行实验,实验结果显示AUC值为0.8809,准确率为0.8317。
In recent years, heart disease has been a serious threat to human life and health safety, and the technology of medical diagnosis assisted by artificial intelligence is becoming more and more common. In order to improve the accuracy of heart disease diagnosis, based on DeepFM model, this paper proposes a novel model—RDF model. The RDF model is composed of three components: Factor Machine is used to model the low-order feature interaction, the BP neural network is used to model the high-order feature interaction, and the integration tree is used to further enhance the accuracy and robustness of the model. The experiment was performed on 303 heart disease samples from the UCI datasets. Experimental results show that the AUC value is 0.8809 and the accuracy is 0.8317.
%K 心脏病,因子分解机,前馈神经网络,集成树
Heart Disease %K Factorization Machine %K Feedforward Neural Network %K Integrated Tree %U http://www.hanspub.org/journal/PaperInformation.aspx?PaperID=44714