用不平衡最优传输解决无配对图像逆问题,提升噪声鲁棒性。
UOTIP: Unbalanced Optimal Transport Map for Unpaired Inverse Problems

- 基于不平衡最优传输构建映射,从噪声数据重建清晰图像。
- 在多类噪声和数据分布不均场景下表现更优,跨噪声类型泛化性强。
- 理论证明映射存在唯一性,适合处理线性与非线性逆问题。
我们研究无配对图像逆问题,即训练时仅有独立的噪声测量与干净目标信号数据集,无一一对应关系。提出一种基于不平衡最优传输的新方法——无配对逆问题最优传输映射(UOTIP)。将重建任务建模为从噪声测量分布到干净信号分布的不平衡最优传输映射学习,引入基于似然的成本函数。通过放松精确边缘约束,该框架具备对多级观测噪声的鲁棒性、对噪声与干净数据集类别不平衡的适应性,以及对多种噪声类型的泛化能力。进一步理论上证明,引入二次代价项可满足扭动条件,确保即使在病态逆问题中运输映射仍存在且唯一。实验表明,UOTIP在各类线性和非线性逆问题的无配对基准上达到当前最优性能。
原文摘要 · Abstract (English)
We investigate unpaired image inverse problems, a challenging setting where only independent, non-paired sets of noisy measurements and clean target signals are available for training. We propose a novel inverse problem solver based on Unbalanced Optimal Transport, called Unbalanced Optimal Transport Map for Inverse Problems (UOTIP). Our method formulates the reconstruction task, predicting clean target signals from noisy measurements, as learning a UOT Map from noisy measurement distribution to clean signal distribution by incorporating a likelihood-based cost function. By relaxing the exact marginal constraint, the UOT framework provides key advantages to our model: robustness to multi-level observation noise, adaptability to class imbalance between noisy and clean datasets, and generalizability to diverse noise-type scenarios. Furthermore, we theoretically demonstrate that incorporating a quadratic cost term ensures the existence and uniqueness of the transport map by satisfying the twist condition, even for ill-posed inverse problems. Our experiments demonstrate that UOTIP achieves state-of-the-art performance on unpaired image inverse problem benchmarks, across linear and nonlinear inverse problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。