All Categories → No Category → contrastive-learning

Top 41 contrastive-learning open source projects

SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners

✭ 2,720

Jupyter Notebook python computer-vision unsupervised-learning representation-learning self-supervised-learning simclr contrastive-learning simclrv2

Pytorch Metric Learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

✭ 3,936

python deep-learning machine-learning pytorch computer-vision image-retrieval metric-learning deep-metric-learning self-supervised-learning contrastive-learning

PIC

Parametric Instance Classification for Unsupervised Visual Feature Learning, NeurIPS 2020

✭ 41

python shell pytorch unsupervised-learning self-supervised-learning contrastive-learning

CONTRIQUE

Official implementation for "Image Quality Assessment using Contrastive Learning"

✭ 33

python deep-neural-networks computer-vision deep-learning image-processing convolutional-neural-networks image-quality image-quality-assessment blind-image-quality-assessment contrastive-loss no-reference full-reference-iqa contrastive-learning

simclr-pytorch

PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results

✭ 89

Jupyter Notebook python deep-learning pytorch representation-learning self-supervised-learning pytorch-implementation contrastive-learning

ContrastiveLearning4Dialogue

The codebase for "Group-wise Contrastive Learning for Neural Dialogue Generation" (Cai et al., Findings of EMNLP 2020)

✭ 54

python javascript HTML dialog pytorch conversational-ai parlai contrastive-learning

GRACE

