DiffInf: Influence-Guided Diffusion for Supervision Alignment in Facial Attribute Learning

TL;DR

DiffInf is a framework that uses influence-guided diffusion to correct annotation inconsistencies in facial attribute datasets, improving classification accuracy by refining training samples without reducing dataset size.

Contribution

It introduces a novel influence-guided diffusion method that identifies and corrects influential annotation errors in facial attribute datasets, enhancing model generalization.

Findings

Consistently improves facial attribute classification accuracy.

Effectively identifies and corrects influential annotation inconsistencies.

Enhances robustness over standard noisy-label training methods.

Abstract

Facial attribute classification relies on large-scale annotated datasets in which many traits, such as age and expression, are inherently ambiguous and continuous but are discretized into categorical labels. Annotation inconsistencies arise from subjectivity and visual confounders such as pose, illumination, expression, and demographic variation, creating mismatch between images and assigned labels. These inconsistencies introduce supervision errors that impair representation learning and degrade downstream prediction. We introduce DiffInf, a self-influence--guided diffusion framework for mitigating annotation inconsistencies in facial attribute learning. We first train a baseline classifier and compute sample-wise self-influence scores using a practical first-order approximation to identify training instances that disproportionately destabilize optimization. Instead of discarding these influential samples, we apply targeted generative correction via a latent diffusion autoencoder to better align visual content with assigned labels while preserving identity and realism. To enable differentiable guidance during correction, we train a lightweight predictor of high-influence membership and use it as a surrogate influence regularizer. The edited samples replace the originals, yielding an influence-refined dataset of unchanged size. Across multi-class facial attribute classification, DiffInf consistently improves generalization compared with standard noisy-label training, robust optimization baselines, and influence-based filtering. Our results demonstrate that repairing influential annotation inconsistencies at the image level enhances downstream facial attribute classification without sacrificing distributional coverage.