Paper ID: 2410.20652

Visualizing attention zones in machine reading comprehension models

Yiming Cui, Wei-Nan Zhang, Ting Liu

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.

Submitted: Oct 28, 2024