[GRL+ @ ICML 2020] PyTorch implementation for "Deep Graph Contrastive Representation Learning" (https://arxiv.org/abs/2006.04131v2)

✭ 144

python machine-learning deep-learning graph-representation-learning contrastive-learning

GCA

[WWW 2021] Source code for "Graph Contrastive Learning with Adaptive Augmentation"

✭ 69

python deep-learning pytorch self-supervised-learning graph-representation-learning contrastive-learning graph-contrastive-learning

AdCo

AdCo: Adversarial Contrast for Efficient Learning of Unsupervised Representations from Self-Trained Negative Adversaries

✭ 148

python self-supervised-learning contrastive-learning adco

MOON

Model-Contrastive Federated Learning (CVPR 2021)

✭ 93

python pytorch federated-learning contrastive-learning cvpr2021

cl-ica

Code for the paper "Contrastive Learning Inverts the Data Generating Process".

✭ 65

python Dockerfile shell machine-learning computer-vision contrastive-learning

contrastive loss

Experiments with supervised contrastive learning methods with different loss functions

✭ 143

Jupyter Notebook python deep-learning contrastive-learning

G-SimCLR

This is the code base for paper "G-SimCLR : Self-Supervised Contrastive Learning with Guided Projection via Pseudo Labelling" by Souradip Chakraborty, Aritra Roy Gosthipaty and Sayak Paul.

✭ 69

Jupyter Notebook clustering keras-tensorflow visual-recognition self-supervised-learning tensorflow2 contrastive-learning

SimCLR

Pytorch implementation of "A Simple Framework for Contrastive Learning of Visual Representations"

✭ 65

python pytorch representation-learning unsupervised-learning simclr contrastive-learning

Scon-ABSA

[CIKM 2021] Enhancing Aspect-Based Sentiment Analysis with Supervised Contrastive Learning

✭ 17

HTML python sentiment-analysis aspect-based-sentiment-analysis contrastive-learning

VarCLR

VarCLR: Variable Semantic Representation Pre-training via Contrastive Learning

✭ 30

python embeddings source-code pre-training contrastive-learning icse2022

mirror-bert

[EMNLP 2021] Mirror-BERT: Converting Pretrained Language Models to universal text encoders without labels.

✭ 56

python shell nlp unsupervised-learning bert contrastive-learning emnlp2021

SoCo

[NeurIPS 2021 Spotlight] Aligning Pretraining for Detection via Object-Level Contrastive Learning

✭ 125

python shell Dockerfile object-detection self-supervised-learning contrastive-learning neurips-2021

GCL

List of Publications in Graph Contrastive Learning

✭ 25

graph self-supervised-learning gnn contrastive-learning

HEAPUtil

Code for the RA-L (IROS) 2021 paper "A Hierarchical Dual Model of Environment- and Place-Specific Utility for Visual Place Recognition"

✭ 46

python Jupyter Notebook utility localization deep-learning visual-place-recognition netvlad contrastive-learning

DiGCL

The PyTorch implementation of Directed Graph Contrastive Learning (DiGCL), NeurIPS-2021

✭ 27

python graph directed-graph graph-neural-networks contrastive-learning neurips-2021

CVC

CVC: Contrastive Learning for Non-parallel Voice Conversion (INTERSPEECH 2021, in PyTorch)

✭ 45

python speech pytorch voice-conversion cyclegan contrastive-learning

awesome-graph-self-supervised-learning-based-recommendation

A curated list of awesome graph & self-supervised-learning-based recommendation.

✭ 37

machine-learning deep-learning recommendation-system graph-neural-networks self-supervised-learning knowledge-graph-for-recommendation contrastive-learning graph-based-recommendation

info-nce-pytorch

PyTorch implementation of the InfoNCE loss for self-supervised learning.

✭ 160

python contrastive-loss self-supervised-learning contrastive-learning

GeDML

Generalized Deep Metric Learning.

✭ 30

python computer-vision metric-learning loss-functions deep-metric-learning self-supervised-learning contrastive-learning

TCE

This repository contains the code implementation used in the paper Temporally Coherent Embeddings for Self-Supervised Video Representation Learning (TCE).

✭ 51

python computer-vision deep-learning pytorch embeddings metric-learning representation-learning action-recognition hmdb51 contrastive-loss self-supervised-learning ucf-101 kinetics-datasets contrastive-learning tsne-visualisations

CLMR

Official PyTorch implementation of Contrastive Learning of Musical Representations

✭ 216

python music-information-retrieval music-classification self-supervised-learning contrastive-learning

DisCont

Code for the paper "DisCont: Self-Supervised Visual Attribute Disentanglement using Context Vectors".

✭ 13

python disentanglement center-loss context-vectors disentangled-representations contrastive-loss self-supervised-learning visual-attributes contrastive-learning

ViCC

[WACV'22] Code repository for the paper "Self-supervised Video Representation Learning with Cross-Stream Prototypical Contrasting", https://arxiv.org/abs/2106.10137.

✭ 33

python shell unsupervised-learning action-recognition video-recognition self-supervised-learning video-retrieval contrastive-learning

Parametric-Contrastive-Learning

Parametric Contrastive Learning (ICCV2021)

✭ 155

python Jupyter Notebook shell Dockerfile Batchfile Makefile pytorch supervised-learning imagenet image-classification class-imbalance imbalanced-data imbalanced-learning contrastive-learning supervised-contrastive-learning long-tailed-recognition iccv2021 parametric-contrastive-learning

CCL

PyTorch Implementation on Paper [CVPR2021]Distilling Audio-Visual Knowledge by Compositional Contrastive Learning

✭ 76

python shell pytorch video-recognition distillation audio-visual-learning contrastive-learning cvpr2021 compositional-contrastive-learning audio-teacher-models multi-modal-distillation

S2-BNN

S2-BNN: Bridging the Gap Between Self-Supervised Real and 1-bit Neural Networks via Guided Distribution Calibration (CVPR 2021)

✭ 53

python binary-neural-networks contrastive-loss self-supervised-learning efficient-model contrastive-learning distillation-loss

awesome-efficient-gnn

Code and resources on scalable and efficient Graph Neural Networks

✭ 498

python shell knowledge-distillation graph-convolutional-networks graph-neural-networks graph-representation-learning scalable-deep-learning gnn efficient-neural-networks contrastive-learning efficient-deep-learning

Revisiting-Contrastive-SSL

Revisiting Contrastive Methods for Unsupervised Learning of Visual Representations. [NeurIPS 2021]

✭ 81

python shell clustering transfer-learning representation-learning unsupervised-learning moco self-supervised-learning neurips pretraining contrastive-learning neurips2021

object-aware-contrastive

Object-aware Contrastive Learning for Debiased Scene Representation (NeurIPS 2021)

✭ 44

python representation-learning self-supervised-learning contrastive-learning

CLSA

official implemntation for "Contrastive Learning with Stronger Augmentations"

✭ 48

python shell deep-learning unsupervised-learning self-supervised-learning contrastive-learning clsa stronger-augmentations

RSC-Net

Implementation for "3D human pose, shape and texture from low-resolution images and videos", TPAMI 2021

✭ 43

python shell computer-vision deep-learning 3d weakly-supervised-learning 3d-human-mesh 3d-human-shape-and-pose-estimation contrastive-learning

Supervised-Contrastive-Learning-in-TensorFlow-2

Implements the ideas presented in https://arxiv.org/pdf/2004.11362v1.pdf by Khosla et al.

✭ 117

Jupyter Notebook computer-vision deep-learning tensorflow keras representation-learning contrastive-learning supervised-contrastive-learning

COCO-LM

[NeurIPS 2021] COCO-LM: Correcting and Contrasting Text Sequences for Language Model Pretraining

✭ 109

python shell Cuda C++cython lua natural-language-processing deep-learning transformers representation-learning language-model natural-language-understanding pretraining pretrained-language-model contrastive-learning

SCL

📄 Spatial Contrastive Learning for Few-Shot Classification (ECML/PKDD 2021).

✭ 42

python shell deep-learning image-classification few-shot-learning self-supervised-learning contrastive-learning

ConDigSum

Code for EMNLP 2021 paper "Topic-Aware Contrastive Learning for Abstractive Dialogue Summarization"

✭ 62

python Cuda topic dialogue bart summarization coherence self-supervised contrastive-learning sub-summary

1-41 of 41 contrastive-learning projects