HEAT: a heterogeneous entity-aware transformer for efficient and accurate cloze-style machine reading comprehension
摘要
Transformer models excel in machine reading comprehension but struggle with complex reasoning tasks due to the lack of explicit relational knowledge in input sequences. Existing solutions often rely on external knowledge graphs, introducing challenges in knowledge selection, availability, and processing overhead. To address this, we propose HEAT (Heterogeneous Entity-Aware Transformer), a novel method that integrates reasoning knowledge from a document-derived heterogeneous graph directly into the transformer’s self-attention mechanism, eliminating external dependencies. HEAT employs a fusion attention pattern with three components: global-local attention for word tokens, graph-based attention prioritizing entity tokens connected in the graph, and relationship-aware attention between entity and word tokens. These are enhanced by relative position labels integrated into LUKE’s entity-aware self-attention. Evaluated on ReCoRD (commonsense reasoning), WikiHop (multi-hop reasoning), and CliCR (clinical reasoning), HEAT achieves superior performance, surpassing LUKE by 1.1%, 5.6%, and 2.2%, respectively, and outperforming LUKE-Graph, demonstrating its effectiveness in reasoning-intensive tasks.