lucidrains / slot-attention
Implementation of Slot Attention from GoogleAI
☆405Updated 4 months ago
Alternatives and similar repositories for slot-attention:
Users that are interested in slot-attention are comparing it to the libraries listed below
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆560Updated 2 years ago
- Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)☆454Updated 2 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆258Updated 3 years ago
- EsViT: Efficient self-supervised Vision Transformers☆409Updated last year
- Pytorch Implementation of paper "Object-Centric Learning with Slot Attention"☆86Updated last year
- NeurIPS 2020, Debiased Contrastive Learning☆282Updated last year
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆480Updated 2 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆193Updated 3 years ago
- Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)☆268Updated 3 years ago
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆418Updated 2 years ago
- Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021 Oral.☆550Updated last year
- ☆156Updated last year
- A Pytorch-Lightning implementation of self-supervised algorithms☆536Updated 2 years ago
- PyTorch implementation of Barlow Twins.☆966Updated 2 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆446Updated last year
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆808Updated 2 years ago
- VICRegL official code base☆224Updated last year
- FiLM: Visual Reasoning with a General Conditioning Layer☆328Updated 3 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆202Updated last year
- Understanding Training Dynamics of Deep ReLU Networks☆284Updated 2 months ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,115Updated last year
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆260Updated 3 years ago
- ☆314Updated 9 months ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆197Updated 8 months ago
- PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"☆576Updated 2 years ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆339Updated 5 months ago
- Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)☆538Updated 5 months ago
- Code to reproduce the results in the FAIR research papers "Semi-Supervised Learning of Visual Features by Non-Parametrically Predicting V…☆487Updated last year
- Probing the representations of Vision Transformers.☆319Updated 2 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆130Updated last year