arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2010.02577cs.LG
收藏

基于二进制嵌入和三进制模型系数的内存和计算高效核SVM

Memory and Computation-Efficient Kernel SVM via Binary Embedding and Ternary Model Coefficients

Hong Kong Baptist University(香港浸会大学);Department of Computer Science(计算机科学系);Hong Kong SAR, China(中国香港特别行政区)

Zijian Lei, Liang Lan

摘要:

核近似被广泛用于扩展核SVM训练和预测。然而,如果我们想将内核近似模型部署在内存有限的设备(如手机、智能手表和物联网设备)上,其内存和计算成本仍然太高。为了解决这一挑战,我们提出了一种新的内存和计算效率的核SVM模型,通过使用二进制嵌入和二进制模型系数。首先,我们提出了一种有效的方法来生成紧凑的二进制嵌入的数据,保持内核的相似性。其次,我们提出了一个简单但有效的算法来学习一个线性分类模型的三元系数,可以支持不同类型的损失函数和正则化。我们的算法可以实现更好的泛化精度比现有的作品学习二进制系数,因为我们允许系数是$-1$,$0$,或$1$在训练阶段,和系数$0$可以被删除在模型推断二进制分类。此外,我们提供了一个详细的分析,我们的算法的收敛性和我们的模型的推理复杂度。分析表明,该模型保证了收敛到局部最优,推理复杂度远低于其他竞争方法。在5个大型真实数据集上的实验结果表明,该方法能够以小于30KB的内存开销建立精确的非线性SVM模型。

英文摘要:

Kernel approximation is widely used to scale up kernel SVM training and prediction. However, the memory and computation costs of kernel approximation models are still too high if we want to deploy them on memory-limited devices such as mobile phones, smartwatches, and IoT devices. To address this challenge, we propose a novel memory and computation-efficient kernel SVM model by using both binary embedding and binary model coefficients. First, we propose an efficient way to generate compact binary embedding of the data, preserving the kernel similarity. Second, we propose a simple but effective algorithm to learn a linear classification model with ternary coefficients that can support different types of loss function and regularizer. Our algorithm can achieve better generalization accuracy than existing works on learning binary coefficients since we allow coefficient to be $-1$, $0$, or $1$ during the training stage, and coefficient $0$ can be removed during model inference for binary classification. Moreover, we provide a detailed analysis of the convergence of our algorithm and the inference complexity of our model. The analysis shows that the convergence to a local optimum is guaranteed, and the inference complexity of our model is much lower than other competing methods. Our experimental results on five large real-world datasets have demonstrated that our proposed method can build accurate nonlinear SVM models with memory costs less than 30KB.

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

↑