arXiv · 2410.20652
Visualizing attention zones in machine reading comprehension models
Abstract
The attention mechanism plays an important role in the machine reading comprehension (MRC) model. Here, we describe a pipeline for building an MRC model with a pretrained language model and visualizing the effect of each attention zone in different layers, which can indicate the explainability of the model. With the presented protocol and accompanying code, researchers can easily visualize the relevance of each attention zone in the MRC model. This approach can be generalized to other pretrained language models.
Explore related subjects
Keep this discovery
Yiming Cui, Wei-Nan Zhang, Ting Liu. 2024-10-28. Visualizing attention zones in machine reading comprehension models. https://doi.org/10.1016/j.xpro.2022.101481
Cite the original work for its findings. Save a collection to share your selection of sources.