MOON: A Mixed Objective Optimization Network for the Recognition of Facial Attributes
This addresses the challenge of imbalanced training data in multi-task deep learning for facial attribute recognition, offering an incremental improvement over existing methods.
The paper tackles the problem of multi-label imbalance in facial attribute recognition by introducing a mixed objective optimization network (MOON) with a domain-adaptive loss function, which outperforms independently trained models and advances state-of-the-art performance.
Attribute recognition, particularly facial, extracts many labels for each image. While some multi-task vision problems can be decomposed into separate tasks and stages, e.g., training independent models for each task, for a growing set of problems joint optimization across all tasks has been shown to improve performance. We show that for deep convolutional neural network (DCNN) facial attribute extraction, multi-task optimization is better. Unfortunately, it can be difficult to apply joint optimization to DCNNs when training data is imbalanced, and re-balancing multi-label data directly is structurally infeasible, since adding/removing data to balance one label will change the sampling of the other labels. This paper addresses the multi-label imbalance problem by introducing a novel mixed objective optimization network (MOON) with a loss function that mixes multiple task objectives with domain adaptive re-weighting of propagated loss. Experiments demonstrate that not only does MOON advance the state of the art in facial attribute recognition, but it also outperforms independently trained DCNNs using the same data. When using facial attributes for the LFW face recognition task, we show that our balanced (domain adapted) network outperforms the unbalanced trained network.