用牙齿测量数据,机器学习精准预测活体年龄与性别。
An Explainable Machine Learning Approach for Age and Gender Estimation in Living Individuals Using Dental Biometrics
- 结合多种牙齿指标,用集成学习模型进行预测。
- XGB模型年龄估计F1达73.26%,随机森林性别识别达77.53%。
- 引入SHAP解释模型,帮助牙科专家理解预测依据。
年龄与性别估算是法医调查和人类学研究中的关键任务。本研究基于862名活体个体(男性459人,女性403人)的牙科数据,利用上颌和下颌的前磨牙及磨牙的根尖片,提取冠高(CH)、冠髓腔高(CPCH)和牙冠指数(TCI)等指标,构建机器学习预测系统。采用CatBoost、GBM、AdaBoost、随机森林(RF)、XGBoost(XGB)、LightGBM和极端梯度提升树(ETC)等多种模型,提出一种新型集成学习方法,针对不同牙科指标分别训练模型以提升准确性。同时引入可解释AI框架SHAP,增强结果可读性。实验结果显示,随机森林与XGB模型表现最佳,其中XGB在年龄估计中达到73.26%的F1分数,随机森林在性别分类中达到77.53%的F1分数。该研究显著推动了牙科法医学的智能化发展。
原文摘要 · Abstract (English)
Objectives: Age and gender estimation is crucial for various applications, including forensic investigations and anthropological studies. This research aims to develop a predictive system for age and gender estimation in living individuals, leveraging dental measurements such as Coronal Height (CH), Coronal Pulp Cavity Height (CPCH), and Tooth Coronal Index (TCI). Methods: Machine learning models were employed in our study, including Cat Boost Classifier (Catboost), Gradient Boosting Machine (GBM), Ada Boost Classifier (AdaBoost), Random Forest (RF), eXtreme Gradient Boosting (XGB), Light Gradient Boosting Machine (LGB), and Extra Trees Classifier (ETC), to analyze dental data from 862 living individuals (459 males and 403 females). Specifically, periapical radiographs from six teeth per individual were utilized, including premolars and molars from both maxillary and mandibular. A novel ensemble learning technique was developed, which uses multiple models each tailored to distinct dental metrics, to estimate age and gender accurately. Furthermore, an explainable AI model has been created utilizing SHAP, enabling dental experts to make judicious decisions based on comprehensible insight. Results: The RF and XGB models were particularly effective, yielding the highest F1 score for age and gender estimation. Notably, the XGB model showed a slightly better performance in age estimation, achieving an F1 score of 73.26%. A similar trend for the RF model was also observed in gender estimation, achieving a F1 score of 77.53%. Conclusions: This study marks a significant advancement in dental forensic methods, showcasing the potential of machine learning to automate age and gender estimation processes with improved accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。