CosFairNet:A Parameter-Space based Approach for Bias Free Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dwivedi, Rajeev Ranjan, Kumari, Priyadarshini, Kurmi, Vinod K
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910657868201984
author Dwivedi, Rajeev Ranjan
Kumari, Priyadarshini
Kurmi, Vinod K
author_facet Dwivedi, Rajeev Ranjan
Kumari, Priyadarshini
Kurmi, Vinod K
contents Deep neural networks trained on biased data often inadvertently learn unintended inference rules, particularly when labels are strongly correlated with biased features. Existing bias mitigation methods typically involve either a) predefining bias types and enforcing them as prior knowledge or b) reweighting training samples to emphasize bias-conflicting samples over bias-aligned samples. However, both strategies address bias indirectly in the feature or sample space, with no control over learned weights, making it difficult to control the bias propagation across different layers. Based on this observation, we introduce a novel approach to address bias directly in the model's parameter space, preventing its propagation across layers. Our method involves training two models: a bias model for biased features and a debias model for unbiased details, guided by the bias model. We enforce dissimilarity in the debias model's later layers and similarity in its initial layers with the bias model, ensuring it learns unbiased low-level features without adopting biased high-level abstractions. By incorporating this explicit constraint during training, our approach shows enhanced classification accuracy and debiasing effectiveness across various synthetic and real-world datasets of different sizes. Moreover, the proposed method demonstrates robustness across different bias types and percentages of biased samples in the training data. The code is available at: https://visdomlab.github.io/CosFairNet/
format Preprint
id arxiv_https___arxiv_org_abs_2410_15094
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CosFairNet:A Parameter-Space based Approach for Bias Free Learning
Dwivedi, Rajeev Ranjan
Kumari, Priyadarshini
Kurmi, Vinod K
Computer Vision and Pattern Recognition
Machine Learning
Deep neural networks trained on biased data often inadvertently learn unintended inference rules, particularly when labels are strongly correlated with biased features. Existing bias mitigation methods typically involve either a) predefining bias types and enforcing them as prior knowledge or b) reweighting training samples to emphasize bias-conflicting samples over bias-aligned samples. However, both strategies address bias indirectly in the feature or sample space, with no control over learned weights, making it difficult to control the bias propagation across different layers. Based on this observation, we introduce a novel approach to address bias directly in the model's parameter space, preventing its propagation across layers. Our method involves training two models: a bias model for biased features and a debias model for unbiased details, guided by the bias model. We enforce dissimilarity in the debias model's later layers and similarity in its initial layers with the bias model, ensuring it learns unbiased low-level features without adopting biased high-level abstractions. By incorporating this explicit constraint during training, our approach shows enhanced classification accuracy and debiasing effectiveness across various synthetic and real-world datasets of different sizes. Moreover, the proposed method demonstrates robustness across different bias types and percentages of biased samples in the training data. The code is available at: https://visdomlab.github.io/CosFairNet/
title CosFairNet:A Parameter-Space based Approach for Bias Free Learning
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2410.15094