arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.02442cs.LGeess.IV

一种用于基于ReLU的非线性矩阵分解的动量加速算法

A Momentum Accelerated Algorithm for ReLU-based Nonlinear Matrix Decomposition

School of Mathematics and Computational Science, Xiangtan University(湘潭大学数学与计算科学学院);School of Mathematical Sciences, Beihang University(北京航空航天大学数学科学学院)

Qingsong Wang, Chunfeng Cui, Deren Han

AI总结:

本文针对ReLU-NMD模型的过拟合问题,提出Tikhonov正则化的ReLU-NMD-T模型,并设计了一种同时引入正负动量参数的动量加速算法,在真实数据集上验证了其有效性。

摘要:

近年来,由于非线性矩阵分解(NMD)与神经网络的密切联系,人们对NMD的探索兴趣日益增长。NMD旨在通过逐元素非线性函数,从稀疏非负矩阵中寻找低秩矩阵,其中一种典型选择是整流线性单元(ReLU)激活函数。为解决现有基于ReLU的NMD模型(ReLU-NMD)中的过拟合问题,我们提出了一种Tikhonov正则化的ReLU-NMD模型,称为ReLU-NMD-T。随后,我们引入了一种动量加速算法来处理ReLU-NMD-T模型。与大多数现有研究不同的一个显著特点是,我们的算法同时引入了正、负动量参数。我们在真实世界数据集上的数值实验表明了所提模型和算法的有效性。此外,代码可在 https://github.com/nothing2wang/NMD-TM 获取。

英文摘要:

Recently, there has been a growing interest in the exploration of Nonlinear Matrix Decomposition (NMD) due to its close ties with neural networks. NMD aims to find a low-rank matrix from a sparse nonnegative matrix with a per-element nonlinear function. A typical choice is the Rectified Linear Unit (ReLU) activation function. To address over-fitting in the existing ReLU-based NMD model (ReLU-NMD), we propose a Tikhonov regularized ReLU-NMD model, referred to as ReLU-NMD-T. Subsequently, we introduce a momentum accelerated algorithm for handling the ReLU-NMD-T model. A distinctive feature, setting our work apart from most existing studies, is the incorporation of both positive and negative momentum parameters in our algorithm. Our numerical experiments on real-world datasets show the effectiveness of the proposed model and algorithm. Moreover, the code is available at https://github.com/nothing2wang/NMD-TM.

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

↑