Schwierigkeitsabhängige Dekorrelation von Aufmerksamkeitsköpfen für robuste Gesichtserkennung
Autoren & Institutionen
Tahar Chettaoui
Fraunhofer Institute for Computer Graphics Research IGD
Guray Ozgur
Fraunhofer Institute for Computer Graphics Research IGD
TU Darmstadt
Eduarda Caldeira
Fraunhofer Institute for Computer Graphics Research IGD
TU Darmstadt
Naser Damer
Fraunhofer Institute for Computer Graphics Research IGD
TU Darmstadt
Fadi Boutros
Fraunhofer Institute for Computer Graphics Research IGD
Welches Problem es löst
ViT-HarDis zielt insbesondere dort auf redundante Aufmerksamkeit ab, wo die Einbettung einer anderen Identität gefährlich nahe kommt. Es wurde entwickelt, um die Gesichtserkennung in großem Maßstab und mit niedriger Auflösung zu verbessern und gleichzeitig die Inferenzlatenz, die Modellform und die Einbettungsschnittstellen unverändert zu lassen.
Zentrales Ergebnis
Bei der kontrollierten ViT-S-Ablation steigt die TAR bei FAR 1e-4 von 84,46 % auf 85,72 % bei IJB-B und von 88,18 % auf 89,09 % bei IJB-C. Mit ViT-B, das auf WebFace4M trainiert wurde, meldet die Methode 95,84 % für IJB-B, 97,24 % für IJB-C und 75,97 % für TinyFace Rank-1 und liegt damit an der Spitze der verglichenen ViT-basierten Systeme. Die Inferenzkosten bleiben unverändert, da die Strafe nur für das Training gilt, obwohl die Autoren von etwa 30 % längerem Training auf WebFace4M berichten.
Abstract
Die Mehrkopfaufmerksamkeit in Vision Transformers (ViTs) ermöglicht unabhängige Aufmerksamkeitsmuster über mehrere Köpfe hinweg, aber ohne explizite Regularisierung können Köpfe eher redundante Muster als komplementäre Rollen lernen. Korrelationsbasierte Strafen fördern die Kopfentwirrung, wenden jedoch eine einheitliche Regularisierung über alle Trainingsbeispiele hinweg an. Bei der Gesichtserkennung (FR) sind harte Proben, deren Einbettungen nahe an konkurrierenden Identitäten liegen, für die Optimierung und Verallgemeinerung von entscheidender Bedeutung. Aus diesem Grund schlagen wir ViT-HarDis vor, das die Aufmerksamkeits-Kopf-Entwirrung von der Stichprobenschwierigkeit abhängig macht: Stichproben mit hoher Kosinus-Ähnlichkeit zu negativen Klassenzentren erhalten eine stärkere Kopf-Dekorrelation, wodurch die Spezialisierung dort erzwungen wird, wo die Identitätsmehrdeutigkeit am höchsten ist. Experimente zeigen bei den meisten Benchmarks Verbesserungen gegenüber der Basislinie und der gleichmäßigen Entflechtung. Unsere Analyse zeigt, dass ViT-HarDis die Ähnlichkeit harter Stichproben mit konkurrierenden Klassenzentren verringert und gleichzeitig die Aufmerksamkeitsähnlichkeit zwischen den Köpfen verringert, und dass die Anwendung der Entflechtung nur auf die letzten Transformer-Schichten die Regularisierung des gesamten Netzwerks übertrifft. Auf WebFace4M trainiert, erzielt ViT-HarDis SOTA-Ergebnisse unter den ViT-basierten Methoden auf IJB-B, IJB-C und TinyFace Rank-1, während es beim Training auf MS1MV3 die besten IJB-B- und TinyFace-Ergebnisse erzielt, wodurch FR ohne Architekturänderungen oder zusätzliche Inferenzkosten verbessert wird. Code: https://github.com/TaharChettaoui/Hard-Sample-Conditioned-Attention-for-FR.
Ausgangspunkt der Forschung
Vision Transformers können unterschiedliche Gesichtsregionen unterschiedlichen Aufmerksamkeitsköpfen zuordnen, aber diese Köpfe konvergieren oft in redundanten Mustern. Einheitliche Dekorrelationsstrafen behandeln ein einfaches Frontalporträt und eine harte Stichprobe mit unklarer Identität gleichermaßen, auch wenn schwierige Stichproben in der Nähe konkurrierender Klassenzentren diejenigen sind, die am wahrscheinlichsten Verifizierungsfehler feststellen. Das Papier stellt die Frage, ob die Schulungskapazität auf diese schwierigen Identitäten konzentriert werden kann, ohne die eingesetzte Architektur zu ändern.
Methode
Die Methode misst die Probenhärte anhand der Kosinusähnlichkeit zum nächstgelegenen negativen Klassenzentrum. Für jeden ausgewählten Transformer-Block wird ein Inter-Head-Aufmerksamkeitskorrelations-Strafwert berechnet, dieser wird für schwerer korrekt klassifizierte Proben stärker gewichtet und zum normalen marginbasierten Gesichtserkennungsverlust addiert. Experimente vergleichen die Grundlinie, die gleichmäßige Entwirrung und die härtebedingte Entwirrung; Schichtablationen zeigen, dass die Regularisierung nur des letzten Viertels der Transformer-Blöcke effektiver ist, als die Einschränkung überall anzuwenden.
Einordnung
Für Teams mit einem ViT-Gesichtsencoder ist dies eine Verbesserung des Trainingsverfahrens, kein Umbau der Inferenz. Die Vorteile zeigen sich vor allem bei schwierigen Tests mit vielen Identitäten und niedriger Auflösung; dafür steigt die Trainingsdauer, und eine Prüfung an der Zielpopulation bleibt nötig.