最近、野生での顔認識は目覚ましい成功を収めており、重要なエンジンの1つはトレーニングデータのサイズの増加です。たとえば、最大の顔データセットであるWebFace42Mには、約200万のIDと4200万の顔が含まれています。ただし、膨大な数の面により、トレーニング時間、コンピューティングリソース、およびメモリコストの制約が生じます。この問題に関する現在の研究は、主に、多数のIDによって引き起こされるGPUメモリ消費を削減するための効率的な完全接続層(FC)の設計に焦点を当てています。この作業では、貪欲な収集操作(つまり、コアセット選択の観点)によって引き起こされる最新の顔データセットの冗長性の問題を解決することにより、これらの制約を緩和します。顔認識の問題に関するこの観点での最初の試みとして、既存の方法はパフォーマンスと効率の両方で制限されていることがわかりました。優れた費用対効果のために、Face-NMSと呼ばれる新しいフィルタリング戦略を提供します。 Face-NMSはフィーチャスペースで動作し、コアセットを生成する際にローカルおよびグローバルのスパース性を同時に考慮します。実際には、Face-NMSは、オブジェクト検出コミュニティのNon-Maximum Suppression(NMS)に類似しています。全体的なスパース性への潜在的な寄与によって顔をランク付けし、ローカルのスパース性との類似性が高いペアの余分な顔を除外します。効率の側面に関しては、Face-NMSは、プロキシモデルのトレーニングで、小さいながらも十分なプロキシデータセットを適用することにより、パイプライン全体を高速化します。その結果、Face-NMSを使用して、主要なベンチマークでのパフォーマンスを維持しながら、WebFace42Mデータセットを60%に正常にスケールダウンし、40%のリソース節約と1.64倍の高速化を実現しました。このコードは、https://github.com/HuangJunJie2017/Face-NMSで参照できるように公開されています。
Recently, face recognition in the wild has achieved remarkable success and one key engine is the increasing size of training data. For example, the largest face dataset, WebFace42M contains about 2 million identities and 42 million faces. However, a massive number of faces raise the constraints in training time, computing resources, and memory cost. The current research on this problem mainly focuses on designing an efficient Fully-connected layer (FC) to reduce GPU memory consumption caused by a large number of identities. In this work, we relax these constraints by resolving the redundancy problem of the up-to-date face datasets caused by the greedily collecting operation (i.e. the core-set selection perspective). As the first attempt in this perspective on the face recognition problem, we find that existing methods are limited in both performance and efficiency. For superior cost-efficiency, we contribute a novel filtering strategy dubbed Face-NMS. Face-NMS works on feature space and simultaneously considers the local and global sparsity in generating core sets. In practice, Face-NMS is analogous to Non-Maximum Suppression (NMS) in the object detection community. It ranks the faces by their potential contribution to the overall sparsity and filters out the superfluous face in the pairs with high similarity for local sparsity. With respect to the efficiency aspect, Face-NMS accelerates the whole pipeline by applying a smaller but sufficient proxy dataset in training the proxy model. As a result, with Face-NMS, we successfully scale down the WebFace42M dataset to 60% while retaining its performance on the main benchmarks, offering a 40% resource-saving and 1.64 times acceleration. The code is publicly available for reference at https://github.com/HuangJunJie2017/Face-NMS.