提出新算法解决ReLU矩阵分解的计算难题
An Accelerated Alternating Partial Bregman Algorithm for ReLU-based Matrix Decomposition
- 设计含正则项的非线性分解框架,兼顾低秩、稀疏与非负
- 算法实现多变量同步更新,理论证明收敛性与效率
- 适用于图正则聚类和稀疏NMF压缩,实测效果优异
尽管低秩估计在数据挖掘中表现卓越,但面对缺乏低秩结构的数据时性能下降。本文聚焦非负稀疏矩阵,探究修正线性单元(ReLU)激活函数的内在低秩特性。提出一种新型非线性矩阵分解框架,包含综合正则项,同时促进聚类与压缩任务中的低秩性、稀疏性及非负性。该公式因多块结构、非凸性、非光滑性及全局梯度Lipschitz连续性的缺失,带来显著计算挑战。为此,开发加速交替部分Bregman近端梯度法(AAPB),其独特之处在于支持多变量同步更新。在温和且理论合理的假设下,建立了子线性与全局收敛性质。通过精心选择适配不同正则项的核生成距离,推导出闭式解,且对任意 $L\ge 1$ 均保持 $L$-平滑可调性。数值实验在图正则聚类与稀疏NMF基压缩任务中验证了模型与算法的有效性。
原文摘要 · Abstract (English)
Despite the remarkable success of low-rank estimation in data mining, its effectiveness diminishes when applied to data that inherently lacks low-rank structure. To address this limitation, in this paper, we focus on non-negative sparse matrices and aim to investigate the intrinsic low-rank characteristics of the rectified linear unit (ReLU) activation function. We first propose a novel nonlinear matrix decomposition framework incorporating a comprehensive regularization term designed to simultaneously promote useful structures in clustering and compression tasks, such as low-rankness, sparsity, and non-negativity in the resulting factors. This formulation presents significant computational challenges due to its multi-block structure, non-convexity, non-smoothness, and the absence of global gradient Lipschitz continuity. To address these challenges, we develop an accelerated alternating partial Bregman proximal gradient method (AAPB), whose distinctive feature lies in its capability to enable simultaneous updates of multiple variables. Under mild and theoretically justified assumptions, we establish both sublinear and global convergence properties of the proposed algorithm. Through careful selection of kernel generating distances tailored to various regularization terms, we derive corresponding closed-form solutions while maintaining the $L$-smooth adaptable property always holds for any $L\ge 1$. Numerical experiments, on graph regularized clustering and sparse NMF basis compression confirm the effectiveness of our model and algorithm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。