[论文解读] A Framework for Wasserstein-1-Type Metrics

Bernhard Schmitzer, Benedikt Wirth|arXiv (Cornell University)|Jan 8, 2017

Geometric Analysis and Curvature Flows参考文献 6被引用 16

一句话总结

本文提出了一种统一的非平衡Wasserstein-1型度量框架，该框架将标准$W_1$度量推广至处理质量不等的非负测度。通过利用对偶公式和与局部差异度量的下确界卷积，该方法在保持计算效率的同时，实现了对质量变化的鲁棒建模——在图像重建、视频光流估计和噪声分解等应用中表现出优于传统$W_1$和总变差方法的性能。

ABSTRACT

We propose a unifying framework for generalising the Wasserstein-1 metric to a discrepancy measure between nonnegative measures of different mass. This generalization inherits the convexity and computational efficiency from the Wasserstein-1 metric, and it includes several previous approaches from the literature as special cases. For various specific instances of the generalized Wasserstein-1 metric we furthermore demonstrate their usefulness in applications by numerical experiments.

研究动机与目标

• 为解决标准$W_1$度量要求测度质量相等的局限性，将其推广至质量可变化的非平衡设置。
• 在单一凸优化框架下统一现有非平衡运输方法，同时保持计算效率。
• 在成像和视频分析中实现实际应用，以应对由噪声、光照变化或结构动态引起的质量波动。
• 证明通过下确界卷积将$W_1$型运输与局部差异度量结合，可获得优于标准$W_1$和总变差方法的重建与去噪效果。

提出的方法

• 该框架采用$W_1$的对偶公式，将问题简化为定义域上的局部约束，通过最小费用流方法实现高效计算。
• 通过将$W_1$与局部差异度量$D$进行下确界卷积类型扩展，引入质量生成/销毁机制。
• 在适当假设下，证明了对偶公式与基于下确界卷积的方法之间的等价性，从而实现灵活建模。
• 开发了一种采用变量分裂的原始-对偶算法，确保收敛性和可扩展性。
• 通过有限差分法进行离散化，并采用算子分裂方法，对非光滑项应用近端算子。
• 该方法可在统一优化框架中集成额外的正则化项，如总变差。

实验结果

研究问题

• RQ1 如何在保持计算效率的前提下，将$W_1$度量推广至处理质量不等的非负测度？
• RQ2 非平衡$W_1$的对偶公式与基于下确界卷积的局部差异度量扩展之间存在何种关系？
• RQ3 不同局部差异度量如何影响非平衡运输在图像重建任务中的行为？
• RQ4 所提出的框架是否能在重建细长结构和处理视频序列中的噪声方面优于标准$W_1$和总变差方法？

主要发现

• 所提出的非平衡$W_1$型度量在重建细长结构（如竖直线）方面优于标准$W_1$和总变差方法，归因于其能够一致地传输质量。
• 在视频光流估计中，非平衡度量能正确识别生物细胞运动，即使存在亮度变化和噪声，而标准$W_1$因质量失衡而失效。
• 下确界卷积模型$W_{2D^{ ext{TV}}, D^{ ext{d}}, D^{ ext{d}}}$成功结合了噪声滤除与结构重建，性能优于单独使用$D^{ ext{TV}}$或$W_1$。
• 数值实验表明，在下确界卷积中间使用$D^{ ext{H}}$可实现平滑、对称的局部质量波动平衡。
• 增加正则化参数$(\lambda_0, \lambda_1)$会使结果呈现更简化的、均匀的线条，证实了模型对不同重建目标的适应能力。