apple / ml-probabilistic-attention
☆28Updated 2 years ago
Related projects: ⓘ
- ☆44Updated last month
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆48Updated 3 months ago
- Making Heads or Tails Towards Semantically Consistent Visual Counterfactuals☆30Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆74Updated 2 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- ☆34Updated last year
- PyTorch implementation of FNet: Mixing Tokens with Fourier transforms☆25Updated 3 years ago
- ☆18Updated 2 years ago
- Official implementation of the paper "Provable Stochastic Optimization for Global Contrastive Learning: Small Batch Does Not Harm Perform…☆19Updated last year
- ☆26Updated last year
- Github code for the paper Maximum Class Separation as Inductive Bias in One Matrix. Arxiv link: https://arxiv.org/abs/2206.08704☆27Updated last year
- Code for "Don't trust your eyes: on the (un)reliability of feature visualizations"☆30Updated 10 months ago
- Visual Representation Learning Benchmark for Self-Supervised Models☆35Updated 5 months ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆23Updated 2 years ago
- Official code for the paper: "Metadata Archaeology"☆18Updated last year
- Directed masked autoencoders☆13Updated last year
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆54Updated last year
- State-of-the-art pretrained vision model from Bing Multimedia☆18Updated 11 months ago
- Code for the paper Self-Supervised Learning of Split Invariant Equivariant Representations☆24Updated last year
- Code for paper "Can contrastive learning avoid shortcut solutions?" NeurIPS 2021.☆47Updated 2 years ago
- Research publication code for "Forward Compatible Training for Large-Scale Embedding Retrieval Systems", CVPR 2022, and "FastFill: Effici…☆52Updated last year
- A Pytorch implementation of Attention on Attention module (both self and guided variants), for Visual Question Answering☆40Updated 3 years ago
- ICML 2020, Estimating Generalization under Distribution Shifts via Domain-Invariant Representations☆21Updated 4 years ago
- We investigated corruption robustness across different architectures including Convolutional Neural Networks, Vision Transformers, and th…☆15Updated 2 years ago
- The project is about predicting sets (of classes) from images.☆22Updated 3 years ago
- ☆74Updated 2 years ago
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 5 months ago
- Improving Transformation Invariance in Contrastive Representation Learning☆13Updated 3 years ago
- A Benchmark for Efficient and Compositional Visual Reasoning☆17Updated last year
- [ICLR2024] (EvALign-ICL Benchmark) Beyond Task Performance: Evaluating and Reducing the Flaws of Large Multimodal Models with In-Context …☆20Updated 6 months ago