arrow
Return

Sparse Matrix Classification on Imbalanced Datasets Using Convolutional Neural Networks

delete2019-01-01
delete13
delete
OA
AI
J
Juan C. Pichel *
B
Beatriz Pateiro‐López
DOI:10.1109/ACCESS.2019.2924060delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
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 classification 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 classifiers are able to select the best performing format 92.8% of the time, obtaining 98.3% of the maximum attainable SpMV performance. A comparison to other state-of-the-art classification methods is also provided, demonstrating the benefits of our proposal.
Keywords:
Sparse matrix
classification
imbalance
deep learning
CNN
performance
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

U
Universidade de Santiago de Compostela
Scholars:
1.5W
Papers: 1.3W
Citations: 1.4W
Cited Papers

Cited Papers

Quantitative MRI texture analysis in chronic active multiple sclerosis lesions
err2021-06-01
err0
PREAI
errClaudia E. Weber; Matthias Wittayer; Matthias Kraemer; Andreas Dabringhaus; Michael Platten; Achim Gass; Philipp Eisele
errShare
errSave
A systematic study of the class imbalance problem in convolutional neural networks
err2018-10-01
err1.7K
errOAAI
errBuda, Mateusz; Maki, Atsuto; Mazurowski, Maciej A.
errShare
errSave
errShare
errSave
Long term changes of biodiversity of benthic macroalgae in the intertidal zone of the Nanji Islands
err2010-04-01
err0
PREAI
errJian-Zhang Sun; Xiu-Ren Ning; Feng-Feng Le; Wan-Dong Chen; Ding-Gen Zhuang
errShare
errSave
Arsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent
err2015-01-01
err0
PREAI
errTuan. A. Vu; Giang. H. Le; Canh. D. Dao; Lan. Q. Dang; Kien. T. Nguyen; Quang. K. Nguyen; Phuong. T. Dang; Hoa. T. K. Tran; Quang. T. Duong; Tuyen. V. Nguyen; Gun. D. Lee
errShare
errSave
Structural study of lanthanides(III) in aqueous nitrate and chloride solutions by EXAFS
err1999-02-01
err0
PREAI
errT. Yaita; H. Narita; Sh. Suzuki; Sh. Tachimori; H. Motohashi; H. Shiwaku
errShare
errSave
researcher View more