用机器学习区分牙科医生类型,准确率达94.1%
Classifying Dental Care Providers Through Machine Learning with Features Ranking
- 通过特征排序筛选出治疗服务次数等关键指标
- 神经网络模型准确率最高,达94.1%且对缺失数据鲁棒
- 适合医疗资源分配研究者参考,尤其关注数据不完整场景
本研究利用2018年包含24,300条记录、20个特征的牙科服务数据集,采用机器学习模型对牙科提供者进行分类,分为标准服务提供者与安全网诊所(SNC)提供者。数据集存在38.1%的高缺失率,包含预防、治疗、检查等服务数量,以及按需付费(FFS)与管理式医疗等交付模式,及受益人人口统计信息。通过信息增益、基尼指数和ANOVA等特征排序方法,识别出治疗相关指标(TXMT_USER_CNT、TXMT_SVC_CNT)为最重要预测变量。评估了包括kNN、决策树、支持向量机(SVM)、随机梯度下降(SGD)、随机森林、神经网络和梯度提升在内的12种模型,使用10折交叉验证。在所有特征下,神经网络达到最高准确率94.1%,其次为梯度提升(93.2%)和随机森林(93.0%)。随着特征数量增加,模型性能持续提升,特别是SGD与集成方法在缺失数据下表现稳定。特征分析显示,治疗服务数量和编码信息是区分提供者类型的关键,而年龄组(AGE_GROUP)和年份(CALENDAR_YEAR)影响较小。研究强调特征选择对提升模型效率与准确性的重要性,尤其在不平衡医疗数据中,建议结合特征排序与先进算法优化牙科提供者分类,助力精准资源分配。
原文摘要 · Abstract (English)
This study investigates the application of machine learning (ML) models for classifying dental providers into two categories - standard rendering providers and safety net clinic (SNC) providers - using a 2018 dataset of 24,300 instances with 20 features. The dataset, characterized by high missing values (38.1%), includes service counts (preventive, treatment, exams), delivery systems (FFS, managed care), and beneficiary demographics. Feature ranking methods such as information gain, Gini index, and ANOVA were employed to identify critical predictors, revealing treatment-related metrics (TXMT_USER_CNT, TXMT_SVC_CNT) as top-ranked features. Twelve ML models, including k-Nearest Neighbors (kNN), Decision Trees, Support Vector Machines (SVM), Stochastic Gradient Descent (SGD), Random Forest, Neural Networks, and Gradient Boosting, were evaluated using 10-fold cross-validation. Classification accuracy was tested across incremental feature subsets derived from rankings. The Neural Network achieved the highest accuracy (94.1%) using all 20 features, followed by Gradient Boosting (93.2%) and Random Forest (93.0%). Models showed improved performance as more features were incorporated, with SGD and ensemble methods demonstrating robustness to missing data. Feature ranking highlighted the dominance of treatment service counts and annotation codes in distinguishing provider types, while demographic variables (AGE_GROUP, CALENDAR_YEAR) had minimal impact. The study underscores the importance of feature selection in enhancing model efficiency and accuracy, particularly in imbalanced healthcare datasets. These findings advocate for integrating feature-ranking techniques with advanced ML algorithms to optimize dental provider classification, enabling targeted resource allocation for underserved populations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。