HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

PocketNet: Extreme Lightweight Face Recognition Network using Neural Architecture Search and Multi-Step Knowledge Distillation

Boutros Fadi ; Siebke Patrick ; Klemt Marcel ; Damer Naser ; Kirchbuchner Florian ; Kuijper Arjan

PocketNet: Extreme Lightweight Face Recognition Network using Neural
  Architecture Search and Multi-Step Knowledge Distillation

Abstract

Deep neural networks have rapidly become the mainstream method for facerecognition (FR). However, this limits the deployment of such models thatcontain an extremely large number of parameters to embedded and low-enddevices. In this work, we present an extremely lightweight and accurate FRsolution, namely PocketNet. We utilize neural architecture search to develop anew family of lightweight face-specific architectures. We additionally proposea novel training paradigm based on knowledge distillation (KD), the multi-stepKD, where the knowledge is distilled from the teacher model to the studentmodel at different stages of the training maturity. We conduct a detailedablation study proving both, the sanity of using NAS for the specific task ofFR rather than general object classification, and the benefits of our proposedmulti-step KD. We present an extensive experimental evaluation and comparisonswith the state-of-the-art (SOTA) compact FR models on nine different benchmarksincluding large-scale evaluation benchmarks such as IJB-B, IJB-C, and MegaFace.PocketNets have consistently advanced the SOTA FR performance on ninemainstream benchmarks when considering the same level of model compactness.With 0.92M parameters, our smallest network PocketNetS-128 achieved verycompetitive results to recent SOTA compacted models that contain up to 4Mparameters.

Code Repositories

fdbtrs/pocketnet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
lightweight-face-recognition-on-agedb-30PocketNetS
Accuracy: 0.9635
lightweight-face-recognition-on-calfwPocketNetS
Accuracy: 0.955
MParams: 0.99
lightweight-face-recognition-on-cfp-fpPocketNetS
Accuracy: 0.9334
lightweight-face-recognition-on-cplfwPocketNetS
Accuracy: 0.8893
lightweight-face-recognition-on-lfwPocketNetS
Accuracy: 0.9966
MFLOPs: 587.24
MParams: 0.99

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