用混合模型精准预测糖尿病,准确率达99.79%
Predicting Diabetes Using Machine Learning: A Comparative Study of Classifiers
- 融合CNN与LSTM的DNet模型,捕捉特征与时间依赖
- 在真实数据集上达99.79%准确率与99.98%AUC-ROC
- 适合医疗诊断与疾病早期预测场景
糖尿病是全球重大健康挑战,常引发肾病、失明和心脏病等并发症。机器学习在医疗中可用于高效精准的疾病预测,助力早期干预。本研究提出一种创新糖尿病预测框架,结合逻辑回归、SVM、朴素贝叶斯、随机森林等传统方法,以及AdaBoost、梯度提升、极端梯树和XGBoost等集成方法。核心是新提出的DNet模型,一种融合卷积神经网络(CNN)与长短期记忆(LSTM)层的混合架构,用于有效特征提取与序列学习。DNet包含初始卷积块、带有跳跃连接的残差块以促进信息流动,采用批归一化与丢弃层实现稳健正则化,并通过LSTM层捕捉数据中的时序依赖。基于Kaggle来源的真实世界糖尿病数据集,评估涵盖交叉验证准确率、精确率、召回率、F1分数及ROC-AUC。DNet表现最优,准确率达99.79%,ROC-AUC达99.98%,证明其在糖尿病预测中的优越性。该混合架构凸显了在医学诊断与疾病预测任务中的应用价值。
原文摘要 · Abstract (English)
Diabetes remains a significant health challenge globally, contributing to severe complications like kidney disease, vision loss, and heart issues. The application of machine learning (ML) in healthcare enables efficient and accurate disease prediction, offering avenues for early intervention and patient support. Our study introduces an innovative diabetes prediction framework, leveraging both traditional ML techniques such as Logistic Regression, SVM, Naïve Bayes, and Random Forest and advanced ensemble methods like AdaBoost, Gradient Boosting, Extra Trees, and XGBoost. Central to our approach is the development of a novel model, DNet, a hybrid architecture combining Convolutional Neural Network (CNN) and Long Short-Term Memory (LSTM) layers for effective feature extraction and sequential learning. The DNet model comprises an initial convolutional block for capturing essential features, followed by a residual block with skip connections to facilitate efficient information flow. Batch Normalization and Dropout are employed for robust regularization, and an LSTM layer captures temporal dependencies within the data. Using a Kaggle-sourced real-world diabetes dataset, our model evaluation spans cross-validation accuracy, precision, recall, F1 score, and ROC-AUC. Among the models, DNet demonstrates the highest efficacy with an accuracy of 99.79% and an AUC-ROC of 99.98%, establishing its potential for superior diabetes prediction. This robust hybrid architecture showcases the value of combining CNN and LSTM layers, emphasizing its applicability in medical diagnostics and disease prediction tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。