← Search

Rémi Flamary

27 accepted papers

2026

A Spectral-Grassmann Wasserstein metric for operator representations of dynamical systems

ICLR 2026poster

The geometry of dynamical systems estimated from trajectory data is a major challenge for machine learning applications. Koopman and transfer operators provide a linear representation of nonlinear dynamics through their spectral decomposition, offering a natural framework for comparison. We propose…

Cited by 0SourcecodeScholar
2026

Neural Optimal Transport Meets Multivariate Conformal Prediction

ICLR 2026poster

We propose a framework for conditional vector quantile regression (CVQR) that combines neural optimal transport with amortized optimization, and apply it to multivariate conformal prediction. Classical quantile regression does not extend naturally to multivariate responses, while existing approaches…

Cited by 0SourceScholar
2026

PSDNorm: Temporal Normalization for Deep Learning in Sleep Staging

ICLR 2026poster

Distribution shift poses a significant challenge in machine learning, particularly in biomedical applications using data collected across different subjects, institutions, and recording devices, such as sleep data. While existing normalization layers, BatchNorm, LayerNorm and InstanceNorm, h…

Cited by 0SourcecodeScholar
2025

The quest for the GRAph Level autoEncoder (GRALE)

NeurIPS 2025poster

Although graph-based learning has attracted a lot of attention, graph representation learning is still a challenging task whose resolution may impact key application fields such as chemistry or biology. To this end, we introduce GRALE, a novel graph autoencoder that encodes and decodes graphs of var…

Cited by 0SourceScholar
2025

Unsupervised Learning for Optimal Transport plan prediction between unbalanced graphs

NeurIPS 2025poster

Optimal transport between graphs, based on Gromov-Wasserstein and other extensions, is a powerful tool for comparing and aligning graph structures. However, solving the associated non-convex optimization problems is computationally expensive, which limits the scalability of these methods to…

Cited by 0SourceScholar
2024

Any2Graph: Deep End-To-End Supervised Graph Prediction With An Optimal Transport Loss

NeurIPS 2024spotlight

We propose Any2graph, a generic framework for end-to-end Supervised Graph Prediction (SGP) i.e. a deep learning model that predicts an entire graph for any kind of input. The framework is built on a novel Optimal Transport loss, the Partially-Masked Fused Gromov-Wasserstein, that exhibits all necess…

2023

Convolution Monge Mapping Normalization for learning on sleep data

NeurIPS 2023poster

