GPU-Accelerated Primal Learning for Extremely Fast Large-Scale Classification
John T. Halloran, David M. Rocke
Abstract
One of the most efficient methods to solve L 2 -regularized primal problems, such as logistic regression and linear support vector machine (SVM) classification, is the widely used trust region Newton algorithm, TRON [38]. While TRON has recently been shown to enjoy substantial speedups on shared-memory multi-core systems [35, 22] , exploiting graphical processing units (GPUs) to speed up the method is significantly more difficult, owing to the highly complex and heavily sequential nature of the algorithm. In this work, we show that using judicious GPU-optimization principles, TRON training time for different losses and feature representations may be drastically reduced. For sparse feature sets, we show that using GPUs to train logistic regression classifiers in LIBLINEAR is up to an orderof-magnitude faster than solely using multithreading. For dense feature sets-which impose far more stringent memory constraints-we show that GPUs substantially reduce the lengthy SVM learning times required for state-of-the-art proteomics analysis, leading to dramatic improvements over recently proposed speedups. Furthermore, we show how GPU speedups may be mixed with multithreading to enable such speedups when the dataset is too large for GPU memory requirements; on a massive dense proteomics dataset of nearly a quarter-billion data instances, these mixed-architecture speedups reduce SVM analysis time from over half a week to less than a single day while using limited GPU memory.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Related papers
- Differentiable Optimization of Generalized Nondecomposable Functions using Linear ProgramsZihang Meng, Lopamudra Mukherjee, Yichao Wu, Vikas Singh et al.NeurIPS 2021 · 1 citation
- Giga-scale Kernel Matrix-Vector Multiplication on GPURobert Hu, Siu Lun Chau, Dino Sejdinovic, Joan GlaunèsNeurIPS 2022 · 3 citations
- SecureML: A System for Scalable Privacy-Preserving Machine LearningPayman Mohassel, Yupeng ZhangS&P 2017 · 2,107 citations
- Beyond L1: Faster and Better Sparse Models with skglmQuentin Bertrand, Quentin Klopfenstein, Pierre-Antoine Bannier, Gauthier Gidel et al.NeurIPS 2022 · 32 citations
- PRISM: Distribution-free Adaptive Computation of Matrix Functions for Accelerating Neural Network TrainingShenghao Yang, Zhichao Wang, Oleg Balabanov, N. Benjamin Erichson et al.ICML 2026 · 3 citations
