arXiv:cs.LG· Luis A. Ortega, Andr\'es R. Masegosa, Thomas D. Nielsen·· 2 天前AI 评分30
Flow-Transformed Implicit Processes:用归一化流提升函数空间变分推理的表达力
Flow-Transformed Implicit Processes for Function-Space Variational Inference
AI 导读
研究者提出 Flow-Transformed Implicit Processes(FTIP),一种用归一化流替代高斯分布来定义组合权重变分分布的变分推理方法,使隐式过程先验下的函数空间后验近似更具表达力。
正文
Abstract:Implicit-process priors define distributions over functions through flexible generative mechanisms, making them attractive for Bayesian function-space modelling. However, performing posterior inference with such priors is challenging because their induced function-space distributions are typically not available in closed form. One practical strategy is to approximate the prior using a finite collection of sampled functions, and then represent posterior functions as learned combinations of these samples. Existing approaches commonly place a Gaussian variational distribution over the combination weights. While tractable, this choice limits the shapes of posterior uncertainty that can be represented, especially when the true posterior is asymmetric, heavy-tailed, or multimodal. We propose Flow-Transformed Implicit Processes (FTIP), a variational inference method that makes this finite-dimensional function-space approximation more expressive. Instead of using a Gaussian distribution over the combination weights, FTIP uses a normalizing flow to define a richer variational distribution. This induces a flexible posterior distribution over functions while preserving tractable optimization. We train the model using a Black-Box {\alpha} objective, allowing us to compare mass-covering and mode-seeking variational behaviour. Experiments show that FTIP captures asymmetric and multimodal posterior structure in function space that Gaussian coefficient approximations tend to smooth or collapse.
| Comments: | 27 pages, 5 figures, 11 tables. Accepted at the 40th Conference on Neural Information Processing Systems (NeurIPS 2026) |
| Subjects: | Machine Learning (cs.LG); Machine Learning (stat.ML) |
| Cite as: | arXiv:2606.01954 [cs.LG] |
| (or arXiv:2606.01954v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2606.01954 arXiv-issued DOI via DataCite |
Submission history
From: Luis A. Ortega [view email]
[v1]
Mon, 1 Jun 2026 09:14:09 UTC (974 KB)
[v2]
Thu, 1 Oct 2026 08:30:32 UTC (976 KB)
来源:arXiv:cs.LG · arxiv.org