arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.08075cs.LGstat.ML

TERM模型:用于密度估计的张量环混合模型

TERM Model: Tensor Ring Mixture Model for Density Estimation

University of Electronic Science and Technology of China(电子科技大学);Skolkovo Institute of Science and Technology(斯科尔科沃科学技术研究院)

Ruituo Wu, Jiani Liu, Ce Zhu, Anh-Huy Phan, Ivan V. Oseledets, Yipeng Liu

AI总结:

针对张量网络密度估计中排列候选多、表达能力受限的问题,提出基于张量环分解的TERM模型及带自适应权重的混合模型,在中等维度数据集的密度估计和采样任务上取得更优效果。

摘要:

高效概率密度估计是统计机器学习中的核心挑战。基于张量的概率图方法解决了神经网络方法中存在的可解释性与稳定性问题。然而,大量潜在的张量排列可能会产生结构相同但表达能力各异的张量网络。本文将张量环分解用于密度估计器,与现有常用分解方法相比,该方法大幅减少了排列候选的数量,同时提升了表达能力。此外,本文进一步设计了一种融合多个带自适应权重的排列候选的混合模型,提升了表达的灵活性与全面性。与当前主流的张量网络结构/排列搜索方向不同,本文方法受集成学习启发,提供了一种新视角:该方法认可次优排列除最优排列的信息外,还能提供独特信息。实验表明,所提方法在中等维度数据集的概率密度估计以及捕捉复杂细节的采样任务上表现更优。

英文摘要:

Efficient probability density estimation is a core challenge in statistical machine learning. Tensor-based probabilistic graph methods address interpretability and stability concerns encountered in neural network approaches. However, a substantial number of potential tensor permutations can lead to a tensor network with the same structure but varying expressive capabilities. In this paper, we take tensor ring decomposition for density estimator, which significantly reduces the number of permutation candidates while enhancing expressive capability compared with existing used decompositions. Additionally, a mixture model that incorporates multiple permutation candidates with adaptive weights is further designed, resulting in increased expressive flexibility and comprehensiveness. Different from the prevailing directions of tensor network structure/permutation search, our approach provides a new viewpoint inspired by ensemble learning. This approach acknowledges that suboptimal permutations can offer distinctive information besides that of optimal permutations. Experiments show the superiority of the proposed approach in estimating probability density for moderately dimensional datasets and sampling to capture intricate details.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