← Search

Kilian FATRAS

8 accepted papers

2024

Generating and Imputing Tabular Data via Diffusion and Flow-based Gradient-Boosted Trees

AISTATS 2024poster

Tabular data is hard to acquire and is subject to missing values. This paper introduces a novel approach for generating and imputing mixed-type (continuous and categorical) tabular data utilizing score-based diffusion and conditional flow matching. In contrast to prior methods that rely on neural ne…

2024

No Wrong Turns: The Simple Geometry Of Neural Networks Optimization Paths

ICML 2024poster

Understanding the optimization dynamics of neural networks is necessary for closing the gap between theory and practice. Stochastic first-order optimization algorithms are known to efficiently locate favorable minima in deep neural networks. This efficiency, however, contrasts with the non-convex an…

2024

SE(3)-Stochastic Flow Matching for Protein Backbone Generation

ICLR 2024spotlight

The computational design of novel protein structures has the potential to impact numerous scientific disciplines greatly. Toward this goal, we introduce \foldflow, a series of novel generative models of increasing modeling power based on the flow-matching paradigm over $3\mathrm{D}$ rigid motions---…

2024

Sequence-Augmented SE(3)-Flow Matching For Conditional Protein Generation

NeurIPS 2024poster

Proteins are essential for almost all biological processes and derive their diverse functions from complex $3 \rm D$ structures, which are in turn determined by their amino acid sequences. In this paper, we exploit the rich biological inductive bias of amino acid sequences and introduce FoldFlow++,…

Cited by 31SourcePDFScholar
2024

Simulation-Free Schrödinger Bridges via Score and Flow Matching

AISTATS 2024poster

We present simulation-free score and flow matching ([SF]$^2$M), a simulation-free objective for inferring stochastic dynamics given unpaired samples drawn from arbitrary source and target distributions. Our method generalizes both the score-matching loss used in the training of diffusion models and…

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…