← Search

Jens Lagergren

8 accepted papers

2026

Multi-Marginal Flow Matching with Adversarially Learnt Interpolants

ICLR 2026poster

Learning the dynamics of a process given sampled observations at several time points is an important but difficult task in many scientific applications. When no ground-truth trajectories are available, but one has only snapshots of data taken at discrete time steps, the problem of modelling the dyna…

Cited by 0SourcecodeScholar
2024

Benefits of Non-Linear Scale Parameterizations in Black Box Variational Inference through Smoothness Results and Gradient Variance Bounds

AISTATS 2024poster

Black box variational inference has consistently produced impressive empirical results. Convergence guarantees require that the variational objective exhibits specific structural properties and that the noise of the gradient estimator can be controlled. In this work we study the smoothness and the v…

Cited by 3SourcePDFScholar
2024

Efficient Mixture Learning in Black-Box Variational Inference

ICML 2024poster

Mixture variational distributions in black box variational inference (BBVI) have demonstrated impressive results in challenging density estimation tasks. However, currently scaling the number of mixture components can lead to a linear increase in the number of learnable parameters and a quadratic in…

2024

Indirectly Parameterized Concrete Autoencoders

ICML 2024poster

Feature selection is a crucial task in settings where data is high-dimensional or acquiring the full set of features is costly. Recent developments in neural network-based embedded feature selection show promising results across a wide range of applications. Concrete Autoencoders (CAEs), considered…

2023

Cooperation in the Latent Space: The Benefits of Adding Mixture Components in Variational Autoencoders

ICML 2023poster

In this paper, we show how the mixture components cooperate when they jointly adapt to maximize the ELBO. We build upon recent advances in the multiple and adaptive importance sampling literature. We then model the mixture components using separate encoder networks and show empirically that the ELBO…

2022

Multiple Importance Sampling ELBO and Deep Ensembles of Variational Approximations

AISTATS 2022poster

In variational inference (VI), the marginal log-likelihood is estimated using the standard evidence lower bound (ELBO), or improved versions as the importance weighted ELBO (IWELBO). We propose the multiple importance sampling ELBO (MISELBO), a versatile yet simple framework. MISELBO is applicable i…

2022

VaiPhy: a Variational Inference Based Algorithm for Phylogeny

NeurIPS 2022accept

Phylogenetics is a classical methodology in computational biology that today has become highly relevant for medical investigation of single-cell data, e.g., in the context of development of cancer. The exponential size of the tree space is unfortunately a formidable obstacle for current Bayesian ph…