CUDA-based acceleration of KNN and Bayesian algorithms

Konferenz: CIBDA 2022 - 3rd International Conference on Computer Information and Big Data Applications
25.03.2022 - 27.03.2022 in Wuhan, China

Tagungsband: CIBDA 2022

Seiten: 4Sprache: EnglischTyp: PDF

Autoren:
Jia, Peiyan; Chen, Huiping; Huang, Yabo (College of Computer and Information Engineering, Henan University, Kaifeng, Henan Province, China)

Inhalt:
Machine learning algorithms are data-driven algorithms that have produced significant advancements in areas like natural language processing, image classification, and machine translation. The computation time, on the other hand, climbs exponentially as the dataset grows. This paper proposes a method of utilizing CUDA to employ GPU parallel computing to cut the computational complexity of frequently used algorithms such as the K-nearest neighbor (KNN) and Naive Bayes (NB), in an effort to expand the algorithm's computational performance. On the basis of assuring the algorithm's correctness, this method considerably reduces the algorithm's running time. Experiments show that the method proposed in this paper can reduce the complexity of KNN from O(N2) to O(N), and also the complexity of the NB algorithm from O(N) to O(1).