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

通过对数据和模型系数进行二值化的高效内存分解机

Memory-Efficient Factorization Machines via Binarizing both Data and Model Coefficients

Hong Kong Baptist University(香港浸会大学)

Yu Geng, Liang Lan

摘要:

因子分解机(FM),一个通用的预测器,可以有效地在线性时间模型的特征交互,主要是提出了协作推荐,并已被广泛用于回归,分类和排名任务。子空间编码分解机(SEFM)是近年来提出的一种新的分解机,它通过对每个输入特征进行one-hot编码,对单个特征和特征间的交互进行显式非线性特征映射,克服了分解机(FM)的表达能力限制。尽管SEFM很有效,但它使FM的内存成本增加了$b$倍,其中$b$是对每个输入特征应用独热编码时的箱数。为了减少SEFM的内存开销,我们提出了一种称为二进制FM的新方法,该方法将模型参数约束为二进制值(即,1或$-1$)。然后,每个参数值可以有效地存储在一个比特中。我们提出的方法可以显着减少SEFM模型的内存开销。此外,我们提出了一种新的算法,有效地和高效地学习建议FM与二进制约束使用直通估计(STE)与自适应梯度下降(Adagrad)。最后,我们评估了我们提出的方法在八个不同的分类数据集上的性能。我们的实验结果表明,我们提出的方法达到了相当的精度与SEFM,但少得多的内存成本。

英文摘要:

Factorization Machines (FM), a general predictor that can efficiently model feature interactions in linear time, was primarily proposed for collaborative recommendation and have been broadly used for regression, classification and ranking tasks. Subspace Encoding Factorization Machine (SEFM) has been proposed recently to overcome the expressiveness limitation of Factorization Machines (FM) by applying explicit nonlinear feature mapping for both individual features and feature interactions through one-hot encoding to each input feature. Despite the effectiveness of SEFM, it increases the memory cost of FM by $b$ times, where $b$ is the number of bins when applying one-hot encoding on each input feature. To reduce the memory cost of SEFM, we propose a new method called Binarized FM which constraints the model parameters to be binary values (i.e., 1 or $-1$). Then each parameter value can be efficiently stored in one bit. Our proposed method can significantly reduce the memory cost of SEFM model. In addition, we propose a new algorithm to effectively and efficiently learn proposed FM with binary constraints using Straight Through Estimator (STE) with Adaptive Gradient Descent (Adagrad). Finally, we evaluate the performance of our proposed method on eight different classification datasets. Our experimental results have demonstrated that our proposed method achieves comparable accuracy with SEFM but with much less memory cost.

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

↑