arXiv:2603.03389cs.LG2026-03被引 1

用图神经网络增强词向量聚合,让句子表示更准更快

Towards Improved Sentence Representations using Token Graphs

  • 构建词间相似图,用图网络优化词向量再聚合
  • 90%干扰词下仍保持97%准确率,远超传统方法
  • 参数少20倍、训练快100倍,适合快速部署

从大语言模型的词级输出中获取单向量句子表示是绝大多数句级任务的关键步骤。然而,标准的平均或最大池化方法将词视为独立集合,忽略了模型自注意力层捕捉的丰富关系结构,易受信号稀释影响。为此,我们提出GLOT——一种轻量级、结构感知的池化模块,将池化重定义为关系学习与聚合的结合。在冻结的LLM输出上,GLOT先构建潜在词相似图,再用图神经网络精炼词表示,最后通过读出层聚合。实验表明,该方法极为稳健高效:在90%词为随机干扰的诊断测试中,准确率仍超97%,而基线方法性能崩溃;在GLUE和MTEB等基准上表现媲美最先进方法,参数量减少20倍,训练速度提升超100倍。理论分析证实其表达能力,证明基于词图的学习是高效适配冻结LLM的强大范式。代码已开源。

原文摘要 · Abstract (English)

Obtaining a single-vector representation from a Large Language Model's (LLM) token-level outputs is a critical step for nearly all sentence-level tasks. However, standard pooling methods like mean or max aggregation treat tokens as an independent set, discarding the rich relational structure captured by the model's self-attention layers and making them susceptible to signal dilution. To address this, we introduce GLOT, a lightweight, structure-aware pooling module that reframes pooling as relational learning followed by aggregation. Operating on the outputs of a frozen LLM, GLOT first constructs a latent token-similarity graph, then refines token representations with a graph neural network, and finally aggregates them using a readout layer. Experimentally, our approach is remarkably robust and efficient: on a diagnostic stress test where 90% of tokens are random distractors, GLOT maintains over 97% accuracy while baseline methods collapse. Furthermore, it is competitive with state-of-the-art techniques on benchmarks like GLUE and MTEB with 20x fewer trainable parameters and speeds up the training time by over 100x compared with parameter-efficient fine-tuning methods. Supported by a theoretical analysis of its expressive power, our work shows that learning over token graphs is a powerful paradigm for the efficient adaptation of frozen LLMs. Our code is published at https://github.com/ipsitmantri/GLOT.

句子表示图神经网络高效微调

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