Technical Summary: AlignFace: Human-Aligned Face Similarity Metric with Interpretable Concept Relations
Problem Statement
Current computer vision models for generated facial content (e.g., face editing, privacy protection) rely on similarity metrics that often fail to serve as faithful proxies for human perception. While perceptual evaluation has evolved from signal-based heuristics (e.g., PSNR, SSIM) to representation-based metrics (e.g., LPIPS, DreamSim), existing approaches remain limited to behavioral modeling without cognitive alignment. These "black box" models rely on implicit and spurious relations, assume a universal observer, and fail to account for inherent variations across diverse human populations. Consequently, they provide inaccurate evaluative models for stakeholders and misleading guidance for debugging generative models, such as underestimating privacy risks or identity shifts when a human would still perceive faces as similar.
The authors argue that perception models must adhere to scientific cognitive effects to achieve true human alignment. Specifically, human face perception is characterized by:
- Dual Processing: Dependence on both featural attributes (discrete components like eye color) and configural attributes (spatial layout like pupillary distance).
- Nonlinear Scaling: Human perception of attribute differences does not scale linearly; it follows nonlinear psychophysical laws.
- Demographic Variance: The presence of own-group bias, where individuals recognize faces from their own demographic groups (ethnicity, gender) more accurately than other groups.
Methodology
1. The FACETS Dataset
To ground the model in human cognition, the authors introduced FACETS, a large-scale human-annotated dataset.
- Stimuli: Constructed from CMU Multi-PIE and CelebA datasets, featuring 120 identities (balanced for White and Asian demographics). Images were edited using diffusion-based inpainting, landmark-based warping, and GAN-based transfer to create variations in 20 cognitively-grounded attributes (14 featural, 6 configural).
- Data Collection: Utilizing a Two-Alternative Forced Choice (2AFC) protocol, participants compared edited faces against a reference.
- Overall Similarity: 9.36k triplet ratings from 78 participants.
- Attribute Similarity: 73.32k triplet ratings for 20 attributes from 611 participants.
- Analysis: Preliminary Generalized Additive Model (GAM) analysis on this data confirmed nonlinear attribute influence and significant differences in sensitivity between own-group and other-group perceptions.
2. AlignFace Architecture
AlignFace is an ante-hoc interpretable model designed to encode these cognitive principles. It consists of four primary modules:
Vision-Language Model (VLM) Encoding:
- Uses a shared vision encoder to extract representations for a face pair (z^A,z^B) and computes relational differences (Z^Δ).
- A text encoder processes attribute prompts (e.g., "Pupillary distance") to create an anchor (Z^attr) guiding the comparison along specific dimensions.
Attribute-Gated Cross-Attention (AGCA):
- Performs multi-modal fusion where image-difference features act as queries (Q) and textual attribute features act as keys (K) and values (V).
- A learnable gating factor (tanh(α)) controls the injection of attribute-conditioned signals, effectively retrieving attribute-relevant visual features from the difference representation.
Concept Bottleneck Model (CBM):
- The output of the AGCA is constrained into interpretable multi-label binary concepts (Δc^attr∈[−1,+1]).
- This forces the model's reasoning to be explicitly based on these disentangled face attributes rather than latent, uninterpretable features.
Attribute-Influence Neural Generalized Additive Model (GAM):
- Maps the attribute differences to the overall similarity score using a neural GAM: F(Δc^)=σ(β+∑fi(Δc^i)).
- Each fi is a spline function implemented via a small neural network, capturing nonlinear partial influences.
- A monotonicity constraint is imposed (∂Δc^i∂fi≥0) to ensure that higher attribute differences never decrease the overall face distance, aligning with perceptual consistency.
3. Training Objective
The model is trained as AlignFace2AFC, a multi-task triplet metric. It predicts pairwise distances for two comparisons (⟨A,Ref⟩ and ⟨B,Ref⟩) and minimizes the squared hinge loss between the predicted relative distance differences and the human binary labels (which indicate which face is more similar). The training is staged: first, attribute-level predictions are trained and frozen; then, the Neural GAM is trained on the overall distance loss.
Key Contributions
- Cognitive Characterization: Identification of specific cognitive characteristics for human face similarity perception, including featural/configural dependence, nonlinear psychophysical scaling, and demographic own-group bias.
- AlignFace Model: A cognitively-grounded, interpretable ante-hoc model integrating VLMs, Gated Cross-Attention, Concept Bottlenecks, and Neural GAMs to enforce human alignment.
- AlignFace2AFC Metric: An explainable metric that quantifies perception via the nonlinear contributions of grounded attributes.
- FACETS Dataset: A new dataset of face triplets with similarity judgments at both overall and attribute levels across 20 attributes, including demographic metadata.
Results
- Behavioral Agreement: AlignFace significantly outperforms baselines (including heuristic metrics like SSIM, learned metrics like LPIPS/DreamSim, and face-specific models like ArcFace) in aligning with human 2AFC judgments. It achieves the highest agreement scores for overall face perception.
- Attribute Alignment: The model demonstrates high agreement on attribute-level predictions (Mean agreement of 0.79 with FLIP backbone), particularly for attributes like hair length and eyebrow shape.
- Interpretability & Nonlinearity: Partial Dependence Plots (PDPs) reveal that AlignFace captures highly nonlinear relationships between attribute differences and overall similarity, achieving high Pearson correlations (ρ≈0.94–0.98) with human GAMs. This confirms the hypothesis that attribute influence is nonlinear.
- Demographic Sensitivity: The model successfully captures "own-group" effects, showing different sensitivity trends for White and Asian participants when evaluating faces from their own demographic versus other demographics.
Significance
The paper claims that by bridging learned representations with human cognitive processes, AlignFace enables more transparent and aligned perceptual evaluation metrics for face images. Unlike previous data-driven approaches that risk spurious correlations, AlignFace is theoretically justified through cognitive psychology. It provides a framework for evaluating generative models that is not only more accurate in predicting human perception but also explainable, allowing developers to understand why a model perceives two faces as similar or different based on specific featural and configural attributes. This is critical for applications involving human-centric tasks like digital makeup, restoration, and privacy protection, where the "true effect on people" must be accurately estimated.