Saved in:
Bibliographic Details
Main Authors: Zeng, Dewen, Wu, Yawen, Hu, Xinrong, Xu, Xiaowei, Shi, Yiyu
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.16965
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913806846787584
author Zeng, Dewen
Wu, Yawen
Hu, Xinrong
Xu, Xiaowei
Shi, Yiyu
author_facet Zeng, Dewen
Wu, Yawen
Hu, Xinrong
Xu, Xiaowei
Shi, Yiyu
contents Contrastive learning with the nearest neighbor has proved to be one of the most efficient self-supervised learning (SSL) techniques by utilizing the similarity of multiple instances within the same class. However, its efficacy is constrained as the nearest neighbor algorithm primarily identifies "easy" positive pairs, where the representations are already closely located in the embedding space. In this paper, we introduce a novel approach called Contrastive Learning with Synthetic Positives (CLSP) that utilizes synthetic images, generated by an unconditional diffusion model, as the additional positives to help the model learn from diverse positives. Through feature interpolation in the diffusion model sampling process, we generate images with distinct backgrounds yet similar semantic content to the anchor image. These images are considered "hard" positives for the anchor image, and when included as supplementary positives in the contrastive loss, they contribute to a performance improvement of over 2% and 1% in linear evaluation compared to the previous NNCLR and All4One methods across multiple benchmark datasets such as CIFAR10, achieving state-of-the-art methods. On transfer learning benchmarks, CLSP outperforms existing SSL frameworks on 6 out of 8 downstream datasets. We believe CLSP establishes a valuable baseline for future SSL studies incorporating synthetic data in the training process.
format Preprint
id arxiv_https___arxiv_org_abs_2408_16965
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Contrastive Learning with Synthetic Positives
Zeng, Dewen
Wu, Yawen
Hu, Xinrong
Xu, Xiaowei
Shi, Yiyu
Computer Vision and Pattern Recognition
Contrastive learning with the nearest neighbor has proved to be one of the most efficient self-supervised learning (SSL) techniques by utilizing the similarity of multiple instances within the same class. However, its efficacy is constrained as the nearest neighbor algorithm primarily identifies "easy" positive pairs, where the representations are already closely located in the embedding space. In this paper, we introduce a novel approach called Contrastive Learning with Synthetic Positives (CLSP) that utilizes synthetic images, generated by an unconditional diffusion model, as the additional positives to help the model learn from diverse positives. Through feature interpolation in the diffusion model sampling process, we generate images with distinct backgrounds yet similar semantic content to the anchor image. These images are considered "hard" positives for the anchor image, and when included as supplementary positives in the contrastive loss, they contribute to a performance improvement of over 2% and 1% in linear evaluation compared to the previous NNCLR and All4One methods across multiple benchmark datasets such as CIFAR10, achieving state-of-the-art methods. On transfer learning benchmarks, CLSP outperforms existing SSL frameworks on 6 out of 8 downstream datasets. We believe CLSP establishes a valuable baseline for future SSL studies incorporating synthetic data in the training process.
title Contrastive Learning with Synthetic Positives
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.16965