| Bandit-based optimization on graphs with application to library performance tuning |
| Full text |
Pdf
(672 KB)
|
| Source
|
ACM International Conference Proceeding Series; Vol. 382
archive
Proceedings of the 26th Annual International Conference on Machine Learning
table of contents
Montreal, Quebec, Canada
Pages: 729-736
Year of Publication: 2009
ISBN:978-1-60558-516-1
|
|
Authors
|
|
Frédéric de Mesmay
|
Carnegie Mellon University, Pittsburgh, PA
|
|
Arpad Rimmel
|
TAO (Inria), LRI, UMR (CNRS - Université Paris-Sud), Orsay, France
|
|
Yevgen Voronenko
|
Carnegie Mellon University, Pittsburgh, PA
|
|
Markus Püschel
|
Carnegie Mellon University, Pittsburgh, PA
|
|
| Sponsors |
|
| Publisher |
|
| Bibliometrics |
Downloads (6 Weeks): 5, Downloads (12 Months): 29, Citation Count: 0
|
|
|
ABSTRACT
The problem of choosing fast implementations for a class of recursive algorithms such as the fast Fourier transforms can be formulated as an optimization problem over the language generated by a suitably defined grammar. We propose a novel algorithm that solves this problem by reducing it to maximizing an objective function over the sinks of a directed acyclic graph. This algorithm valuates nodes using Monte-Carlo and grows a subgraph in the most promising directions by considering local maximum k-armed bandits. When used inside an adaptive linear transform library, it cuts down the search time by an order of magnitude compared to the existing algorithm. In some cases, the performance of the implementations found is also increased by up to 10% which is of considerable practical importance since it consequently improves the performance of all applications using the library.
REFERENCES
Note: OCR errors may be found in this Reference List extracted from the full text article. ACM has opted to expose the complete List rather than only correct and linked references.
| |
1
|
|
| |
2
|
Chaslot, G., De Jong, S., Saito, J.-T., & Uiterwijk, J. W. H. M. (2006). Monte-Carlo tree search in production management problems. Proc. 18th BeNeLux Conf. on Artificial Intel. (BNAIC) (pp. 91--98).
|
| |
3
|
Cicirello, V., & Smith, S. (2005). The max k-armed bandit: A new model for exploration applied to search heuristic selection. Proc. 20th National Conf. on Artificial Intelligence (AAAI) (pp. 1355--1361).
|
| |
4
|
Coulom, R. (2006). Efficient selectivity and backup operators in Monte-Carlo tree search. Proc. 5th Int. Conf. on Computers and Games (CG), 72--83.
|
| |
5
|
Frigo, M., & Johnson, S. (2005). The design and implementation of FFTW3. Proc. IEEE, 93, 216--231.
|
 |
6
|
|
| |
7
|
Intel (2008). Integrated performance primitives.
|
| |
8
|
|
| |
9
|
Kocsis, L., & Szepesvi, C. (2006). Bandit based Monte-Carlo planning. Euro. Conf. on Mach. Learn. (ECML), LNCS 4212 (pp. 282--293). Springer.
|
| |
10
|
Lai, T., & Robbins, H. (1985). Asymptotically efficient adaptive allocation rules. Advances in Applied Mathematics, 6, 4--22.
|
| |
11
|
Püschel, M., Moura, J. M. F., Johnson, J., Padua, D., Veloso, M., et al. (2005). SPIRAL: Code generation for DSP transforms. Proc. of the IEEE, 93, 232--275.
|
| |
12
|
|
| |
13
|
Streeter, M. J., & Smith, S. F. (2006). A simple distribution-free approach to the max k-armed bandit problem. Principles and Practice of Constraint Programming (CP) (pp. 560--574).
|
| |
14
|
|
| |
15
|
|
| |
16
|
Vuduc, R., Demmel, J. W., & Yelick, K. A. (2005). Oski: A library of automatically tuned sparse matrix kernels. Journal of Physics: Conf. Series, 16, 521--530.
|
| |
17
|
Wang, Y., & Gelly, S. (2007). Modifications of UCT and sequence-like simulations for Monte-Carlo Go. IEEE Symp. on Computational Intelligence and Games (CIG) (pp. 175--182).
|
| |
18
|
|
|