Title
End-to-End Deep Learning of Optimization Heuristics
Abstract
Accurate automatic optimization heuristics are necessary for dealing with thecomplexity and diversity of modern hardware and software. Machine learning is aproven technique for learning such heuristics, but its success is bound by thequality of the features used. These features must be hand crafted by developersthrough a combination of expert domain knowledge and trial and error. This makesthe quality of the final model directly dependent on the skill and availabletime of the system architect. Our work introduces a better way for building heuristics. We develop a deepneural network that learns heuristics over raw code, entirely without using codefeatures. The neural network simultaneously constructs appropriaterepresentations of the code and learns how best to optimize, removing the needfor manual feature creation. Further, we show that our neural nets can transferlearning from one optimization problem to another, improving the accuracy of newmodels, without the help of human experts. We compare the effectiveness of our automatically generated heuristics againstones with features hand-picked by experts. We examine two challenging tasks:predicting optimal mapping for heterogeneous parallelism and GPU threadcoarsening factors. In 89% of the cases, the quality of our fully automaticheuristics matches or surpasses that of state-of-the-art predictive models usinghand-crafted features, providing on average 14% and 12% more performance withno human effort expended on designing features.
Year
DOI
Venue
2017
10.1109/PACT.2017.24
2017 26th International Conference on Parallel Architectures and Compilation Techniques (PACT)
Keywords
Field
DocType
Optimization Heuristics,Machine Learning,Compiler Optimizations,Heterogeneous Systems
Trial and error,Theano,Domain knowledge,Computer science,Parallel computing,Feature extraction,Heuristics,Artificial intelligence,Deep learning,Artificial neural network,Optimization problem,Machine learning
Conference
ISSN
ISBN
Citations 
1089-795X
978-1-5090-6765-7
23
PageRank 
References 
Authors
0.74
43
4
Name
Order
Citations
PageRank
chris cummins1544.55
Pavlos Petoumenos220013.23
Zheng Wang321518.10
Hugh Leather418214.33