collaborators

21 papers

cs.LG2026

Warm-Starting Iterative Gaussian Processes for Faster Sequential Inference

Alan Yufei Dong, Jihao Andreas Lin, José Miguel Hernández-Lobato

Efficient Gaussian process (GP) inference is critical for sequential decision-making tasks such as active learning, online prediction, and Bayesian optimization. Iterative approach…

cs.LG2026

Compression as Adaptation: Implicit Visual Representation with Diffusion Foundation Models

Zongyu Guo, Jiajun He, Zhaoyang Jia +6

Modern visual generative models acquire rich visual knowledge through large-scale training, yet existing visual representations (such as pixels, latents, or tokens) remain external…

cs.LG2026

Richer Bayesian Last Layers with Subsampled NTK Features

Sergio Calvo-Ordoñez, Jonathan Plenk, Richard Bergna +4

Bayesian Last Layers (BLLs) provide a convenient and computationally efficient way to estimate uncertainty in neural networks. However, they underestimate epistemic uncertainty bec…

stat.ML2026

A Diffusive Classification Loss for Learning Energy-based Generative Models

RuiKang OuYang, Louis Grenioux, José Miguel Hernández-Lobato

Score-based generative models have recently achieved remarkable success. While they are usually parameterized by the score, an alternative way is to use a series of time-dependent…

cs.LG2026

RNE: plug-and-play diffusion inference-time control and energy-based training

Jiajun He, José Miguel Hernández-Lobato, Yuanqi Du +1

Diffusion models generate data by removing noise gradually, which corresponds to the time-reversal of a noising process. However, access to only the denoising kernels is often insu…

cs.LG2026

Wiener Chaos Expansion based Neural Operator for Singular Stochastic Partial Differential Equations

Dai Shi, Luke Thompson, Andi Han +3

In this paper, we explore how our recently developed Wiener Chaos Expansion (WCE)-based neural operator (NO) can be applied to singular stochastic partial differential equations, e…