Provable Optimization for Adversarial Fair Self-supervised Contrastive Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Qi, Qi, Hu, Quanqi, Lin, Qihang, Yang, Tianbao
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929378866233344
author Qi, Qi
Hu, Quanqi
Lin, Qihang
Yang, Tianbao
author_facet Qi, Qi
Hu, Quanqi
Lin, Qihang
Yang, Tianbao
contents This paper studies learning fair encoders in a self-supervised learning (SSL) setting, in which all data are unlabeled and only a small portion of them are annotated with sensitive attribute. Adversarial fair representation learning is well suited for this scenario by minimizing a contrastive loss over unlabeled data while maximizing an adversarial loss of predicting the sensitive attribute over the data with sensitive attribute. Nevertheless, optimizing adversarial fair representation learning presents significant challenges due to solving a non-convex non-concave minimax game. The complexity deepens when incorporating a global contrastive loss that contrasts each anchor data point against all other examples. A central question is ``{\it can we design a provable yet efficient algorithm for solving adversarial fair self-supervised contrastive learning}?'' Building on advanced optimization techniques, we propose a stochastic algorithm dubbed SoFCLR with a convergence analysis under reasonable conditions without requring a large batch size. We conduct extensive experiments to demonstrate the effectiveness of the proposed approach for downstream classification with eight fairness notions.
format Preprint
id arxiv_https___arxiv_org_abs_2406_05686
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Provable Optimization for Adversarial Fair Self-supervised Contrastive Learning
Qi, Qi
Hu, Quanqi
Lin, Qihang
Yang, Tianbao
Machine Learning
Computer Vision and Pattern Recognition
Computers and Society
This paper studies learning fair encoders in a self-supervised learning (SSL) setting, in which all data are unlabeled and only a small portion of them are annotated with sensitive attribute. Adversarial fair representation learning is well suited for this scenario by minimizing a contrastive loss over unlabeled data while maximizing an adversarial loss of predicting the sensitive attribute over the data with sensitive attribute. Nevertheless, optimizing adversarial fair representation learning presents significant challenges due to solving a non-convex non-concave minimax game. The complexity deepens when incorporating a global contrastive loss that contrasts each anchor data point against all other examples. A central question is ``{\it can we design a provable yet efficient algorithm for solving adversarial fair self-supervised contrastive learning}?'' Building on advanced optimization techniques, we propose a stochastic algorithm dubbed SoFCLR with a convergence analysis under reasonable conditions without requring a large batch size. We conduct extensive experiments to demonstrate the effectiveness of the proposed approach for downstream classification with eight fairness notions.
title Provable Optimization for Adversarial Fair Self-supervised Contrastive Learning
topic Machine Learning
Computer Vision and Pattern Recognition
Computers and Society
url https://arxiv.org/abs/2406.05686