Hierarchical Attention-Based Age Estimation and Bias Analysis

IEEE Trans Pattern Anal Mach Intell. 2023 Dec;45(12):14682-14692. doi: 10.1109/TPAMI.2023.3319472. Epub 2023 Nov 3.

Abstract

In this work, we present a Deep Learning approach to estimate age from facial images. First, we introduce a novel attention-based approach to image augmentation-aggregation, which allows multiple image augmentations to be adaptively aggregated using a Transformer-Encoder. A hierarchical probabilistic regression model is then proposed that combines discrete probabilistic age estimates with an ensemble of regressors. Each regressor is adapted and trained to refine the probability estimate over a given age range. We show that our age estimation scheme outperforms current schemes and provides a new state-of-the-art age estimation accuracy when applied to the MORPH II and CACD datasets. We also present an analysis of the biases in the results of the state-of-the-art age estimates.