A Step Forward Towards Trustworthy Risk-Aware Facial Retrieval (RA-FR)
Abstract
Facial image retrieval in unconstrained surveillance environments is a high-stakes challenge where missing a subject of interest -- a single false negative -- is simply not an option. Despite near-perfect performance on curated benchmarks, current recognition systems falter under real-world domain shifts such as low resolution, motion blur, and uncontrolled illumination (e.g., SCFace). Addressing this reliability gap, we propose Risk-Aware Facial Retrieval (RA-FR), a framework that moves beyond fixed Top- retrieval to adaptive set generation, guaranteeing ground truth inclusion within a user-specified risk level () and confidence level (). Our approach integrates three core contributions: (1) reducing aleatoric uncertainty via a hybrid blind face restoration technique coupling Latent Consistency Models (InterLCM) and DiffBIR; (2) extracting discriminative, restoration-robust features via self-supervised DINOv1 ViT-B with GGeM pooling; and (3) employing conformal prediction with Hoeffding's inequality to dynamically calibrate retrieval set sizes based on query uncertainty. On the IMFDB benchmark, it consistently satisfies a 5% risk target with an average retrieval set size of approximately 10 images. By unifying domain-specific restoration, robust representation learning, and provable decision rules, RA-FR offers a pipeline that makes facial retrieval in surveillance both reliable and auditable. The code is available at: https://github.com/MuhammadEmmadSiddiqui/RA-FR.
Cite
@article{arxiv.2607.16279,
title = {A Step Forward Towards Trustworthy Risk-Aware Facial Retrieval (RA-FR)},
author = {Muhammad Emmad Siddiqui and Muhammad Rafi},
journal= {arXiv preprint arXiv:2607.16279},
year = {2026}
}
Comments
Accepted at the 28th International Conference on Pattern Recognition (ICPR 2026). Extended version with additional citations and expanded sections