
























Abstract:Distribution-to-distribution generative models support scientific imaging tasks ranging from modeling cellular perturbation responses to translating medical images across conditions. Trustworthy generation requires reliability, or generalization across labs, devices, and experimental conditions, and accountability, or detecting out-of-distribution cases where predictions may be unreliable. We leverage Stochastic Flow Matching (SFM), a marginal-preserving stochastic extension of flow matching for improved generalization under distribution shift. SFM augments deterministic flows with a diffusion term together with a learned score-based drift correction, retaining the learned transport marginals while modeling conditional variability. Building on this SFM framework, we introduce Bayesian Stochastic Flow Matching (BSFM) as a companion uncertainty quantification mechanism and develop AVUQ (Antithetic Variance-reduction Uncertainty Quantification) to approximately estimate epistemic and aleatoric uncertainty via sample-efficient antithetic sampling with approximate posterior inference. We further use AVUQ to yield anomaly scores for unreliable generation detection. Experiments on cellular imaging (BBBC021, JUMP) and brain fMRI (Theory of Mind) across diverse unseen scenarios show that SFM improves generalization while AVUQ provides effective uncertainty-based anomaly scores under practical sampling budgets.
| Subjects: | Machine Learning (cs.LG) |
| Cite as: | arXiv:2603.21717 [cs.LG] |
| (or arXiv:2603.21717v4 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2603.21717 arXiv-issued DOI via DataCite |
From: Dongxia Wu [view email]
[v1]
Mon, 23 Mar 2026 09:01:02 UTC (18,358 KB)
[v2]
Tue, 24 Mar 2026 15:25:41 UTC (18,358 KB)
[v3]
Sun, 5 Apr 2026 09:53:03 UTC (18,358 KB)
[v4]
Wed, 20 May 2026 20:53:53 UTC (18,620 KB)
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。