HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

ATF: Towards Robust Face Alignment via Leveraging Similarity and Diversity across Different Datasets

{Jian Cheng Cong Leng Fangzhou Xiong Qinghao Hu Xing Lan}

Abstract

Face alignment is an important task in the field of multi-media. Together with the impressive progress of algorithms, various benchmark datasets have been released in recent years. Intuitively, it is meaningful to integrate multiple labeled datasets with different annotations to achieve higher performance on a target landmark detector. Although numerous efforts have been made in joint usage, there yet remain three shortages in recent works, e.g., additional computation, limitation of the markups scheme, and limited support for the regression method. To address the above problems, we proposed a novel Alternating Training Framework (ATF), which leverages similarity and diversity across multi-media sources for a more robust detector. Our framework mainly contains two sub-modules: Alternating Training with Decreasing Proportions (ATDP) and Mixed Branch Loss (mathcal LMB). In particular, ATDP trains multiple datasets simultaneously to take advantage of the diversity between them, while mathcal LMB utilizes similar landmark pairs to constrain different branches of corresponding datasets. Extensive experiments on various benchmarks show the effectiveness of our framework, and ATF is feasible for both heatmap-based network and direct coordinate regression. Specifically, the mean error even reaches 3.17 on the experiment on 300W leveraging WFLW, which significantly outperforms state-of-the-art methods. Both in an ordinary convolutional network (OCN) and HRNET, ATF achieves up to 9.96% relative improvement. Our source codes are made publicly available at https://github.com/starhiking/ATF.

Benchmarks

BenchmarkMethodologyMetrics
face-alignment-on-300wATF
NME_inter-ocular (%, Challenge): 4.89
NME_inter-ocular (%, Common): 2.75
NME_inter-ocular (%, Full): 3.17
face-alignment-on-aflw-19ATF
NME_diag (%, Full): 1.55
face-alignment-on-cofwATF
NME (inter-ocular): 3.32%
face-alignment-on-wfw-extra-dataATF
NME (inter-ocular): 4.49

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
ATF: Towards Robust Face Alignment via Leveraging Similarity and Diversity across Different Datasets | Papers | HyperAI