iamalexkorotin / Wasserstein2Barycenters
PyTorch implementation of the paper "Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimization" (ICLR 2021)
☆23Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for Wasserstein2Barycenters
- Sinkhorn Barycenters via Frank-Wolfe algorithm☆24Updated 4 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆20Updated 2 months ago
- ☆23Updated 3 years ago
- Distributional and Outlier Robust Optimization (ICML 2021)☆27Updated 3 years ago
- Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness☆42Updated 3 years ago
- Code for the paper "Semi-Conditional Normalizing Flows for Semi-Supervised Learning"☆10Updated 4 years ago
- PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)☆50Updated last year
- ☆13Updated last year
- ☆15Updated last year
- ☆19Updated 4 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 3 years ago
- Kernel Continual Learning☆7Updated 3 years ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- Code for the ICML 2021 and ICLR 2022 papers: Skew Orthogonal Convolutions, Improved deterministic l2 robustness on CIFAR-10 and CIFAR-100☆18Updated 2 years ago
- Wrap around any model to output differentially private prediction sets with finite sample validity on any dataset.☆17Updated 8 months ago
- ☆52Updated 3 months ago
- Diagnosing Vulnerability of Variational Auto-Encoders to Adversarial Attacks☆13Updated 2 years ago
- ☆31Updated 4 years ago
- Simple data balancing baselines for worst-group-accuracy benchmarks.☆40Updated last year
- ☆13Updated 5 years ago
- Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …☆39Updated 4 years ago
- Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J☆65Updated 7 months ago
- Featurized Density Ratio Estimation☆20Updated 3 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆35Updated 2 years ago
- Code for ICLR 2022 Paper, "Controlling Directions Orthogonal to a Classifier"☆34Updated last year
- Code for the paper "Getting a CLUE: A Method for Explaining Uncertainty Estimates"☆36Updated 6 months ago
- Official PyTorch implementation of NPwSA: "Neural Processes with Stochastic Attention: Paying more attention to the context dataset (ICLR…☆10Updated 2 years ago
- Robust Optimal Transport code☆40Updated 2 years ago
- Source code for the ICML2019 paper "Subspace Robust Wasserstein Distances"☆28Updated 5 years ago
- ☆18Updated 2 years ago