arXiv reaDer
PocketNet:ニューラルアーキテクチャ検索とマルチステップ知識蒸留を使用した超軽量顔認識ネットワーク
PocketNet: Extreme Lightweight Face Recognition Network using Neural Architecture Search and Multi-Step Knowledge Distillation
ディープニューラルネットワークは、急速に顔認識の主流の方法になりました。ただし、非常に多くのパラメータを含むこのようなモデルを組み込みデバイスに展開したり、メモリフットプリントが制限されたアプリケーションシナリオに展開したりすることは困難です。この作業では、非常に軽量で正確な顔認識ソリューションを紹介します。ニューラルアーキテクチャ検索を利用して、顔認識モデルの新しいファミリ、つまりPocketNetを開発します。また、知識蒸留に基づく新しいトレーニングパラダイム、つまり多段階知識蒸留を提示することにより、コンパクトモデルの検証パフォーマンスを向上させることを提案します。 IJB-B、IJB-C、MegaFaceなどの大規模な評価ベンチマークを含む9つの異なるベンチマークで、広範な実験的評価と最近のコンパクトな顔認識モデルとの比較を示します。 PocketNetsは、同じレベルのモデルのコンパクトさを考慮した場合、9つの主流ベンチマークで最先端(SOTA)の顔認識パフォーマンスを一貫して向上させてきました。最小のネットワークPocketNetS-128は、0.2Mのパラメーターを使用して、4Mを超えるパラメーターを含む最近のSOTAコンパクトモデルに対して非常に競争力のある結果を達成しました。トレーニングコードと事前トレーニング済みモデルは、https://github.com/fdbtrs/PocketNetで公開されています。
Deep neural networks have rapidly become the mainstream method for face recognition. However, deploying such models that contain an extremely large number of parameters to embedded devices or in application scenarios with limited memory footprint is challenging. In this work, we present an extremely lightweight and accurate face recognition solution. We utilize neural architecture search to develop a new family of face recognition models, namely PocketNet. We also propose to enhance the verification performance of the compact model by presenting a novel training paradigm based on knowledge distillation, namely the multi-step knowledge distillation. We present an extensive experimental evaluation and comparisons with the recent compact face recognition models on nine different benchmarks including large-scale evaluation benchmarks such as IJB-B, IJB-C, and MegaFace. PocketNets have consistently advanced the state-of-the-art (SOTA) face recognition performance on nine mainstream benchmarks when considering the same level of model compactness. With 0.92M parameters, our smallest network PocketNetS-128 achieved very competitive results to recent SOTA compacted models that contain more than 4M parameters. Training codes and pre-trained models are publicly released https://github.com/fdbtrs/PocketNet.
updated: Tue Aug 24 2021 13:19:08 GMT+0000 (UTC)
published: Tue Aug 24 2021 13:19:08 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アソシエイト