Sparse matrix classification on imbalanced datasets using convolutional neural networks

Abstract

This paper deals with the class imbalance problem in the context of the automatic selection of the best storage format for a sparse matrix with the aim of maximizing the performance of the sparse matrix vector multiplication (SpMV) on GPUs. Our classi cation method uses convolutional neural networks (CNNs) and proposes several solutions to mitigate the bias toward the majority classes when the data are not balanced. First, the CNNs are trained using images that represent the sparsity pattern of the matrices, whose pixels are colored according to different matrix features. In addition, we introduce a new network called SpNet, which achieves better results than a standard network as AlexNet in terms of prediction accuracy even having a more simple architecture. Finally, sampling techniques and cost-sensitive methods have been studied to give more emphasis on minority classes. The experiments conducted show that our classi ers are able to select the best performing format 92.8% of the time, obtaining 98.3% of the maximum attainable SpMV performance.Acomparison to other state-of-the-art classi cation methods is also provided, demonstrating the bene ts of our proposal