基于图形处理器的径向基函数极限学习机无监督特征学习分类

Unsupervised Feature Learning Classification With Radial Basis Function Extreme Learning Machine Using Graphic Processors

IEEE Transactions on Cybernetics · 2016
被引 30
ABS 3

中文导读

提出一种结合无监督特征学习和径向基函数极限学习机的方法,利用GPU并行计算加速,在保证高精度的同时实现最高20倍的速度提升。

Abstract

Ever-increasing size and complexity of data sets create challenges and potential tradeoffs of accuracy and speed in learning algorithms. This paper offers progress on both fronts. It presents a mechanism to train the unsupervised learning features learned from only one layer to improve performance in both speed and accuracy. The features are learned by an unsupervised feature learning (UFL) algorithm. Then, those features are trained by a fast radial basis function (RBF) extreme learning machine (ELM). By exploiting the massive parallel computing attribute of modern graphics processing unit, a customized compute unified device architecture (CUDA) kernel is developed to further speed up the computing of the RBF kernel in the ELM. Results tested on Canadian Institute for Advanced Research and Mixed National Institute of Standards and Technology data sets confirm the UFL RBF ELM achieves high accuracy, and the CUDA implementation is up to 20 times faster than CPU and the naive parallel approach.

机器学习无监督学习并行计算图形处理器