Aug, 2023

基于最优输运的深度学习框架用于滞缓问题:利用 Sinkhorn 损失和 Wasserstein 核

TL;DR我们提出了一种将最优输运论和基于神经网络的方法集成到新颖的减小阶模型框架中的方法,通过使用 Wasserstein 距离作为自定义核函数的核 POD 方法,并利用 Sinkhorn 算法高效训练得到的神经网络,可以捕捉数据的几何结构,从而实现精确学习降维解流形。与传统指标(如均方误差或交叉熵)相比,利用 Sinkhorn 散度作为损失函数可以增强训练的稳定性,对抗过拟合和噪声并加速收敛。通过在具有 Kolmogorov n-width 缓慢衰减特性的一系列具有挑战性的测试案例上进行实验,结果表明我们的框架在准确性和计算效率方面优于传统的减小阶模型方法。