In order to reduce the dimensionality of the dataset to obtain the best feature subset as well as to improve the accuracy of the prognostic survival classification of gastric cancer, a hybrid network model of deep belief network and support vector machine combined with feature selection algorithm was proposed. Based on the filtered feature selection algorithm, a distance coefficient was introduced to adjust the overall degree of bias and reduce the instability of the calculated weight values, so as to construct new sample weight values, and then analyze the subset of features that have a greater impact on the survival period of gastric cancer through the Pearson’s correlation coefficient; The constrained Boltzmann machine module was adopted in the deep belief network, and then the subset of features in the hidden layer was subjected to the feature extraction; Finally, the support vector machine was used to classify the output values of the last layer of the deep belief network to realize the classification of gastric cancer survival. By improving the feature selection algorithm and combining the advantages of deep belief network and support vector machine, the model showed better accuracy, AUC value and F1 value in the experiments, which are 81.2%, 83.4% and 81.5%, respectively, compared with the traditional single machine learning method.
SUNGH, FERLAYJ, SIEGELR L, et al. Global cancer statistics 2020: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries[J]. CA: A Cancer Journal for Clinicians, 2021, 71(3): 209⁃249.
GAOMeihong, SHANGXuequn. Artificial intelligence based prediction for cancer susceptibility, recurrence and survival[J]. Progress in Biochemistry and Biophysics, 2022, 49(9): 1687⁃1702.
CHENWen, WANGXu, DUANHuihong, et al. Application of deep learning in cancer prognosis prediction model[J]. Journal of Biomedical Engineering, 2020, 37(5): 918⁃929.
CHANGZiwei, LIUHui, ZHANGQiumeng, et al. Establishment of lncRNA predictive model based on TCGA and LASSO for prognosis of gastric cancer[J]. Chinese Clinical Oncology, 2020, 25(9): 823⁃829.
DENGDingwen, YINZhongqiang. Constructing lncRNA prediction model of gastric cancer to promote the treatment of gastric cancer[J]. Advances in Clinical Medicine, 2023, 13(12): 19461⁃19470.
[10]
AFRASHM R, MIRBAGHERIE, MASHOUFIM, et al. Optimizing prognostic factors of five year survival in gastric cancer patients using feature selection techniques with machine learning algorithms: A comparative study[J]. BMC Medical Informatics and Decision Making, 2023, 23(1): 54.
MENGPenghui, HUANGKaiwen, XULei. Optimized feature selection algorithm based on ReliefF and ant colony for cardiomyopathy classification[J]. Software Engineering and Applications, 2022, 11(2): 267⁃281.
[13]
BURUGADDAV R, PATILV C, VYASN, et al. Enhancing breast cancer diagnosis: A deep belief network approach for mammography image analysis[C]//2023 International Conference on Advances in Computation, Communication and Information Technology (ICAICCIT), Faridabad, 2023: 46⁃51.
[14]
SMOLANDERJ, DEHMERM, EMMERT‑STREIBF. Comparing deep belief networks with support vector machines for classifying gene expression data from complex disorders[J]. FEBS Open Bio, 2019, 9(7): 1232⁃1248.
WUChenwen, LIChenyang, GUOShujin, et al. Feature gene selection method based on ReliefF and ant colony optimization[J]. Application Research of Computers, 2018, 35(9): 2610⁃2613.
[17]
CHAOHao, SONGCheng, LUBaoyun, et al. Feature extraction based on DBN⁃SVM for tone recognition[J]. Journal of Information Processing Systems, 2019, 15(1): 91⁃99.
[18]
SUXiyuan, CAOChangqing, ZENGXiaodong, et al. Application of DBN and GWO⁃SVM in analog circuit fault diagnosis[J]. Scientific Reports, 2021, 11(1): 7969.
[19]
高鹏. 基于改进深度置信网络的胃癌诊断预测模型研究与应用[D]. 合肥: 安徽大学, 2021.
[20]
GAOPeng. Research and application of gastric cancer diagnosis prediction model based on improved deep belief network[D]. Hefei: Anhui University, 2021.
[21]
MOHAMMEDA J, HASSSANM M, KADIRD H. Improving classification performance for a novel imbalanced medical dataset using SMOTE method[J]. International Journal of Advanced Trends in Computer Science and Engineering, 2020, 9(3): 3161⁃3172.
[22]
AFRASHM R, SHAFIEEM, KAZEMI⁃ARPANAHIH. Establishing machine learning models to predict the early risk of gastric cancer based on lifestyle factors[J]. BMC Gastroenterology, 2023, 23(1): 6.
[23]
TIANHuakai, LIUZitao, LIUJiang, et al. Application of machine learning algorithm in predicting distant metastasis of T1 gastric cancer[J]. Scientific Reports, 2023, 13(1): 5741.
[24]
DURKAYA KURTCANB, OZCANT. Predicting customer churn using grey wolf optimization⁃based support vector machine with principal component analysis[J]. Journal of Forecasting, 2023, 42(6): 1329⁃1340.
[25]
MABaoshan, YANGe, CHAIBingjie, et al. XGBLC: An improved survival prediction model based on XGBoost[J]. Bioinformatics, 2022, 38(2): 410⁃418.