Bias Mitigation in Face Recognition via Demographic-based Supervised Contrastive Learning
This paper proposes Demographic-based Supervised Contrastive Learning (DeSCon), a novel loss function that improves fairness in face recognition by specifically targeting error rates in the tail of the non-match score distribution through demographic-aware batch composition and pair selection, thereby mitigating bias beyond what simple training data balancing can achieve.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
Imagine you are walking through a massive, crowded digital hallway where a security guard checks everyone's ID. This guard is an AI, a super-smart computer program trained to recognize faces. For a long time, these guards have been incredibly good at their job, but they had a secret flaw: they were much better at recognizing some people than others. If you had a certain skin tone, age, or gender, the guard might squint harder, make more mistakes, or even let the wrong person slip by. This isn't just a glitch; it's a bias. The problem started because the AI was trained on a "class photo" of the internet that had way more pictures of some groups than others. It's like teaching a student to recognize animals using a textbook that only has pictures of golden retrievers and no pictures of pugs; the student will be confused when they see a pug.
To fix this, scientists have tried a few things. Some tried to "re-tune" the guard's brain after the fact, adjusting the rules for different groups. Others tried to make the training class photo perfectly balanced, giving every group an equal number of pictures. But here's the tricky part: even if you balance the photo album perfectly, the AI can still get confused when it's actually working in the real world, especially when it has to make a "no" decision on a very tricky face. The real test isn't just getting the average right; it's about making sure the AI doesn't fail unfairly when it's on the edge of making a mistake. This is where a new study steps in to try a different approach, one that focuses on how the AI learns to tell similar faces apart without getting tripped up by who they are.
The researchers behind this study, Yu Linghu and their team from the University of Zurich, propose a new way to train these face-recognition systems called DeSCon (Demographic-based Supervised Contrastive Learning). Think of the AI's learning process like a game of musical chairs, but instead of chairs, the players are faces.
Usually, when training an AI, the goal is to make sure all the faces of "Person A" sit close together in a specific corner of the room, while "Person B" sits in a different corner. This is like using a magnet to pull all the "Person A" faces together. However, the old way of doing this sometimes ignored the fact that "Person A" might look a bit like "Person C" if they share a similar skin tone or age, even if they are different people. The AI might get confused and think "Person A" is "Person C" just because they look similar in those specific ways.
The team's new method, DeSCon, adds a special rule to this game. It says: "Okay, pull all the 'Person A' faces together, but also make sure you push the 'Person A' faces away from the 'Person C' faces that look the most like them, specifically within the same demographic group." It's like a strict teacher who not only tells the students to sit with their own friends but also specifically tells the students who look very similar to each other to sit far apart so they don't get mixed up.
The researchers tested three different ways to pick which "look-alike" pairs to push apart:
- The "Everyone" approach: Push apart any two people who aren't the same person, no matter who they are.
- The "Group" approach: Only push apart people who are in the same demographic group (e.g., only worry about pushing apart two different Asian faces, or two different older faces).
- The "Hard Mode" approach (DeSCon-Hard): This is the star of the show. It specifically finds the hardest pairs—the ones that look the most confusingly similar within the same group—and forces the AI to learn the difference between them.
The results suggest that this "Hard Mode" approach works best. When they tested their system on various datasets, including ones that were perfectly balanced and ones that were messy and unbalanced, DeSCon-Hard managed to make the AI fairer without making it worse at its main job. In fact, it often got better at recognizing faces overall while reducing the number of times it made unfair mistakes.
The paper argues against the idea that simply balancing the training data is enough. They show that even with a balanced dataset, the AI can still be biased if it doesn't learn to handle the "hard" cases where different people look very similar. They also found that trying to remove demographic information entirely from the AI's brain (so it can't "see" race or gender) often makes the AI worse at recognizing faces in general. Instead, their method teaches the AI to use those details to be more precise, ensuring that a mistake isn't made just because of who the person is.
In their experiments, the team used standard benchmarks to measure success. They found that their new method, particularly the "Hard Mode" version, consistently reduced the gap in error rates between different groups. For example, on one test, the "Hard Mode" version reduced the unfairness in false matches (saying two different people are the same) significantly more than the standard methods, while keeping the overall accuracy high. They also noted that while their method works well on the datasets they used, the training data they had was smaller than the massive datasets used by big tech companies, so there is still work to be done to see how it scales up.
Ultimately, this paper suggests that to build a truly fair face recognition system, we need to stop just trying to balance the books and start teaching the AI to pay extra attention to the tricky, confusing cases where bias usually hides. By focusing on the "hardest" look-alikes within each group, the AI learns to be fairer and sharper at the same time.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.