arXiv reaDer
Adaptive Wing Loss for Robust Face Alignment via Heatmap Regression
 ディープネットワークによるヒートマップ回帰は、顔のランドマークをローカライズするための主流のアプローチの1つになっています。ただし、ヒートマップ回帰の損失関数はほとんど研究されていません。このペーパーでは、顔の位置合わせの問題におけるヒートマップ回帰の理想的な損失関数の特性を分析します。次に、アダプティブウィング損失と呼ばれる新しい損失関数を提案します。この関数は、形状をさまざまなタイプのグラウンドトゥルースヒートマップピクセルに適合させることができます。この適応性により、前景のピクセルでは損失が減り、背景のピクセルでは損失が減ります。前景と背景のピクセル間の不均衡に対処するために、前景と困難な背景のピクセルに高い重みを割り当てる加重損失マップも提案します。これにより、トレーニングプロセスは、ランドマークのローカリゼーションに不可欠なピクセルにより焦点を当てるのに役立ちます。顔の位置合わせの精度をさらに向上させるために、境界予測と境界座標によるCoordConvを導入します。 COFW、300W、WFLWなどのさまざまなベンチマークでの広範囲な実験により、さまざまな評価指標において、当社のアプローチが最先端を大幅に上回っていることを示しています。さらに、アダプティブウィング損失は他のヒートマップ回帰タスクにも役立ちます。コードはで公開されます。
Heatmap regression with a deep network has become one of the mainstream approaches to localize facial landmarks. However, the loss function for heatmap regression is rarely studied. In this paper, we analyze the ideal loss function properties for heatmap regression in face alignment problems. Then we propose a novel loss function, named Adaptive Wing loss, that is able to adapt its shape to different types of ground truth heatmap pixels. This adaptability penalizes loss more on foreground pixels while less on background pixels. To address the imbalance between foreground and background pixels, we also propose Weighted Loss Map, which assigns high weights on foreground and difficult background pixels to help training process focus more on pixels that are crucial to landmark localization. To further improve face alignment accuracy, we introduce boundary prediction and CoordConv with boundary coordinates. Extensive experiments on different benchmarks, including COFW, 300W and WFLW, show our approach outperforms the state-of-the-art by a significant margin on various evaluation metrics. Besides, the Adaptive Wing loss also helps other heatmap regression tasks. Code will be made publicly available at
updated: Tue May 19 2020 05:30:24 GMT+0000 (UTC)
published: Tue Apr 16 2019 02:00:24 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト