arXiv · 1904.02679
Visualizing Attention in Transformer-Based Language Representation Models
Abstract
We present an open-source tool for visualizing multi-head self-attention in Transformer-based language representation models. The tool extends earlier work by visualizing attention at three levels of granularity: the attention-head level, the model level, and the neuron level. We describe how each of these views can help to interpret the model, and we demonstrate the tool on the BERT model and the OpenAI GPT-2 model. We also present three use cases for analyzing GPT-2: detecting model bias, identifying recurring patterns, and linking neurons to model behavior.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Jesse Vig. 2019-04-11. Visualizing Attention in Transformer-Based Language Representation Models. https://arxiv.org/abs/1904.02679
Cite the original work for its findings. Save a collection to share your selection of sources.