N Yogesh, Shrinivasacharya Purohit, Naik Nagaraj
Siddaganga Institute of Technology, Tumkuru, Karanataka, India.
Visvesveraya Technological University, Belagavi, India.
PeerJ Comput Sci. 2024 Nov 13;10:e2467. doi: 10.7717/peerj-cs.2467. eCollection 2024.
Chronic kidney disease (CKD) involves numerous variables, but only a few significantly impact the classification task. The statistically equivalent signature (SES) method, inspired by constraint-based learning of Bayesian networks, is employed to identify essential features in CKD. Unlike conventional feature selection methods, which typically focus on a single set of features with the highest predictive potential, the SES method can identify multiple predictive feature subsets with similar performance. However, most feature selection (FS) classifiers perform suboptimally with strongly correlated data. The FS approach faces challenges in identifying crucial features and selecting the most effective classifier, particularly in high-dimensional data. This study proposes using the Least Absolute Shrinkage and Selection Operator (LASSO) in conjunction with the SES method for feature selection in CKD identification. Following this, an ensemble deep-learning model combining long short-term memory (LSTM) and gated recurrent unit (GRU) networks is proposed for CKD classification. The features selected by the hybrid feature selection method are fed into the ensemble deep-learning model. The model's performance is evaluated using accuracy, precision, recall, and F1 score metrics. The experimental results are compared with individual classifiers, including decision tree (DT), Random Forest (RF), logistic regression (LR), and support vector machine (SVM). The findings indicate a 2% improvement in classification accuracy when using the proposed hybrid feature selection method combined with the LSTM and GRU ensemble deep-learning model. Further analysis reveals that certain features, such as HEMO, POT, bacteria, and coronary artery disease, contribute minimally to the classification task. Future research could explore additional feature selection methods, including dynamic feature selection that adapts to evolving datasets and incorporates clinical knowledge to enhance CKD classification accuracy further.
慢性肾脏病(CKD)涉及众多变量,但只有少数变量对分类任务有显著影响。受贝叶斯网络基于约束学习启发的统计等效特征(SES)方法,被用于识别CKD中的关键特征。与传统的特征选择方法不同,传统方法通常关注具有最高预测潜力的单一特征集,而SES方法可以识别多个性能相似的预测特征子集。然而,大多数特征选择(FS)分类器在处理强相关数据时表现欠佳。FS方法在识别关键特征和选择最有效的分类器方面面临挑战,尤其是在高维数据中。本研究提出将最小绝对收缩和选择算子(LASSO)与SES方法结合用于CKD识别中的特征选择。在此之后,提出了一种结合长短期记忆(LSTM)和门控循环单元(GRU)网络的集成深度学习模型用于CKD分类。通过混合特征选择方法选择的特征被输入到集成深度学习模型中。使用准确率、精确率、召回率和F1分数指标评估模型的性能。将实验结果与包括决策树(DT)、随机森林(RF)、逻辑回归(LR)和支持向量机(SVM)在内的单个分类器进行比较。研究结果表明,使用所提出的混合特征选择方法结合LSTM和GRU集成深度学习模型时,分类准确率提高了2%。进一步分析表明,某些特征,如血红蛋白、钾、细菌和冠状动脉疾病,对分类任务的贡献最小。未来的研究可以探索其他特征选择方法,包括适应不断变化的数据集并纳入临床知识以进一步提高CKD分类准确率的动态特征选择。