向量与张量上置换不变函数的通用表示
机器学习
2023-10-24 v1
摘要
我们研究的主要对象是多重集函数——即关于变大小输入的置换不变函数。由 \cite{zaheer2017deep} 提出的 Deep Sets 通过可和分解模型为标量上的连续多重集函数提供了\emph{通用表示}。将函数定义域限制为 维向量的有限多重集时,Deep Sets 还提供了一种\emph{通用逼近},其所需潜空间维度为 ——其中 为输入多重集大小的上界。在本文中,我们强化该结果,证明通过 的潜空间维度,对连续与非连续多重集函数均可保证通用表示。随后我们引入\emph{可识别}多重集,可用识别函数唯一标记其元素,即有限精度向量是可识别的。利用对可识别多重集的分析,我们证明通用连续多重集函数的可和分解模型仅需 的潜维度。我们进一步表明该模型的编码器与解码器函数均为连续——这是相对于缺乏此类保证的现有工作的主要贡献。同时,这相比前述针对连续与非连续多重集函数通用表示所推导的 界有显著改进。我们随后推广结果,给出特殊的和分解结构,以通用表示可识别张量上的置换不变张量函数。这些和分解模型族使我们能够设计深度网络架构,并将其部署于序列、图像与图上的多种学习任务中。
引用
@article{arxiv.2310.13829,
title = {Universal Representation of Permutation-Invariant Functions on Vectors and Tensors},
author = {Puoya Tabaghi and Yusu Wang},
journal= {arXiv preprint arXiv:2310.13829},
year = {2023}
}