Title
Preconditioner Auto-Tuning Using Deep Learning for Sparse Iterative Algorithms
Abstract
In numerical libraries for sparse matrix operations, there are many tuning parameters related to implementation selection. Selection of different tuning parameters could result in totally different performance. Moreover, optimal implementation depends on the sparse matrices to be operated. It is difficult to find optimal implementation without executing each implementation and thereby examining its performance on a given sparse matrix. In this study, we propose an implementation selection method for sparse iterative algorithms and preconditioners in a numerical library using deep learning. The proposed method uses full color images to represent the features of a sparse matrix. We present an image generation method for partitioning a given matrix (to generate its feature image) so that the value of each matrix element is considered in the implementation selection. We then evaluate the effectiveness of the proposed method by conducting a numerical experiment. In this experiment, the accuracy of implementation selection is evaluated. The training data comprise a pair of sparse matrix and its optimal implementation. The optimal implementation of each sparse matrix in the training data is obtained in advance by executing every implementation and getting the best one. The experimental results obtained using the proposed method show that the accuracy of selecting the optimal implementation of each sparse matrix is 79.5%.
Year
DOI
Venue
2018
10.1109/CANDARW.2018.00055
2018 Sixth International Symposium on Computing and Networking Workshops (CANDARW)
Keywords
Field
DocType
auto-tuning,preconditioner selection,GMRES,Xabclib,deep learning
Training set,Image generation,Generalized minimal residual method,Preconditioner,Matrix (mathematics),Computer science,Algorithm,Artificial intelligence,Deep learning,Auto tuning,Sparse matrix
Conference
ISBN
Citations 
PageRank 
978-1-5386-9185-4
0
0.34
References 
Authors
1
7
Name
Order
Citations
PageRank
Kenya Yamada100.34
Takahiro Katagiri212117.01
Hiroyuki Takizawa327346.54
Kazuo Minami4608.57
Mitsuo Yokokawa522751.71
Toru Nagai602.03
Masao Ogino713.42