中文

用于生成式建模的蒙日-安培流

机器学习 2018-09-28 v1 统计力学 动力系统 机器学习

摘要

我们提出一种深度生成模型,名为蒙日-安培流(Monge-Ampère flow),其构建于最优传输理论中蒙日-安培方程导出的连续时间梯度流。从潜空间到数据空间的生成映射遵循一个动力系统,其中可学习的势函数引导可压缩流体流向目标密度分布。模型训练等价于求解一个最优控制问题。蒙日-安培流具有可处理的似然,并支持高效采样与推断。通过设计合适的标量势函数,可轻易在生成模型中施加对称性约束。我们将该方法应用于 MNIST 数据集的无监督密度估计以及临界点处二维伊辛模型的变分计算。该方法将蒙日-安培方程、最优传输与流体动力学的见解和技术引入基于可逆流的生成模型。

关键词

引用

@article{arxiv.1809.10188,
  title  = {Monge-Amp\`ere Flow for Generative Modeling},
  author = {Linfeng Zhang and Weinan E and Lei Wang},
  journal= {arXiv preprint arXiv:1809.10188},
  year   = {2018}
}