XAI-Based Feature Importance Analysis on Loop Optimization

Keichi Takahashi,Toshinobu Katayama,Yoichi Shimomura,Hiroyuki Takizawa
DOI: https://doi.org/10.1109/IPDPSW63119.2024.00142
2024-05-27
Abstract:In high-performance computing, performance tuning of a program is crucial and demanded in various situations. For instance, in image recognition applications using Deep Neural Networks (DNNs) such as autonomous driving, not only high accuracy but also high processing speed are required. Thus, Automatic Tuning (AT) has been attracting attention to automatically adjust the parameters of an application program that affect the program performance. In particular, in recent years, the utilization of AT leveraging Machine Learning (ML) models has been discussed. Although black-box models such as DNNs are adopted for advanced AT tasks, those models have highly complex structures, presenting challenges to humans in understanding how they are trained and how they make predictions. EXplainable Artificial Intelligence (XAI) refers to a class of techniques that enable humans to understand how ML models are trained and how ML models make predictions. In this paper, AT with an ML model is first developed to optimize nested loops in DNN inference for image recognition. Then, the feature importance that the model has learned is quantified using Permutation Feature Importance (PFI), which is one of XAI. The quantification aims to reveal the underlying relationship between performance parameters and execution time learned by the model in a human-understandable form, thereby enhancing the interpretability of the model. Additionally, this paper investigates whether important features revealed by XAI align with the tuning results. The evaluation results show that features with significant importance, as elucidated by XAI, significantly influence the program performance. Accordingly, the results demonstrate that important features revealed by XAI align with the tuning results, underscoring the usefulness of employing XAI in AT.
Computer Science
What problem does this paper attempt to address?