Data Augmentation of Contrastive Learning is Estimating Positive-incentive Noise

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Hongyuan, Xu, Yanchen, Huang, Sida, Li, Xuelong
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911659335876608
author Zhang, Hongyuan
Xu, Yanchen
Huang, Sida
Li, Xuelong
author_facet Zhang, Hongyuan
Xu, Yanchen
Huang, Sida
Li, Xuelong
contents Inspired by the idea of Positive-incentive Noise (Pi-Noise or $π$-Noise) that aims at learning the reliable noise beneficial to tasks, we scientifically investigate the connection between contrastive learning and $π$-noise in this paper. By converting the contrastive loss to an auxiliary Gaussian distribution to quantitatively measure the difficulty of the specific contrastive model under the information theory framework, we properly define the task entropy, the core concept of $π$-noise, of contrastive learning. It is further proved that the predefined data augmentation in the standard contrastive learning paradigm can be regarded as a kind of point estimation of $π$-noise. Inspired by the theoretical study, a framework that develops a $π$-noise generator to learn the beneficial noise (instead of estimation) as data augmentations for contrast is proposed. The designed framework can be applied to diverse types of data and is also completely compatible with the existing contrastive models. From the visualization, we surprisingly find that the proposed method successfully learns effective augmentations. Our code is available at https://github.com/hyzhang98/PiNDA.
format Preprint
id arxiv_https___arxiv_org_abs_2408_09929
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data Augmentation of Contrastive Learning is Estimating Positive-incentive Noise
Zhang, Hongyuan
Xu, Yanchen
Huang, Sida
Li, Xuelong
Machine Learning
Computer Vision and Pattern Recognition
Inspired by the idea of Positive-incentive Noise (Pi-Noise or $π$-Noise) that aims at learning the reliable noise beneficial to tasks, we scientifically investigate the connection between contrastive learning and $π$-noise in this paper. By converting the contrastive loss to an auxiliary Gaussian distribution to quantitatively measure the difficulty of the specific contrastive model under the information theory framework, we properly define the task entropy, the core concept of $π$-noise, of contrastive learning. It is further proved that the predefined data augmentation in the standard contrastive learning paradigm can be regarded as a kind of point estimation of $π$-noise. Inspired by the theoretical study, a framework that develops a $π$-noise generator to learn the beneficial noise (instead of estimation) as data augmentations for contrast is proposed. The designed framework can be applied to diverse types of data and is also completely compatible with the existing contrastive models. From the visualization, we surprisingly find that the proposed method successfully learns effective augmentations. Our code is available at https://github.com/hyzhang98/PiNDA.
title Data Augmentation of Contrastive Learning is Estimating Positive-incentive Noise
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.09929