Mathematics of AI - Transformer Networks

Linearized Partial Gromov Wasserstein

Partial Gromov-Wasserstein Metric

Source code for the ICML 2024 Spotlight paper "Stereographic Spherical Sliced Wasserstein Distances"

Out-of-distribution detection, robustness, and generalization resources. The repository contains a curated list of papers, tutorials, books, videos, articles and open-source libraries etc

TopoTrans: Optimal Transport meets Topological Data Analysis

Set Locality Sensitive Hashing via Sliced Wasserstein Embedding

The implementation code for our paper Wasserstein Embedding for Graph Learning (ICLR 2021).

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

Python Library for Signal/Image Data Analysis with Transport Methods

Rep the Set: Neural Networks for Learning Set Representations

Rep the Set: Neural Networks for Learning Set Representations

A curated list of awesome computer vision resources

Totally Versatile Miscellanea for Pytorch

Code for "Explainability methods for graph convolutional neural networks" - PE Pope*, S Kolouri*, M Rostami, CE Martin, H Hoffmann (CVPR 2019)

Official Repository for the CVPR 2020 paper "Universal Litmus Patterns: Revealing Backdoor Attacks in CNNs"

GAT: Generative Adversarial Training for Adversarial Example Detection and Robust Classification

GAT: Generative Adversarial Training for Adversarial Example Detection and Robust Classification

GAT: Generative Adversarial Training for Adversarial Example Detection and Robust Classification

GAT: Generative Adversarial Training for Adversarial Example Detection and Robust Classification

Sliced Wasserstein GMM using automatic differentiation

Random Fourier Features

An implementation of EWC with PyTorch

Visualizing how deep networks make decisions