arXiv reaDer
非常に大規模な顔認識のための効率的なトレーニング アプローチ
An Efficient Training Approach for Very Large Scale Face Recognition
顔認識は、超大規模で適切にラベル付けされたデータセットにより、深層学習時代に大きな進歩を遂げました。ただし、超大規模なデータセットでのトレーニングは時間がかかり、多くのハードウェア リソースを消費します。したがって、効率的なトレーニング アプローチを設計することが重要かつ不可欠です。計算コストとメモリ コストがかかるのは、主に完全接続 (FC) 層の高次元性によるものです。具体的には、次元数は顔のアイデンティティの数によって決定され、100 万レベル以上の場合もあります。この目的のために、Faster Face Classification (F^2C) と呼ばれる、超大規模顔データセットの新しいトレーニング アプローチを提案します。 F^2C では、まず、アイデンティティの中心を生成し、顔認識のために顔の特徴を抽出するために使用される Gallery Net と Probe Net をそれぞれ定義します。 Gallery Net は Probe Net と同じ構造を持ち、移動平均パラダイムで Probe Net からパラメータを継承しています。その後、FC レイヤーのトレーニング時間とハードウェア コストを削減するために、Gallery Net からの特徴を保存し、正のサンプル (その ID が DCP にある) を使用して内積 (ロジット) を計算する動的クラス プール (DCP) を提案します。 ) 各ミニバッチで。 DCP は FC 層の代替と見なすことができますが、はるかに小さいため、計算とメモリのコストが大幅に削減されます。負のサンプル (その ID が DCP にない) の場合、負のサンプルと DCP のそれらの間のコサインの類似性を最小限に抑えます。次に、DCP のパラメーターの更新効率を向上させるために、ID ベースとインスタンス ベースのローダーを含むデュアル データ ローダーを設計して、特定の ID とサンプルをミニバッチで生成します。
Face recognition has achieved significant progress in deep-learning era due to the ultra-large-scale and well-labeled datasets. However, training on ultra-large-scale datasets is time-consuming and takes up a lot of hardware resource. Therefore, designing an efficient training approach is crucial and indispensable. The heavy computational and memory costs mainly result from the high dimensionality of the Fully-Connected (FC) layer. Specifically, the dimensionality is determined by the number of face identities, which can be million-level or even more. To this end, we propose a novel training approach for ultra-large-scale face datasets, termed Faster Face Classification (F^2C). In F^2C, we first define a Gallery Net and a Probe Net that are used to generate identities' centers and extract faces' features for face recognition, respectively. Gallery Net has the same structure as Probe Net and inherits the parameters from Probe Net with a moving average paradigm. After that, to reduce the training time and hardware costs of the FC layer, we propose a Dynamic Class Pool (DCP) that stores the features from Gallery Net and calculates the inner product (logits) with positive samples (whose identities are in the DCP) in each mini-batch. DCP can be regarded as a substitute for the FC layer but it is far smaller, thus greatly reducing the computational and memory costs. For negative samples (whose identities are not in DCP), we minimize the cosine similarities between negative samples and those in DCP. Then, to improve the update efficiency of DCP's parameters, we design a dual data-loader including identity-based and instance-based loaders to generate a certain of identities and samples in mini-batches.
updated: Thu Jun 03 2021 13:20:29 GMT+0000 (UTC)
published: Fri May 21 2021 14:34:00 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト