atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

Mitigating Biases in Blackbox Feature Extractors for Image Classification Tasks

NeurIPS 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Group-robust classification · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Studies bias propagation from frozen blackbox pretrained feature extractors into downstream classifiers and proposes a bias-amplify-then-mitigate method using a clustering-based adaptive margin loss with randomized margins on an MLP adapter.

Problem

Debiasing downstream image classifiers built on frozen, blackbox pretrained feature encoders when only the classifier and a small adapter can be trained, and the bias attribute is unknown.

Approach

Amplify bias into an adapter via ERM with high weight decay, cluster the biased adapter features to estimate per-sample class-proportion-based margins, then train a debiased model with an ArcFace-style angular margin loss whose margins are Gaussian-randomized.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply cluster-based adaptive margin lossfrozen ResNet-18 feature extractor with adapter · Waterbirdsworst-group accuracy: 67.57±1.29 -> 80.29±2.50Helped
apply cluster-based adaptive margin lossfrozen ResNet-18 feature extractor with adapter · CelebA Blond Hairworst-group accuracy: 78.37±0.14 -> 81.61±1.02Helped
apply cluster-based adaptive margin lossfrozen ResNet-18 feature extractor with adapter · CMNIST-0.9average accuracy: 85.22±0.37 -> 87.33±0.35Helped
apply cluster-based adaptive margin lossfrozen ResNet-18 feature extractor with adapter · CMNIST-0.995bias-conflicting accuracy: 65.48±0.21 -> 72.56±0.88Helped
replace adaptive margins with a constant margin of 0.5 (ablation)cluster-based margin method · Waterbirdsworst-group accuracy: 80.29 -> 55.76Hurt
remove Gaussian randomization of adaptive margins (ablation)cluster-based margin method · Waterbirdsworst-group accuracy: 80.29 -> 79.28Hurt
remove Gaussian randomization of adaptive margins (ablation)cluster-based margin method · CelebA Blond Hairworst-group accuracy: 81.61 -> 78.51Hurt
cluster frozen encoder features instead of adapter features (ablation)cluster-based margin method · Waterbirdsworst-group accuracy: 80.29 -> 74.92Hurt
use loss-weighted cross-entropy instead of cluster-based margin loss (ablation)bias mitigation model · CelebA Blond Hairworst-group accuracy: 81.61 -> 45.56Hurt
use cluster-weighted cross-entropy instead of cluster-based margin loss (ablation)bias mitigation model · CelebA Blond Hairworst-group accuracy: 81.61 -> 69.4Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Waterbirds CelebA ColorMNIST UTKFace BAR

Benchmarks: Waterbirds CelebA ColorMNIST-0.9 ColorMNIST-0.995 UTKFace BAR

Metrics: worst-group accuracy average group accuracy bias-conflicting accuracy bias-aligned accuracy

Baselines: ERM DebiAN BPA LfF JTT GEORGE Contrastive Adapter (Co-Ada) DFR

Builds on: ArcFace Learning from Failure (LfF) Just Train Twice (JTT) Contrastive Adapter Deep Feature Reweighting (DFR)

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy