arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.02759stat.MLcs.LG

近通用的非负Einsum分解乘法更新

Near-Universal Multiplicative Updates for Nonnegative Einsum Factorization

Department of Statistics, University of Chicago(统计学系,芝加哥大学);Data Science Institute, University of Chicago(数据科学研究所,芝加哥大学)

John Hood, Aaron Schein

AI总结:

NNEinFact提出了一种基于Einsum的非负张量分解乘法更新算法,能够高效适应各种非负分解模型,实现更快的收敛速度和更优的预测性能。

摘要:

尽管多维数据在科学领域中普遍存在,但很少有用户友好的工具能够适应定制的非负张量分解。研究者可能使用基于梯度的自动微分(通常在非负设置中表现不佳),或者在有限的方法集之间进行选择,或者从头开始实现自己的模型。作为替代方案,我们介绍了NNEinFact,一种基于Einsum的乘法更新算法,能够适应任何可以通过张量收缩表达的非负张量分解,通过最小化多种用户指定的损失函数(包括(α,β)-发散度)之一来实现。要使用NNEinFact,研究者只需用字符串指定他们的模型。NNEinFact收敛到损失的 stationary 点,支持缺失数据,并且能够以秒为单位适应具有数千万个条目的张量。经验上,NNEinFact能够定制模型,在真实世界张量数据的验证预测任务中,比标准模型高出超过37%,并且在测试损失上达到梯度方法的一半,同时收敛速度高达90倍。

英文摘要:

Despite the ubiquity of multiway data across scientific domains, there are few performant and user-friendly methods that fit non-standard nonnegative tensor factorization models tailored to the data at-hand. Researchers may use gradient-based automatic differentiation, which often struggles under nonnegative constraints, choose between a limited set of methods with mature implementations, or implement their own model from scratch. As an alternative, we introduce NNEinFact, an einsum-based multiplicative update algorithm that fits any nonnegative tensor factorization expressible as a tensor contraction by minimizing one of many user-specified loss functions, including the $(α,β)$-divergence. To use NNEinFact, the researcher specifies their model with a string. NNEinFact converges to a stationary point of the loss, supports missing data, and fits to tensors with hundreds of millions of entries in seconds. Empirically, NNEinFact fits custom models which outperform standard ones in prediction tasks on real-world tensor data by over $37\%$ and attains less than half the test loss of gradient-based methods while converging up to 90 times faster. Software is publicly available at https://github.com/jhood3/einfact.

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

↑