contrastive-learning

There are 19 repositories under contrastive-learning topic.

pytorch-metric-learning
KevinMusgrave / pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
metric-learning deep-learning computer-vision machine-learning pytorch deep-metric-learning image-retrieval self-supervised-learning contrastive-learning
Language:Python 5160
google-research / simclr
SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners
simclr contrastive-learning representation-learning self-supervised-learning unsupervised-learning computer-vision simclrv2
Language:Jupyter Notebook 3442
lightly-ai / lightly
A python library for self-supervised learning on images.
deep-learning self-supervised-learning machine-learning computer-vision pytorch embeddings contrastive-learning active-learning
Language:Python 2215
Unsupervised-Classification
wvangansbeke / Unsupervised-Classification
SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]
unsupervised-learning image-classification self-supervised-learning clustering eccv-2020 eccv2020 representation-learning contrastive-learning simclr moco
Language:Python 1156
vturrisi / solo-learn
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
barlow-twins byol contrastive-learning deepcluster dino mae masked-input-prediction moco nnclr nvidia-dali pytorch pytorch-lightning ressl self-supervised-learning simclr simsiam swav transformer-models vibcreg vicreg
Language:Python 1145
tensorflow / similarity
TensorFlow Similarity is a python package focused on making similarity learning quick and easy.
similarity-learning metric-learning similarity-search nearest-neighbor-search nearest-neighbors deep-learning tensorflow contrastive-learning machine-learning unsupervised-learning clustering python simclr simclr2 barlow-twins simsiam cosine-similarity knn
Language:Python 935
JosephKJ / OWOD
(CVPR 2021 Oral) Open World Object Detection
open-world object-detection contrastive-learning energy-based-model detectron2 incremental-learning continual-learning cvpr2021 cvpr
Language:Python 933
salesforce / ALBEF
Code for ALBEF: a new vision-language pre-training method
vision-and-language representation-learning image-text weakly-supervised-learning contrastive-learning
Language:Python 931
lucidrains / CoCa-pytorch
Implementation of CoCa, Contrastive Captioners are Image-Text Foundation Models, in Pytorch
artificial-intelligence attention-mechanism contrastive-learning deep-learning multimodal transformers image-to-text
Language:Python 675
PyGCL / PyGCL
PyGCL: A PyTorch Library for Graph Contrastive Learning
contrastive-learning graph-representation-learning machine-learning graph-contrastive-learning
Language:Python 636
yzhuoning / Awesome-CLIP
Awesome list for research on CLIP (Contrastive Language-Image Pre-Training).
clip contrastive-learning pre-training
622
SimCLR
Spijkervet / SimCLR
PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
simclr pytorch contrastive-learning unsupervised-learning representation-learning
Language:Python 590
lucidrains / x-clip
A concise but complete implementation of CLIP with various experimental improvements from recent papers
artificial-intelligence deep-learning contrastive-learning zero-shot-learning multi-modal-learning
Language:Python 549
tfzhou / ContrastiveSeg
ICCV2021 (Oral) - Exploring Cross-Image Pixel Contrast for Semantic Segmentation
semantic-segmentation contrastive-learning pixel-contrast hard-example-mining cityscapes pascal-context
Language:Python 533
MishaLaskin / curl
CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning
deep-learning contrastive-loss contrastive-learning contrastive-predictive-coding deep-reinforcement-learning deep-rl reinforcement-learning reinforcement-learning-algorithms curl sac gpu off-policy model-free-rl deep-neural-networks deeplearning deep-q-network deep-q-learning deep-learning-algorithms deeplearning-ai reinforcement-agents
Language:Python 511
chaitjo / efficient-gnns
Code and resources on scalable and efficient Graph Neural Networks
contrastive-learning efficient-deep-learning efficient-neural-networks gnn graph-convolutional-networks graph-neural-networks graph-representation-learning knowledge-distillation scalable-deep-learning
Language:Python 509
salesforce / PCL
PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"
representation-learning self-supervised-learning unsupervsied-learning contrastive-learning pre-trained-model
Language:Python 454
quaterion
qdrant / quaterion
Blazing fast framework for fine-tuning similarity learning models
contrastive-learning cosine-similarity deep-learning knn machine-learning metric-learning nearest-neighbor-search python pytorch pytorch-lightning similarity-learning similarity-search
Language:Python 449
Shen-Lab / GraphCL
[NeurIPS 2020] "Graph Contrastive Learning with Augmentations" by Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, Yang Shen
graph-neural-network contrastive-learning self-supervised-learning pre-training
Language:Python 406
Westlake-AI / openmixup
CAIRI Supervised, Semi- and Self-Supervised Visual Representation Learning Toolbox and Benchmark
pytorch awesome-list awesome-mim awesome-mixup contrastive-learning data-augmentation image-classifcation imagenet masked-image-modeling mixup self-supervised-learning semi-supervised-learning vision-transformer deep-learning benchmark automix
Language:Python 388
yxuansu / SimCTG
[NeurIPS'22 Spotlight] A Contrastive Framework for Neural Text Generation
textgeneration contrastive-learning nlp languagemodel decode language-modeling deeplearning
Language:Python 387
JohnGiorgi / DeCLUTR
The corresponding code from our paper "DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations". Do not hesitate to open an issue if you run into any trouble!
contrastive-learning natural-language-processing allennlp pytorch transformers representation-learning sentence-embeddings sentence-similarity semantic-search semantic-text-similarity metric-learning self-supervised-learning
Language:Python 357
Unsupervised-Semantic-Segmentation
wvangansbeke / Unsupervised-Semantic-Segmentation
Unsupervised Semantic Segmentation by Contrasting Object Mask Proposals. [ICCV 2021]
self-supervised-learning representation-learning clustering unsupervised-learning moco contrastive-learning semantic-segmentation pascal-voc iccv2021 iccv unsupervised-semantic-segmentation
Language:Python 346
ryanzhumich / Contrastive-Learning-NLP-Papers
Paper List for Contrastive Learning for Natural Language Processing
contrastive-learning natural-language-processing machine-learning metric-learning representation-learning
341
AngeLouCN / Min_Max_Similarity
A contrastive learning based semi-supervised segmentation network for medical image segmentation
contrastive-learning semi-supervised-learning medical-image-segmentation surgical-tools-segmentation medical-video-analysis video-segmentation
Language:Python 328
SXKDZ / awesome-self-supervised-learning-for-graphs
A curated list for awesome self-supervised learning for graphs.
self-supervised-learning graph-representation-learning machine-learning deep-learning contrastive-learning
323
Haochen-Wang409 / U2PL
[CVPR'22] Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels
mechine-learning computer-vision semantic-segmentation semi-supervised-learning contrastive-learning cvpr2022 pytorch cityscapes pascal-voc deep-learning
Language:Python 316
zdaxie / PixPro
Propagate Yourself: Exploring Pixel-Level Consistency for Unsupervised Visual Representation Learning, CVPR 2021
cvpr2021 unsupervised-learning pixpro contrastive-learning pixel-level-contrastive-learning
Language:Python 308
GCC
THUDM / GCC
GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training @ KDD 2020
contrastive-learning graph-neural-networks pre-training
Language:Python 279
ContrastiveSR / Contrastive_Learning_Papers
A list of contrastive Learning papers
self-supervised-learning contrastive-learning deep-learning computer-vision graph natural-language-processing natural-language-understanding research-paper
263
xyupeng / ContrastiveCrop
[CVPR 2022 Oral] Crafting Better Contrastive Views for Siamese Representation Learning
pytorch self-supervised-learning contrastive-learning
Language:Python 254
tangxyw / RecSysPapers
推荐/广告/搜索领域工业界经典以及最前沿论文集合。A collection of industry classics and cutting-edge papers in the field of recommendation/advertising/search.
calibration causal-inference cold-start contrastive-learning debias distillation diverse fairness match papers rank re-rank recommendation-system feedback-delay pre-rank look-alike multi-scenario multi-task reinforcement-learning multi-modal
Language:Python 248
alinlab / CSI
CSI: Novelty Detection via Contrastive Learning on Distributionally Shifted Instances (NeurIPS 2020)
novelty-detection anomaly-detection out-of-distribution-detection contrastive-learning
Language:Python 246
Spijkervet / CLMR
Official PyTorch implementation of Contrastive Learning of Musical Representations
contrastive-learning music-classification music-information-retrieval self-supervised-learning
Language:Python 245
RElbers / info-nce-pytorch
PyTorch implementation of the InfoNCE loss for self-supervised learning.
contrastive-learning contrastive-loss self-supervised-learning
Language:Python 244
voidism / DiffCSE
Code for the NAACL 2022 long paper "DiffCSE: Difference-based Contrastive Learning for Sentence Embeddings"
contrastive-learning representation-learning self-supervised-learning sentence-embeddings sentence-similarity sentence-transformers
Language:Python 244