Felix-Petersen / difftopkLinks
Differentiable Top-k Classification Learning
☆83Updated 2 years ago
Alternatives and similar repositories for difftopk
Users that are interested in difftopk are comparing it to the libraries listed below
Sorting:
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆89Updated last year
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆41Updated 2 years ago
- Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding☆51Updated 9 months ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 3 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆37Updated last year
- ☆58Updated 2 years ago
- Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.☆51Updated 2 years ago
- Official implementation of ORCA proposed in the paper "Cross-Modal Fine-Tuning: Align then Refine"☆71Updated last year
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆143Updated 2 years ago
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆151Updated 2 years ago
- Meta Optimal Transport☆103Updated 2 years ago
- PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS☆30Updated 5 years ago
- Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.☆77Updated 2 years ago
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated last year
- ☆109Updated 2 years ago
- Differentiable Sorting Networks☆116Updated last year
- Deep Learning & Information Bottleneck☆61Updated 2 years ago
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- Natural Language Descriptions of Deep Visual Features, ICLR 2022☆65Updated 2 years ago
- This repository contains the code for our paper "Probabilistic Contrastive Learning Recovers the Correct Aleatoric Uncertainty of Ambiguo…☆41Updated 2 years ago
- LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS IMPLEMENTATION WITH PYTORCH☆79Updated 2 years ago
- [NeurIPS 2021] A Geometric Analysis of Neural Collapse with Unconstrained Features☆57Updated 2 years ago
- Code for "Generalised Implicit Neural Representations" (NeurIPS 2022).☆71Updated 2 years ago
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆69Updated last year
- A Pytorch implementation of the optimal transport kernel embedding☆117Updated 4 years ago
- Code for paper Fully Hyperbolic Neural Networks☆79Updated 2 years ago
- Official Pytorch's implementation of RotoGrad☆79Updated last year
- Differentiable Optimizers with Perturbations in Pytorch☆68Updated 4 years ago
- Code for paper "Can contrastive learning avoid shortcut solutions?" NeurIPS 2021.☆47Updated 3 years ago