Resolving the non-uniformity in the feature space of age estimation: A deep learning model based on feature clusters of panoramic images

Abstract

Age estimation is important in forensics, and numerous techniques have been investigated to estimate age based on various parts of the body. Among them, dental tissue is considered reliable for estimating age as it is less influenced by external factors. The advancement in deep learning has led to the development of automatic estimation of age using dental panoramic images. Typically, most of the medical datasets used for model learning are non-uniform in the feature space. This causes the model to be highly influenced by dense feature areas, resulting in adequate estimations; however, relatively poor estimations are observed in other areas. An effective solution to address this issue can be pre-dividing the data by age feature and training each regressor to estimate the age for individual features. In this study, we divide the data based on feature clusters obtained from unsupervised learning. The developed model comprises a classification head and multi-regression head, wherein the former predicts the cluster to which the data belong and the latter estimates the age within the predicted cluster. The visualization results show that the model can focus on a clinically meaningful area in each cluster for estimating age. The proposed model outperforms the models without feature clusters by focusing on the differences within the area. The performance improvement is particularly noticeable in the growth and aging periods. Furthermore, the model can adequately estimate the age even for samples with a high probability of classification error as they are located at the border of two feature clusters.

Keywords: Age estimation; Deep learning; Dental panoramic images; Feature cluster; Feature space; Multi-head network.