TorchSVM: A PyTorch-based Library for Large-scale Kernel SVM and Kernel Machines

Music City Center 

In this talk, we introduce TorchKSVM, a PyTorch-based library that trains kernel SVMs and other large-margin classifiers with exact cross-validation error computation. Traditional SVM solvers often encounter scalability and efficiency challenges, particularly when handling large datasets or performing multiple cross-validation runs. TorchKSVM effectively enhances both speed and scalability through CUDA-accelerated matrix operations. By carefully designing the underlying algorithms, TorchKSVM employs advanced strategies, such as spectral algorithms, to fully leverage parallel computing and optimize the use of computational resources. Benchmark experiments demonstrate that TorchKSVM consistently outperforms existing kernel SVM solvers, in both CPU and GPU implementations, in terms of accuracy and speed.

Support Vector Machines

PyTorch

GPU

Parallel Computing,

Large-Margin Classification