ExplainabilityTransparencyTrustworthinesscs.LG
Improving Deep Neural Network Classification Confidence using Heatmap-based eXplainable AI
Erico Tjoa, Hong Jing Khok, Tushar Chouhan, Guan Cuntai
December 30, 2021
arXiv.org
6 citations
1 influential
Key Takeaways
-
Different explanation methods work better for different tasks: The paper finds that the best XAI method varies by dataset—Saliency maps performed best for ImageNet while Deconvolution worked better for medical imaging (Chest X-Ray), suggesting that organizations need to test multiple explanation approaches rather than relying on a one-size-fits-all solution.
-
Good explanations can increase AI system reliability: The research demonstrates that heatmap-based explanations don't just help humans understand AI decisions—they can actually improve the AI model's confidence in making correct predictions, suggesting that explainability and accuracy are complementary rather than competing goals.
-
Explanations reveal patterns in AI errors: The paper introduces a "gap distribution" metric that shows stark differences between how the AI system explains correct versus incorrect predictions, providing a potential tool for identifying when an AI system is likely to make mistakes before deployment.
-
New methods can generate better explanations: The authors introduce "generative augmentative explanation," a technique capable of producing heatmaps that significantly improve prediction confidence, indicating that explanation quality is not fixed but can be actively improved through better methods.
Abstract
This paper quantifies the quality of heatmap-based eXplainable AI (XAI) methods w.r.t image classification problem. Here, a heatmap is considered desirable if it improves the probability of predicting the correct classes. Different XAI heatmap-based methods are empirically shown to improve classification confidence to different extents depending on the datasets, e.g. Saliency works best on ImageNet and Deconvolution on Chest X-Ray Pneumonia dataset. The novelty includes a new gap distribution that shows a stark difference between correct and wrong predictions. Finally, the generative augmentative explanation is introduced, a method to generate heatmaps capable of improving predictive confidence to a high level.