CVAIJan 16, 2024

Achieve Fairness without Demographics for Dermatological Disease Diagnosis

arXiv:2401.08066v116 citationsMedical Image Anal.
AI Analysis

This addresses fairness issues in medical AI for dermatology, which is important for underprivileged populations, but it is incremental as it builds on prior work on feature entanglement.

The paper tackles the problem of achieving fairness in dermatological disease diagnosis without using demographic information during training, by proposing a method that regularizes feature entanglement and uses disease masks to improve fairness and accuracy, showing improved fairness compared to state-of-the-art methods on two datasets.

In medical image diagnosis, fairness has become increasingly crucial. Without bias mitigation, deploying unfair AI would harm the interests of the underprivileged population and potentially tear society apart. Recent research addresses prediction biases in deep learning models concerning demographic groups (e.g., gender, age, and race) by utilizing demographic (sensitive attribute) information during training. However, many sensitive attributes naturally exist in dermatological disease images. If the trained model only targets fairness for a specific attribute, it remains unfair for other attributes. Moreover, training a model that can accommodate multiple sensitive attributes is impractical due to privacy concerns. To overcome this, we propose a method enabling fair predictions for sensitive attributes during the testing phase without using such information during training. Inspired by prior work highlighting the impact of feature entanglement on fairness, we enhance the model features by capturing the features related to the sensitive and target attributes and regularizing the feature entanglement between corresponding classes. This ensures that the model can only classify based on the features related to the target attribute without relying on features associated with sensitive attributes, thereby improving fairness and accuracy. Additionally, we use disease masks from the Segment Anything Model (SAM) to enhance the quality of the learned feature. Experimental results demonstrate that the proposed method can improve fairness in classification compared to state-of-the-art methods in two dermatological disease datasets.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes