paper-with-me

Papers

Decoupled Diffusion Sampling for Inverse Problems on Function Spaces

2026-01-30 · Thomas Y. L. Lin, Jiachen Yao, Lufang Chiang, Julius Berner, Anima Anandkumar arxiv

We propose a data-efficient, physics-aware generative framework in function space for inverse PDE problems. Existing plug-and-play diffusion posterior samplers represent physics implicitly through joint coefficient-solution modeling, requiring substantial paired supervision. In contrast, our Decoupled Diffusion Inverse Solver (DDIS) employs a decoupled design: an unconditional diffusion learns the coefficient prior, while a neural operator explicitly models the forward PDE for guidance. This decoupling enables superior data efficiency and effective physics-informed learning, while naturally supporting Decoupled Annealing Posterior Sampling (DAPS) to avoid over-smoothing in Diffusion Posterior Sampling (DPS). Theoretically, we prove that DDIS avoids the guidance attenuation failure of joint models when training data is scarce. Empirically, DDIS achieves state-of-the-art performance under sparse observation, improving $l_2$ error by 11% and spectral error by 54% on average; when data is limited to 1%, DDIS maintains accuracy with 40% advantage in $l_2$ error compared to joint models.

📄 PDF Abstract BibTeX arXiv:2601.23280

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealing

2024-07-01 · CVPR 2025 1 · Bingliang Zhang, Wenda Chu, Julius Berner, Chenlin Meng 외

Diffusion models have recently achieved success in solving Bayesian inverse problems with learned data priors. Current methods build on top of the diffusion sampling process, where each denoising step makes small modific…

DenoisingImage RestorationRetrieval

Function-Space Decoupled Diffusion for Forward and Inverse Modeling in Carbon Capture and Storage

2026-02-12 · Xin Ju, Jiachen Yao, Anima Anandkumar, Sally M. Benson 외 arxiv

Accurate characterization of subsurface flow is critical for Carbon Capture and Storage (CCS) but remains challenged by the ill-posed nature of inverse problems with sparse observations. We present Function-space Decoupl…

DAPS++: Rethinking Diffusion Inverse Problems with Decoupled Posterior Annealing

2025-11-21 · Hao Chen, Renzheng Zhang, Scott S. Howard arxiv

From a Bayesian perspective, score-based diffusion solves inverse problems through joint inference, embedding the likelihood with the prior to guide the sampling process. However, this formulation fails to explain its pr…

Computational EfficiencyImage Restoration

Improving Decoupled Posterior Sampling for Inverse Problems using Data Consistency Constraint

2024-12-01 · Zhi Qi, Shihong Yuan, Yulin Yuan, Linling Kuang 외

Diffusion models have shown strong performances in solving inverse problems through posterior sampling while they suffer from errors during earlier steps. To mitigate this issue, several Decoupled Posterior Sampling meth…

Solving Linear-Gaussian Bayesian Inverse Problems with Decoupled Diffusion Sequential Monte Carlo

2025-02-10 · Filip Ekström Kelvinius, Zheng Zhao, Fredrik Lindsten

A recent line of research has exploited pre-trained generative diffusion models as priors for solving Bayesian inverse problems. We contribute to this research direction by designing a sequential Monte Carlo method for l…