中文

深度生成模型中用于稀疏多模态分布的证据Softmax

机器学习 2021-10-28 v1 人工智能 计算与语言 计算机视觉与模式识别 机器学习

摘要

生成模型的许多应用依赖于对其高维输出概率分布的边际化。产生稀疏概率分布的归一化函数可使精确边际化在计算上更易处理。然而,稀疏归一化函数通常需要替代损失函数进行训练,因为对数似然对稀疏概率分布是未定义的。此外,许多稀疏归一化函数常会坍缩分布的多模态性。本文提出ev-softmax\textit{ev-softmax},一种保留概率分布多模态性的稀疏归一化函数。我们推导了其性质,包括闭式梯度,并引入了对ev-softmax\textit{ev-softmax}的连续近似族,其具有全支撑且可用负对数似然和Kullback-Leibler散度等概率损失函数训练。我们在多种生成模型上评估了该方法,包括变分自编码器和自回归架构。我们的方法在分布精度上优于现有的稠密与稀疏归一化技术。我们证明ev-softmax\textit{ev-softmax}在保持多模态性的同时成功降低了概率分布的维度。

关键词

引用

@article{arxiv.2110.14182,
  title  = {Evidential Softmax for Sparse Multimodal Distributions in Deep Generative Models},
  author = {Phil Chen and Masha Itkina and Ransalu Senanayake and Mykel J. Kochenderfer},
  journal= {arXiv preprint arXiv:2110.14182},
  year   = {2021}
}

备注

Accepted to NeurIPS 2021. Code is available at https://github.com/sisl/EvSoftmax