Batch-CAM: introduction to better reasoning in convolutional deep learning models

Ignesti G., Moroni D., Martinelli M.

Artificial Intelligence  Batch  CAM  Deep Learning 

Understanding the inner workings of deep learning models is crucial for advancing artificial intelligence, particularly in high-stakes fields such as healthcare, where accurate explanations are as vital as precision. This paper introduces Batch- CAM, a novel training paradigm that fuses a batch implementation of the Grad- CAM algorithm with a prototypical reconstruction loss. This combination guides the model to focus on salient image features, thereby enhancing its performance across classification tasks. Our results demonstrate that Batch-CAM achieves a simultaneous improvement in accuracy and image reconstruction quality while reducing training and inference times. By ensuring models learn from evidence- relevant information, this approach makes a relevant contribution to building more transparent, explainable, and trustworthy AI systems.

Metrics