English
Related papers

Related papers: On Explaining with Attention Matrices

200 papers

Attention-based models are successful when trained on large amounts of data. In this paper, we demonstrate that even in the low-resource scenario, attention can be learned effectively. To this end, we start with discrete human-annotated…

Computation and Language · Computer Science 2018-08-29 Yujia Bao , Shiyu Chang , Mo Yu , Regina Barzilay

While attention is all you need may be proving true, we do not know why: attention-based transformer models such as BERT are superior but how information flows from input tokens to output predictions are unclear. We introduce influence…

Computation and Language · Computer Science 2021-12-02 Kaiji Lu , Zifan Wang , Piotr Mardziel , Anupam Datta

Transformer model with multi-head attention requires caching intermediate results for efficient inference in generation tasks. However, cache brings new memory-related costs and prevents leveraging larger batch size for faster speed. We…

Computation and Language · Computer Science 2021-06-15 Yu Yan , Jiusheng Chen , Weizhen Qi , Nikhil Bhendawade , Yeyun Gong , Nan Duan , Ruofei Zhang

The use of Attention Layers has become a trend since the popularization of the Transformer-based models, being the key element for many state-of-the-art models that have been developed through recent years. However, one of the biggest…

Machine Learning · Computer Science 2023-02-14 Giovanni Araujo Bacochina , Rodrigo Clemente Thom de Souza

Pretrained language models based on the transformer architecture have shown great success in NLP. Textual training data often comes from the web and is thus tagged with time-specific information, but most language models ignore this…

Computation and Language · Computer Science 2022-05-05 Guy D. Rosin , Kira Radinsky

The self-attention mechanism has been a key factor in the advancement of vision Transformers. However, its quadratic complexity imposes a heavy computational burden in high-resolution scenarios, restricting the practical application.…

Computer Vision and Pattern Recognition · Computer Science 2025-12-29 Dongchen Han , Tianyu Li , Ziyi Wang , Gao Huang

Currently, attention mechanism becomes a standard fixture in most state-of-the-art natural language processing (NLP) models, not only due to outstanding performance it could gain, but also due to plausible innate explanation for the…

Computation and Language · Computer Science 2022-11-28 Lijie Hu , Yixin Liu , Ninghao Liu , Mengdi Huai , Lichao Sun , Di Wang

While attention has been an increasingly popular component in deep neural networks to both interpret and boost performance of models, little work has examined how attention progresses to accomplish a task and whether it is reasonable. In…

Computer Vision and Pattern Recognition · Computer Science 2020-07-30 Shi Chen , Ming Jiang , Jinhui Yang , Qi Zhao

The Efficient Adaptive Transformer (EAT) framework unifies three adaptive efficiency techniques - progressive token pruning, sparse attention, and dynamic early exiting - into a single, reproducible architecture for input-adaptive…

Computation and Language · Computer Science 2025-10-16 Jan Miller

Interpretability research often aims to predict how a model will respond to targeted interventions on specific mechanisms. However, it rarely predicts how a model will respond to unseen input data. This paper explores the promises and…

Machine Learning · Computer Science 2025-07-10 Victoria R. Li , Jenny Kaufmann , Martin Wattenberg , David Alvarez-Melis , Naomi Saphra

Attention mechanism in sequence-to-sequence models is designed to model the alignments between acoustic features and output tokens in speech recognition. However, attention weights produced by models trained end to end do not always…

Audio and Speech Processing · Electrical Eng. & Systems 2022-04-27 Gene-Ping Yang , Hao Tang

Despite the remarkable empirical performance of Transformers, their theoretical understanding remains elusive. Here, we consider a deep multi-head self-attention network, that is closely related to Transformers yet analytically tractable.…

Machine Learning · Computer Science 2024-12-10 Lorenzo Tiberi , Francesca Mignacco , Kazuki Irie , Haim Sompolinsky

Vision Transformers (ViTs) have become a dominant architecture in computer vision, yet their prediction process remains difficult to interpret because information is propagated through complex interactions across layers and attention heads.…

Computer Vision and Pattern Recognition · Computer Science 2026-04-21 Sehyeong Jo , Gangjae Jang , Haesol Park

We show the formal equivalence of linearised self-attention mechanisms and fast weight controllers from the early '90s, where a ``slow" neural net learns by gradient descent to program the ``fast weights" of another net through sequences of…

Machine Learning · Computer Science 2021-06-10 Imanol Schlag , Kazuki Irie , Jürgen Schmidhuber

While the Self-Attention mechanism in the Transformer model has proven to be effective in many domains, we observe that it is less effective in more diverse settings (e.g. multimodality) due to the varying granularity of each token and the…

Computer Vision and Pattern Recognition · Computer Science 2024-06-06 Wayner Barrios , SouYoung Jin

In this paper, we empirically analyze a simple, non-learnable, and nonparametric Nadaraya-Watson (NW) prediction head that can be used with any neural network architecture. In the NW head, the prediction is a weighted average of labels from…

Computer Vision and Pattern Recognition · Computer Science 2023-02-24 Alan Q. Wang , Mert R. Sabuncu

Attention-based mechanisms are widely used in machine learning, most prominently in transformers. However, hyperparameters such as the rank of the attention matrices and the number of heads are scaled nearly the same way in all realizations…

Machine Learning · Computer Science 2024-07-24 Noah Amsel , Gilad Yehudai , Joan Bruna

Attention networks show promise for both vision and language tasks, by emphasizing relationships between constituent elements through weighting functions. Such elements could be regions in an image output by a region proposal network, or…

Machine Learning · Computer Science 2019-10-07 Chu Wang , Babak Samari , Vladimir Kim , Siddhartha Chaudhuri , Kaleem Siddiqi

In this paper, we propose a novel gender bias detection method by utilizing attention map for transformer-based models. We 1) give an intuitive gender bias judgement method by comparing the different relation degree between the genders and…

Computation and Language · Computer Science 2021-11-01 Bingbing Li , Hongwu Peng , Rajat Sainju , Junhuan Yang , Lei Yang , Yueying Liang , Weiwen Jiang , Binghui Wang , Hang Liu , Caiwen Ding

Temporal causal analysis means understanding the underlying causes behind observed variables over time. Deep learning based methods such as transformers are increasingly used to capture temporal dynamics and causal relationships beyond mere…

Machine Learning · Computer Science 2024-11-21 Riya Mahesh , Rahul Vashisht , Chandrashekar Lakshminarayanan