Discovering Global False Negatives On the Fly for Self-supervised Contrastive Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Balmaseda, Vicente, Wang, Bokun, Lin, Ching-Long, Yang, Tianbao
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908422154223616
author Balmaseda, Vicente
Wang, Bokun
Lin, Ching-Long
Yang, Tianbao
author_facet Balmaseda, Vicente
Wang, Bokun
Lin, Ching-Long
Yang, Tianbao
contents In self-supervised contrastive learning, negative pairs are typically constructed using an anchor image and a sample drawn from the entire dataset, excluding the anchor. However, this approach can result in the creation of negative pairs with similar semantics, referred to as "false negatives", leading to their embeddings being falsely pushed apart. To address this issue, we introduce GloFND, an optimization-based approach that automatically learns on the fly the threshold for each anchor data to identify its false negatives during training. In contrast to previous methods for false negative discovery, our approach globally detects false negatives across the entire dataset rather than locally within the mini-batch. Moreover, its per-iteration computation cost remains independent of the dataset size. Experimental results on image and image-text data demonstrate the effectiveness of the proposed method. Our implementation is available at https://github.com/vibalcam/GloFND.
format Preprint
id arxiv_https___arxiv_org_abs_2502_20612
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Discovering Global False Negatives On the Fly for Self-supervised Contrastive Learning
Balmaseda, Vicente
Wang, Bokun
Lin, Ching-Long
Yang, Tianbao
Machine Learning
Computer Vision and Pattern Recognition
In self-supervised contrastive learning, negative pairs are typically constructed using an anchor image and a sample drawn from the entire dataset, excluding the anchor. However, this approach can result in the creation of negative pairs with similar semantics, referred to as "false negatives", leading to their embeddings being falsely pushed apart. To address this issue, we introduce GloFND, an optimization-based approach that automatically learns on the fly the threshold for each anchor data to identify its false negatives during training. In contrast to previous methods for false negative discovery, our approach globally detects false negatives across the entire dataset rather than locally within the mini-batch. Moreover, its per-iteration computation cost remains independent of the dataset size. Experimental results on image and image-text data demonstrate the effectiveness of the proposed method. Our implementation is available at https://github.com/vibalcam/GloFND.
title Discovering Global False Negatives On the Fly for Self-supervised Contrastive Learning
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.20612