System-Embedded Diffusion Bridge Models
University of Warsaw, Centre for Credible AI · Massachusetts General Hospital, Harvard University · Massachusetts General Hospital, Harvard Medical School · The Ohio State University · Massachusetts General Hospital， Harvard Medical School · Netezza Polska
PDF 由论文原始站点提供，PaperCompass 不保存论文文件。
摘要
Solving inverse problems—recovering signals from incomplete or noisy measurements—is fundamental in science and engineering. Score-based generative models (SGMs) have recently emerged as a powerful framework for this task. Two main paradigms have formed: unsupervised approaches that adapt pretrained generative models to inverse problems, and supervised bridge methods that train stochastic processes conditioned on paired clean and corrupted data. While the former typically assume knowledge of the measurement model, the latter have largely overlooked this structural information. We introduce System-embedded Diffusion Bridge Models (SDBs), a new class of supervised bridge methods that explicitly embed the known linear measurement system into the coefficients of a matrix-valued SDE. This principled integration yields consistent improvements across diverse linear inverse problems and demonstrates robust generalization under system misspecification between training and deployment, offering a promising solution to real-world applications.