1. [Publications](/publications)
2. Demystifying Map Space Exploration for NPUs
 
 # Demystifying Map Space Exploration for NPUs

  ![Publication image](/sites/default/files/styles/wide/public/default_images/default.jpeg?itok=qUFsuJCP "Publication image")

 Map Space Exploration is the problem of finding optimized mappings of a Deep Neural Network (DNN) model on an accelerator. It is known to be extremely computationally expensive, and there has been active research looking at both heuristics and learning-based methods to make the problem computationally tractable. However, while there are dozens of mappers out there (all empirically claiming to find better mappings than others), the research community lacks systematic insights on how different search techniques navigate the map-space and how different mapping axes contribute to the accelerator’s performance and efficiency. Such insights are crucial to developing mapping frameworks for emerging DNNs that are increasingly irregular (due to neural architecture search) and sparse, making the corresponding map spaces much more complex. In this work, rather than proposing yet another mapper, we do a first-of-its-kind apples-to-apples comparison of search techniques leveraged by different mappers. Next, we extract the learnings from our study and propose two new techniques that can augment existing mappers — warm-start and sparsity-aware — that demonstrate speedups, scalability, and robustness across diverse DNN models.



 ## Authors



Sheng-Chun Kao (Georgia Institute of Technology)

[Angshuman Parashar](/person/angshuman-parashar)

[Po-An Tsai](/person/po-an-tsai)

Tushar Krishna (Georgia Institute of Technology)

 

 

 ## Publication Date



Sunday, November 6, 2022

 

 ## Published in



[International Symposium on Workload Characterization (IISWC)](https://ieeexplore.ieee.org/document/9975389)

 

 ## Research Area



[Artificial Intelligence and Machine Learning ](/research-area/machine-learning-artificial-intelligence)

[Computer Architecture](/research-area/computer-architecture)

[Programming Languages, Systems and Tools](/research-area/programming-languages-systems)

 

 

 ## External Links



[IEEE Digital Library](https://ieeexplore.ieee.org/document/9975389)

 

 

 ## Uploaded Files



[Published manuscript](https://d1qx31qr3h6wln.cloudfront.net/publications/IISWC_2022_Gamma_Timeloop.pdf "Open file in new window")984.74 KB

 

 

 ## Copyright



This material is posted here with permission of the IEEE. Internal or personal use of this material is permitted. However, permission to reprint/republish this material for advertising or promotional purposes or for creating new collective works for resale or redistribution must be obtained from the IEEE by writing to <pubs-permissions@ieee.org>.