Open AccessFacial-image based Age Estimation Using Imbalanced Datasets
Author(s) -
Yasi Liu,
Tianzi Wang,
Zheng Liu,
Yang Lü
Publication year - 2019
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/1267/1/012012
Subject(s) - artificial intelligence , computer science , pattern recognition (psychology) , convolutional neural network , classifier (uml) , probability distribution , population , metric (unit) , support vector machine , artificial neural network , mathematics , statistics , operations management , demography , sociology , economics
Facial image based human age estimation is of great application significance. The public-available facial image datasets used for age estimation suffer greatly from the uneven distribution of images of different age groups, which may lead to the low estimation accuracy of the under-sampled age categories and limit the usage of the age estimation in certain applications. We propose a three-stage probability adjustment based CNN algorithm to solve the imbalanced distribution problem of the dataset. In particular, we construct an ENIN neural network structure by applying the Network in Network (NIN) structure to the traditional convolution neural network (CNN) and use the probability vector adjustment to improve the classification accuracy of the under-sampled age categories. Then, we filter out the images with high possibility of being misclassified after the probability vector adjustment and reset their categories by comparing cosine similarity and retraining the ensembled ENIN classifier. We also introduce a population-age-distribution based accuracy metric Accuracy-P to estimate the performance of the age estimation algorithm in real-world applications. Our experimental results confirm that our algorithm can effectively improve the overall estimation accuracy by significantly improving the accuracy of the under-sampled age groups while maintaining satisfactory accuracy for the other age groups.