arXiv:2508.02532cs.CLcs.LG2025-08被引 2

用图神经网络增强文本模型,提升工程文档信息提取精度。

Contextual Graph Transformer: A Small Language Model for Enhanced Engineering Document Information Extraction

  • 结合图网络与Transformer,动态建模文本结构和语义关系。
  • 在技术文档上比GPT-2高24.7%准确率,参数少62.4%。
  • 适合需要精准实体识别与结构理解的工程领域应用。

标准Transformer语言模型虽在通用文本上表现强劲,但在复杂工程文档的细粒度语法与实体关系处理上常显不足。为此,我们提出上下文图变压器(CGT),一种融合图神经网络(GNN)与Transformer的混合架构,用于特定领域的问答任务。CGT通过序列、跳跃词与语义相似性边构建输入标记的动态图,由GATv2Conv层进行局部结构学习,再将增强嵌入送入Transformer编码器以捕捉全局依赖。相较于通用大模型,技术领域更需具备强上下文感知与结构敏感性的专用模型。CGT提供了一种参数高效解决方案。集成于检索增强生成(RAG)流程中,其性能超越基线如GPT-2与BERT,准确率高出GPT-2 24.7%,参数量仅为后者的37.6%。该优势源于模型对标记间结构互动与长程语义连贯性的联合建模能力。模型采用两阶段训练:先在通用文本上预训练,再在领域特定手册上微调,凸显其对技术语言的适应性,实现更佳的语义锚定、实体追踪与检索增强响应,在真实场景中表现优异。

原文摘要 · Abstract (English)

Standard transformer-based language models, while powerful for general text, often struggle with the fine-grained syntax and entity relationships in complex technical, engineering documents. To address this, we propose the Contextual Graph Transformer (CGT), a hybrid neural architecture that combines Graph Neural Networks (GNNs) and Transformers for domain-specific question answering. CGT constructs a dynamic graph over input tokens using sequential, skip-gram, and semantic similarity edges, which is processed by GATv2Conv layers for local structure learning. These enriched embeddings are then passed to a Transformer encoder to capture global dependencies. Unlike generic large models, technical domains often require specialized language models with stronger contextualization and structure awareness. CGT offers a parameter-efficient solution for such use cases. Integrated into a Retrieval-Augmented Generation (RAG) pipeline, CGT outperforms baselines like GPT-2 and BERT, achieving 24.7% higher accuracy than GPT-2 with 62.4% fewer parameters. This gain stems from CGTs ability to jointly model structural token interactions and long-range semantic coherence. The model is trained from scratch using a two-phase approach: pretraining on general text followed by fine-tuning on domain-specific manuals. This highlights CGTs adaptability to technical language, enabling better grounding, entity tracking, and retrieval-augmented responses in real-world applications.

信息抽取图神经网络小模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。