41 citations · 186 across the 21 of their papers we have counts for
9 papers · 1 filter
Unified Neural Scaling Laws
Ethan Caballero, Priyank Jaini, David Krueger +1
We present a functional form (that we refer to as a Unified Neural Scaling Law (UNSL)) that accurately models and extrapolates the scaling behaviors of deep neural networks as mult…
Video models are zero-shot learners and reasoners
Thaddäus Wiedemer, Yuxuan Li, Paul Vicol +6
The remarkable zero-shot capabilities of Large Language Models (LLMs) have propelled natural language processing from task-specific models to unified, generalist foundation models.…
Amortized In-Context Bayesian Posterior Estimation
Sarthak Mittal, Niels Leif Bracher, Guillaume Lajoie +2
Bayesian inference provides a natural way of incorporating prior beliefs and assigning a probability measure to the space of hypotheses. Current solutions rely on iterative routine…
Particle Dynamics for Learning EBMs
Kirill Neklyudov, Priyank Jaini, Max Welling
Energy-based modeling is a promising approach to unsupervised learning, which yields many downstream applications from a single model. The main difficulty in learning energy-based…
Learning Equivariant Energy Based Models with Equivariant Stein Variational Gradient Descent
Priyank Jaini, Lars Holdijk, Max Welling
We focus on the problem of efficient sampling and learning of probability densities by incorporating symmetries in probabilistic models. We first introduce Equivariant Stein Variat…
Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC
Priyank Jaini, Didrik Nielsen, Max Welling
Hybrid Monte Carlo is a powerful Markov Chain Monte Carlo method for sampling from complex continuous distributions. However, a major limitation of HMC is its inability to be appli…
Self Normalizing Flows
T. Anderson Keller, Jorn W. T. Peters, Priyank Jaini +3
Efficient gradient computation of the Jacobian determinant term is a core problem in many machine learning settings, and especially so in the normalizing flow framework. Most propo…
SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows
Didrik Nielsen, Priyank Jaini, Emiel Hoogeboom +2
Normalizing flows and variational autoencoders are powerful generative models that can represent complicated density functions. However, they both impose constraints on the models:…
Sum-of-Squares Polynomial Flow
Priyank Jaini, Kira A. Selby, Yaoliang Yu
Triangular map is a recent construct in probability theory that allows one to transform any source probability density function to any target density function. Based on triangular…