ViTmiX: Vision Transformer Explainability Augmented by Mixed Visualization Methods

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hogea, Eduard, Onchis, Darian M., Coporan, Ana, Florea, Adina Magda, Istin, Codruta
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912161146601472
author Hogea, Eduard
Onchis, Darian M.
Coporan, Ana
Florea, Adina Magda
Istin, Codruta
author_facet Hogea, Eduard
Onchis, Darian M.
Coporan, Ana
Florea, Adina Magda
Istin, Codruta
contents Recent advancements in Vision Transformers (ViT) have demonstrated exceptional results in various visual recognition tasks, owing to their ability to capture long-range dependencies in images through self-attention mechanisms. However, the complex nature of ViT models requires robust explainability methods to unveil their decision-making processes. Explainable Artificial Intelligence (XAI) plays a crucial role in improving model transparency and trustworthiness by providing insights into model predictions. Current approaches to ViT explainability, based on visualization techniques such as Layer-wise Relevance Propagation (LRP) and gradient-based methods, have shown promising but sometimes limited results. In this study, we explore a hybrid approach that mixes multiple explainability techniques to overcome these limitations and enhance the interpretability of ViT models. Our experiments reveal that this hybrid approach significantly improves the interpretability of ViT models compared to individual methods. We also introduce modifications to existing techniques, such as using geometric mean for mixing, which demonstrates notable results in object segmentation tasks. To quantify the explainability gain, we introduced a novel post-hoc explainability measure by applying the Pigeonhole principle. These findings underscore the importance of refining and optimizing explainability methods for ViT models, paving the way to reliable XAI-based segmentations.
format Preprint
id arxiv_https___arxiv_org_abs_2412_14231
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ViTmiX: Vision Transformer Explainability Augmented by Mixed Visualization Methods
Hogea, Eduard
Onchis, Darian M.
Coporan, Ana
Florea, Adina Magda
Istin, Codruta
Computer Vision and Pattern Recognition
Recent advancements in Vision Transformers (ViT) have demonstrated exceptional results in various visual recognition tasks, owing to their ability to capture long-range dependencies in images through self-attention mechanisms. However, the complex nature of ViT models requires robust explainability methods to unveil their decision-making processes. Explainable Artificial Intelligence (XAI) plays a crucial role in improving model transparency and trustworthiness by providing insights into model predictions. Current approaches to ViT explainability, based on visualization techniques such as Layer-wise Relevance Propagation (LRP) and gradient-based methods, have shown promising but sometimes limited results. In this study, we explore a hybrid approach that mixes multiple explainability techniques to overcome these limitations and enhance the interpretability of ViT models. Our experiments reveal that this hybrid approach significantly improves the interpretability of ViT models compared to individual methods. We also introduce modifications to existing techniques, such as using geometric mean for mixing, which demonstrates notable results in object segmentation tasks. To quantify the explainability gain, we introduced a novel post-hoc explainability measure by applying the Pigeonhole principle. These findings underscore the importance of refining and optimizing explainability methods for ViT models, paving the way to reliable XAI-based segmentations.
title ViTmiX: Vision Transformer Explainability Augmented by Mixed Visualization Methods
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.14231