CVPRMay, 2022

一种隐式 Sinkhorn 微分的统一框架

TL;DR该研究提出了一种能够通过隐式求导获得 Sinkhorn 层解析梯度的算法,相较于以前的工作,该框架基于 Sinkhorn 算子最一般的公式并允许任何类型的损失函数,目标容量和成本矩阵联合求导,提高了梯度的稳定性和精度,并且在计算效率方面的表现更佳。