arXiv:2502.11333cs.LGcs.AI2025-02ICML被引 5

提出逆流框架,让生成模型能无真实数据时完成去噪等逆问题。

Inverse Flow and Consistency Models

  • 设计逆流匹配与逆一致性模型,支持任意连续噪声分布的逆向生成
  • 在合成与真实数据上超越已有方法,可处理此前无法支持的噪声类型
  • 适用于荧光显微与单细胞基因组等科学场景,推动生成模型在逆问题中的应用

逆向生成问题(如无真实观测下的去噪)是众多科学探索和实际应用中的关键挑战。尽管扩散模型、条件流匹配和一致性模型等生成模型通过将生成建模为去噪任务取得了显著进展,但这些方法在缺乏干净数据时无法直接用于逆向生成。本文提出逆流(Inverse Flow, IF)框架,使这类生成模型可用于无真实数据的逆向生成问题,包括无真值去噪。逆流可灵活适配几乎任意连续噪声分布,并支持复杂依赖关系。我们提出了两种学习逆流的算法:逆流匹配(IFM)和逆一致性模型(ICM)。值得注意的是,为获得计算高效且无需模拟的逆一致性模型目标,我们将一致性训练推广至任意前向扩散过程或条件流,拓展了其应用范围。我们在合成与真实数据集上验证了该方法的有效性,性能优于先前方法,并支持以往方法无法处理的噪声分布。最后,我们展示了该技术在荧光显微成像与单细胞基因组数据中的应用,凸显其在科学问题中的实用价值。本工作将强大生成模型的应用扩展至逆向生成领域。

原文摘要 · Abstract (English)

Inverse generation problems, such as denoising without ground truth observations, is a critical challenge in many scientific inquiries and real-world applications. While recent advances in generative models like diffusion models, conditional flow matching, and consistency models achieved impressive results by casting generation as denoising problems, they cannot be directly used for inverse generation without access to clean data. Here we introduce Inverse Flow (IF), a novel framework that enables using these generative models for inverse generation problems including denoising without ground truth. Inverse Flow can be flexibly applied to nearly any continuous noise distribution and allows complex dependencies. We propose two algorithms for learning Inverse Flows, Inverse Flow Matching (IFM) and Inverse Consistency Model (ICM). Notably, to derive the computationally efficient, simulation-free inverse consistency model objective, we generalized consistency training to any forward diffusion processes or conditional flows, which have applications beyond denoising. We demonstrate the effectiveness of IF on synthetic and real datasets, outperforming prior approaches while enabling noise distributions that previous methods cannot support. Finally, we showcase applications of our techniques to fluorescence microscopy and single-cell genomics data, highlighting IF's utility in scientific problems. Overall, this work expands the applications of powerful generative models to inversion generation problems.

逆向生成一致性模型流模型科学计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。