arrow
Return

Gated spatial attention for entity linking in visually rich documents

delete2026-04-04
delete0
delete
OA
AI
P
Ping Wang
W
Wei Wang
X
Xuemin Gao
X
Xiaodong Li
X
Xu Zhang *
DOI:10.1016/j.hcc.2026.100396delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Entity linking in visually rich documents aims to identify semantic relationships between entities (e.g., key–value pairs) by jointly leveraging textual, visual, and spatial information. Despite the success of pre-trained document models such as LayoutLMv3, two challenges remain for relation extraction: (1) spatial position signals injected only at the input embedding layer tend to decay in deeper transformer layers, weakening the model’s ability to capture layout-dependent entity associations; and (2) in long documents, softmax attention distributes weights across many irrelevant tokens, diluting the focus on informative regions. To address these issues, we propose Gated Spatial Attention (GSA), a lightweight, plug-in framework on top of LayoutLMv3 that comprises two complementary modules: Spatial Position Enhancement (SPE), which injects ALiBi-style linear biases into every attention layer with head groups specialized for reading-order, horizontal, vertical, and semantic proximity, and Gated Attention (GA), which applies a per-token scalar gate after the scaled dot-product attention to suppress outputs from irrelevant tokens.Experiments on FUNSD and CORD demonstrate that GSA consistently improves both semantic entity recognition and relation extraction, achieving state-of-the-art results with negligible computational overhead.
Keywords:
Document understanding
Entity linking
Relation extraction
Gated spatial attention
Spatial position encoding
Visually rich documents
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

H
High-Confidence Computing
IF:
3
Papers:
239
Citations:
407

Organization

S
state grid intelligent technology co., ltd.
Scholars:
1
Papers: 1
Citations: 0
S
state grid shandong electric power company
Scholars:
64
Papers: 36
Citations: 0
S
State Grid Shandong Electric Power Company
Scholars:
117
Papers: 57
Citations: 0
researcher View more organizations