FAME: Feature Activation Map Explanation on Image Classification and Face Recognition
About
Deep Learning has revolutionized machine learning, reaching unprecedented levels of accuracy, but at the cost of reduced interpretability. Especially in image processing systems, deep networks transform local pixel information into more global concepts in a highly obscured manner. Explainable AI methods for image processing try to shed light on this issue by highlighting the regions of the image that are important for the prediction task. Among these, Class Activation Mapping (CAM) and its gradient-based variants compute attributions based on the feature map and upscale them to the image resolution, assuming that feature map locations are influenced only by underlying regions. Perturbation-based methods, such as CorrRISE, on the other hand, try to provide pixel-level attributions by perturbing the input with fixed patches and checking how the output of the network changes. In this work, we propose Feature Activation Map Explanation (FAME), which combines both worlds by using network gradients to compute changes to the input image, manipulating it in a gradient-driven way rather than using fixed patches. We apply this technique on two common tasks, image classification and face recognition, and show that CAM's above-mentioned assumption does not hold for deeper networks. We qualitatively and quantitively show that FAME produces attribution maps that are competitive state-of-the-art systems. Our code is available: {\footnotesize https://github.com/AIML-IfI/fame.}
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Face recognition attribution | ARface (glass) | Delete Rate53.06 | 15 | |
| Face recognition attribution | SCface far | Deletion Score53.43 | 15 | |
| Face recognition attribution | ARface (neutral) | Deletion Score50.77 | 15 | |
| Face recognition attribution | SCface medium | Deletion Score57.19 | 15 | |
| Face recognition attribution | CFP-FP | Deletion Score55.82 | 15 | |
| Face recognition attribution | CFP FF | Deletion Score65.7 | 15 | |
| Face recognition attribution | ARface (scarf) | Deletion Score55.13 | 15 | |
| Face recognition attribution | SCface (close) | Deletion Score58.31 | 15 | |
| Image Classification | ImageNet | IoU (thr=0.3)46.09 | 5 | |
| Face recognition attribution | CFP-FP | Runtime (s)2.12e+3 | 5 |