arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.00630cs.CVcs.AI

STAA:用于实时解释基于Transformer的视频模型的时空注意力归因

STAA: Spatio-Temporal Attention Attribution for Real-Time Interpreting Transformer-based Video Models

Zerui Wang, Yan Liu

AI总结:

针对现有视频可解释AI方法仅能提供单维解释、计算复杂度高的问题,本文提出STAA方法,从Transformer注意力值同时获取时空解释,经优化后信噪比更高,计算开销不足传统方法3%,可用于实时视频分析。

摘要:

基于Transformer的模型已在包括图像与视频分析在内的各类计算机视觉任务中取得了当前最优(state-of-the-art)性能。然而,Transformer复杂的架构与黑箱特性给可解释性带来了挑战,而可解释性是现实应用与科学研究的关键要素。现有的可解释人工智能(XAI)方法仅能提供一维的特征重要性,要么是空间解释要么是时间解释,且计算复杂度极高。\n本文提出STAA(Spatio-Temporal Attention Attribution,时空注意力归因),一种用于解释视频Transformer模型的XAI方法。与传统方法分别将图像XAI技术用于空间特征、或将片段贡献分析用于时间维度的做法不同,STAA可从Transformer中的注意力值同时获取空间与时间信息。本研究使用Kinetics-400数据集——一个包含400类人类动作的基准数据集,用于动作识别研究。我们提出了量化解释效果的指标,还通过优化提升了STAA的原始输出:通过引入动态阈值与注意力聚焦机制,我们提升了解释结果的信噪比,实现了更精准的可视化效果与更优的评估结果。在计算开销方面,我们的方法所需计算资源不到传统XAI方法的3%,适用于实时视频XAI分析应用。STAA为不断发展的XAI领域做出了贡献,为研究人员与从业者提供了一种分析Transformer模型的方法。

英文摘要:

Transformer-based models have achieved state-of-the-art performance in various computer vision tasks, including image and video analysis. However, Transformer's complex architecture and black-box nature pose challenges for explainability, a crucial aspect for real-world applications and scientific inquiry. Current Explainable AI (XAI) methods can only provide one-dimensional feature importance, either spatial or temporal explanation, with significant computational complexity. This paper introduces STAA (Spatio-Temporal Attention Attribution), an XAI method for interpreting video Transformer models. Differ from traditional methods that separately apply image XAI techniques for spatial features or segment contribution analysis for temporal aspects, STAA offers both spatial and temporal information simultaneously from attention values in Transformers. The study utilizes the Kinetics-400 dataset, a benchmark collection of 400 human action classes used for action recognition research. We introduce metrics to quantify explanations. We also apply optimization to enhance STAA's raw output. By implementing dynamic thresholding and attention focusing mechanisms, we improve the signal-to-noise ratio in our explanations, resulting in more precise visualizations and better evaluation results. In terms of computational overhead, our method requires less than 3\% of the computational resources of traditional XAI methods, making it suitable for real-time video XAI analysis applications. STAA contributes to the growing field of XAI by offering a method for researchers and practitioners to analyze Transformer models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