arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.06960cs.LG
收藏

大规模低秩多字典选择

Low Rank Multi-Dictionary Selection at Scale

University at Albany, State University of New York(纽约州立大学奥尔巴尼分校)

Boya Ma, Maxwell McNeil, Abram Magner, Petko Bogdanov

AI总结:

本文提出LRMDS低秩稀疏多字典原子选择方法,通过逐步选择行列原子对并利用子字典进行凸松弛编码,在理论保证下实现3–10倍加速和最高两个数量级的表示质量提升。

摘要:

稀疏字典编码框架将信号表示为少量预定义字典原子的线性组合。它已被应用于图像、时间序列、图信号,并且最近通过联合使用时间字典和空间字典,被用于二维(或2D)时空数据。大型且过完备的字典能够实现高质量模型,但也带来了可扩展性挑战,而这些挑战在多字典设置中会进一步加剧。因此,本文解决的一个重要问题是:如何针对大型字典和数据集扩展多字典编码?我们提出了一种用于低秩稀疏编码的多字典原子选择技术,名为LRMDS。为了使其能够扩展到大型字典和数据集,该方法根据行-列原子对与数据的对齐程度逐步选择成组的原子对,并通过相应的子字典执行凸松弛编码。我们在理论和实验上均证明,当数据具有使用稀疏原子子集的低秩编码时,LRMDS能够在温和假设下以强有力的保证选出这些原子。此外,我们在合成数据集和真实世界数据集上,并针对一系列编码字典,证明了LRMDS的可扩展性和质量。与基线方法相比,它实现了3倍至10倍的加速;同时,在固定目标原子数量的情况下,在部分真实世界数据集上,表示质量最高提升达两个数量级。

英文摘要:

The sparse dictionary coding framework represents signals as a linear combination of a few predefined dictionary atoms. It has been employed for images, time series, graph signals and recently for 2-way (or 2D) spatio-temporal data employing jointly temporal and spatial dictionaries. Large and over-complete dictionaries enable high-quality models, but also pose scalability challenges which are exacerbated in multi-dictionary settings. Hence, an important problem that we address in this paper is: How to scale multi-dictionary coding for large dictionaries and datasets? We propose a multi-dictionary atom selection technique for low-rank sparse coding named LRMDS. To enable scalability to large dictionaries and datasets, it progressively selects groups of row-column atom pairs based on their alignment with the data and performs convex relaxation coding via the corresponding sub-dictionaries. We demonstrate both theoretically and experimentally that when the data has a low-rank encoding with a sparse subset of the atoms, LRMDS is able to select them with strong guarantees under mild assumptions. Furthermore, we demonstrate the scalability and quality of LRMDS in both synthetic and real-world datasets and for a range of coding dictionaries. It achieves 3X to 10X speed-up compared to baselines, while obtaining up to two orders of magnitude improvement in representation quality on some of the real world datasets given a fixed target number of atoms.

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

↑