Expected Integral Discrete Gradient: Diagnosing Autonomous Driving Model

Meng Li,Hengyang Sun,Zhihao Cui,Yanjun Huang,Hong Chen
DOI: https://doi.org/10.1109/TVT.2024.3436614
IF: 6.8
2024-01-01
IEEE Transactions on Vehicular Technology
Abstract:The demand for explainability in complex machine learning (ML) models is ever more pressing, particularly within safety-critical domains like autonomous driving. The Integrated Gradient (IG), a prominent attribution-based explainable artificial intelligence method, offers an effective solution for explaining and diagnosing ML models. However, IG is primarily designed for deep neural network models, hindering its broader applicability to various structured machine-learning models. Moreover, the issue of selecting a suitable single baseline point in IG methods still needs to be solved. In response to these challenges, this paper introduces a model-agnostic explainable technique called the Expected Integral Discrete Gradient (EIDG). This approach extends the capabilities of IG to encompass a wide range of machine-learning models by leveraging numerical differentiation. It replaces the previous single baseline point scheme with a distributed multi-baseline method to reveal how varying baselines affect the output. Our method is thoroughly evaluated on standard machine learning models, targeting scenarios in autonomous driving scenarios, thereby validating its effectiveness in explaining and diagnosing models. Our work will inspire and equip developers and users with the necessary tools to promote the adoption of attribution explanations across various machine-learning domains. The code for EIDG is available at https://github.com/limeng-1234/EIDG.
What problem does this paper attempt to address?