arXiv:2412.04766eess.IVcs.AI2024-12被引 1

用流匹配解决逆问题,能处理噪声和缺失数据并量化不确定性。

DAWN-FM: Data-Aware and Noise-Informed Flow Matching for Solving Inverse Problems

  • 通过时间依赖速度场建模,融合数据与噪声信息进行生成
  • 在图像去模糊和断层成像中实现高精度解并生成多组合理结果
  • 适合数据不全或噪声大的场景,优于预训练扩散模型

逆问题涉及从不完整或含噪观测中估计参数,广泛存在于医学成像、地球物理和信号处理等领域。这类问题通常病态,需正则化稳定求解。本文采用流匹配(Flow Matching, FM)框架,通过确定性过程将简单参考分布(如高斯分布)映射到目标分布。提出DAWN-FM:数据感知且噪声知情的流匹配方法,显式引入数据与噪声嵌入,使模型能直接利用观测信息并考虑噪声影响,在数据缺失或噪声较大时表现更稳健。通过学习时间依赖的速度场,FM不仅提供精确解,还能通过生成多组可能结果实现不确定性量化。与依赖预训练的扩散模型不同,本方法针对具体逆问题训练,可适应不同噪声水平。在图像去模糊和断层成像等任务上通过大量数值实验验证了其有效性和鲁棒性。代码已公开于:https://github.com/ahxmeds/DAWN-FM.git。

原文摘要 · Abstract (English)

Inverse problems, which involve estimating parameters from incomplete or noisy observations, arise in various fields such as medical imaging, geophysics, and signal processing. These problems are often ill-posed, requiring regularization techniques to stabilize the solution. In this work, we employ Flow Matching (FM), a generative framework that integrates a deterministic processes to map a simple reference distribution, such as a Gaussian, to the target distribution. Our method DAWN-FM: Data-AWare and Noise-Informed Flow Matching incorporates data and noise embedding, allowing the model to access representations about the measured data explicitly and also account for noise in the observations, making it particularly robust in scenarios where data is noisy or incomplete. By learning a time-dependent velocity field, FM not only provides accurate solutions but also enables uncertainty quantification by generating multiple plausible outcomes. Unlike pretrained diffusion models, which may struggle in highly ill-posed settings, our approach is trained specifically for each inverse problem and adapts to varying noise levels. We validate the effectiveness and robustness of our method through extensive numerical experiments on tasks such as image deblurring and tomography. The code is available at: https://github.com/ahxmeds/DAWN-FM.git.

逆问题流匹配不确定性量化图像重建

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