Work

Roles and Utilization of Attention Heads in Transformer-based Neural Language Models

Year: 2020
Type: conference-paper
Abstract: Sentence encoders based on the transformer architecture have shown promising results on various natural language tasks.The main impetus lies in the pre-trained neural language models that capture long-range dependencies among words, owing to multi-head attention that is unique in the architecture.However, little is known for how linguistic properties are processed, represented, and utilized for downstream tasks among hundreds of attention heads inside the pre-trained transformerbased model.For the initial goal of examining the roles of attention heads in handling a set of linguistic features, we conducted a set of experiments with ten probing tasks and three downstream tasks on four pre-trained transformer families (GPT, GPT2, BERT, and ELECTRA).Meaningful insights are shown through the lens of heat map visualization and utilized to propose a relatively simple sentence representation method that takes advantage of most influential attention heads, resulting in additional performance improvements on the downstream tasks.
Language: English
Open Access status: hybrid
OpenAlex ID: W3034256339
FWCI: 1.754
Cites: 35
Cited by: 33
Related to: 10