Parallel Bayesian ARTMAP and Its OpenCL Implementation

摘要

The Bayesian ARTMAP neural network, introduced by Vigdor and Lerner, is an incremental learning algorithm which can efficiently process massive datasets for classification, regression, and probabilistic inference tasks. We introduce the parallelized version of the BA neural network and implement it in OpenCL. Our implementation runs on both multi-core CPUs and GPUs architectures. We test the Parallel Bayesian ARTMAP on several classification and regression benchmarks focusing on speedup and scalability. In some cases, the parallel BA runs by an order of magnitude faster than the sequential implementation. Our implementation has the potential to scale for OpenCL devices with increasing number of compute units.