English(EN) TorchKM: A GPU-Oriented Library for Kernel Learning and Model Selection

TorchKM 库加速 GPU 上的核机器学习

作者 PulseAugur 编辑部 · [2 个来源] · 2026-06-04 21:56

研究人员开发了 TorchKM，这是一个开源库，旨在加速 GPU 上的核机器学习任务。该库提供了一个与 scikit-learn 兼容的 API，并利用 GPU 友好的线性代数来加速训练和模型选择。基准测试表明，TorchKM 在提供比现有方法显著加速的同时，实现了具有竞争力的预测性能。 AI

影响加速核机器学习工作流程，可能实现 SVM 等模型的更快实验和部署。

排序理由该集群包含一篇详细介绍新的机器学习开源库的学术论文。

AI 生成摘要 · Google Gemini · 来自 2 个来源。我们如何撰写摘要 →

报道来源 [2]

arXiv stat.ML TIER_1 English(EN) · Yikai Zhang, Gaoxiang Jia, Jie Ding, Boxiang Wang · 2026-06-08 04:00

TorchKM：面向 GPU 的核学习与模型选择库

arXiv:2606.06742v1 Announce Type: cross Abstract: TorchKM is an open-source library for kernel machines, including support vector machines, kernel logistic regression, and kernel quantile regression, with GPU acceleration. The library features a scikit-learn-style API and is desi…
arXiv stat.ML TIER_1 English(EN) · Boxiang Wang · 2026-06-04 21:56

TorchKM：面向 GPU 的核学习与模型选择库

TorchKM is an open-source library for kernel machines, including support vector machines, kernel logistic regression, and kernel quantile regression, with GPU acceleration. The library features a scikit-learn-style API and is designed to exploit GPU-friendly linear algebra, accel…