arXiv · 2101.09808
Analytical Characterization and Design Space Exploration for Optimization of CNNs
Abstract
Moving data through the memory hierarchy is a fundamental bottleneck that can limit the performance of core algorithms of machine learning, such as convolutional neural networks (CNNs). Loop-level optimization, including loop tiling and loop permutation, are fundamental transformations to reduce data movement. However, the search space for finding the best loop-level optimization configuration is explosively large. This paper develops an analytical modeling approach for finding the best loop-level optimization configuration for CNNs on multi-core CPUs. Experimental evaluation shows that this approach achieves comparable or better performance than state-of-the-art libraries and auto-tuning based optimizers for CNNs.
Explore related subjects
Keep this discovery
Rui Li, Yufan Xu, Aravind Sukumaran-Rajam, Atanas Rountev, P. Sadayappan. 2021-01-24. Analytical Characterization and Design Space Exploration for Optimization of CNNs. https://doi.org/10.1145/3445814.3446759
Cite the original work for its findings. Save a collection to share your selection of sources.