In many machine learning applications on signals and biomedical data, especially electroencephalogram (EEG), one major challenge is the variability of the data across subjects, sessions, and hardware devices. In this work, we propose a new method called Convolutional Monge Mapping Normalization ($\t…

Cited by 6SourcePDFScholar
2023

SNEkhorn: Dimension Reduction with Symmetric Entropic Affinities

NeurIPS 2023poster

Many approaches in machine learning rely on a weighted graph to encode the similarities between samples in a dataset. Entropic affinities (EAs), which are notably used in the popular Dimensionality Reduction (DR) algorithm t-SNE, are particular instances of such graphs. To ensure robustness to heter…

2023

Unbalanced CO-optimal Transport

AAAI 2023technical

Optimal transport (OT) compares probability distributions by computing a meaningful alignment between their samples. CO-optimal transport (COOT) takes this comparison further by inferring an alignment between features as well. While this approach leads to better alignments and generalizes both OT an…

Cited by 22SourcePDFScholar
2022

Aligning individual brains with fused unbalanced Gromov Wasserstein

NeurIPS 2022accept

Individual brains vary in both anatomy and functional organization, even within a given species. Inter-individual variability is a major impediment when trying to draw generalizable conclusions from neuroimaging data collected on groups of subjects. Current co-registration procedures rely on limited…

2022

Convergent Working Set Algorithm for Lasso with Non-Convex Sparse Regularizers

AISTATS 2022poster

Non-convex sparse regularizers are common tools for learning with high-dimensional data. For accelerating convergence for Lasso problem involving those regularizers, a working set strategy addresses the optimization problem through an iterative algorithm by gradually incrementing the number of varia…

2022

Learning to Predict Graphs with Fused Gromov-Wasserstein Barycenters

ICML 2022spotlight

This paper introduces a novel and generic framework to solve the flagship task of supervised labeled graph prediction by leveraging Optimal Transport tools. We formulate the problem as regression with the Fused Gromov-Wasserstein (FGW) loss and propose a predictive model relying on a FGW barycenter…

2022

Multi-source domain adaptation via weighted joint distributions optimal transport

UAI 2022poster

This work addresses the problem of domain adaptation on an unlabeled target dataset using knowledge from multiple labelled source datasets. Most current approaches tackle this problem by searching for an embedding that is invariant across source and target domains, which corresponds to searching for…

Cited by 47SourcePDFScholar
2022

Semi-relaxed Gromov-Wasserstein divergence and applications on graphs

ICLR 2022poster

Comparing structured objects such as graphs is a fundamental operation involved in many learning tasks. To this end, the Gromov-Wasserstein (GW) distance, based on Optimal Transport (OT), has proven to be successful in handling the specific nature of the associated objects. More specifically, throug…

Cited by 52SourcePDFScholar
2022

Template based Graph Neural Network with Optimal Transport Distances

NeurIPS 2022accept

Current Graph Neural Networks (GNN) architectures generally rely on two important components: node features embedding through message passing, and aggregation with a specialized form of pooling. The structural (or topological) information is implicitly taken into account in these two steps. We propo…

2021

Online Graph Dictionary Learning

ICML 2021spotlight

Dictionary learning is a key tool for representation learning, that explains the data as linear combination of few basic elements. Yet, this analysis is not amenable in the context of graph learning, as graphs usually belong to different metric spaces. We fill this gap by proposing a new online Grap…

2021

Unbalanced Optimal Transport through Non-negative Penalized Linear Regression

NeurIPS 2021poster

This paper addresses the problem of Unbalanced Optimal Transport (UOT) in which the marginal conditions are relaxed (using weighted penalties in lieu of equality) and no additional regularization is enforced on the OT plan. In this context, we show that the corresponding optimization problem can be…

Cited by 61SourcePDFScholar
2021

Unbalanced minibatch Optimal Transport; applications to Domain Adaptation

ICML 2021spotlight

Optimal transport distances have found many applications in machine learning for their capacity to compare non-parametric probability distributions. Yet their algorithmic complexity generally prevents their direct use on large scale datasets. Among the possible strategies to alleviate this issue, pr…

2020

Learning with minibatch Wasserstein : asymptotic and gradient properties

AISTATS 2020poster

Optimal transport distances are powerful tools to compare probability distributions and have found many applications in machine learning. Yet their algorithmic complexity prevents their direct use on large scale datasets. To overcome this challenge, practitioners compute these distances on minibatch…

2019

Optimal Transport for Multi-source Domain Adaptation under Target Shift

AISTATS 2019poster

In this paper, we tackle the problem of reducing discrepancies between multiple domains, i.e. multi-source domain adaptation, and consider it under the target shift assumption: in all domains we aim to solve a classification problem with the same output classes, but with different labels proportions…

2019

Optimal Transport for structured data with application on graphs

ICML 2019oral

This work considers the problem of computing distances between structured objects such as undirected graphs, seen as probability distributions in a specific metric space. We consider a new transportation distance ( i.e. that minimizes a total cost of transporting probability masses) that unveils the…

Cited by 207SourcePDFScholar
2019

Sliced Gromov-Wasserstein

NeurIPS 2019poster

Recently used in various machine learning contexts, the Gromov-Wasserstein distance (GW) allows for comparing distributions whose supports do not necessarily lie in the same metric space. However, this Optimal Transport (OT) distance requires solving a complex non convex quadratic program which is…

2017

Joint distribution optimal transportation for domain adaptation

NeurIPS 2017poster

This paper deals with the unsupervised domain adaptation problem, where one wants to estimate a prediction function $f$ in a given target domain without any labeled sample by exploiting the knowledge available from a source domain where labels are known. Our work makes the following assumption: ther…

2016

Mapping Estimation for Discrete Optimal Transport

NeurIPS 2016poster

We are interested in the computation of the transport map of an Optimal Transport problem. Most of the computational approaches of Optimal Transport use the Kantorovich relaxation of the problem to learn a probabilistic coupling $\mgamma$ but do not address the problem of learning the underlying tra…

Cited by 150SourcePDFScholar
2016

Optimal spectral transportation with application to music transcription

NeurIPS 2016poster

Many spectral unmixing methods rely on the non-negative decomposition of spectral data onto a dictionary of spectral templates. In particular, state-of-the-art music transcription systems decompose the spectrogram of the input signal onto a dictionary of representative note spectra. The typical meas…