arXiv:2505.23790cs.CLcs.AI2025-05被引 1

用互信息衡量大模型理解力,发现编码器模型更擅长保留信息。

Rethinking the Understanding Ability across LLMs through Mutual Information

  • 通过互信息量化输入与隐向量的关联,评估模型理解能力。
  • 编码器模型比解码器模型信息保留更好,后者存在后期遗忘现象。
  • 提升信息可恢复性可增强模型泛化能力,适合无监督任务。

大语言模型(LLMs)在自然语言处理中取得突破,但其内在语言理解能力的评估仍具挑战性。本文提出基于互信息(MI)的信息理论框架,将理解能力定义为输入句子与隐表示之间的互信息(句级MI),衡量信息在隐表示中的保留程度。由于LLMs为单个词元学习嵌入,我们将其分解为词元与句子嵌入间的词元级MI,并建立理论边界。基于法诺不等式,推导出可计算的词元级MI下界,直接关联词元可恢复性——即从句子嵌入中重构原始词元的能力。通过实现该可恢复性任务,对比不同LLMs的MI表现,发现编码器模型始终优于解码器模型,后者呈现晚层“遗忘”模式:互信息先上升后下降。此外,通过微调最大化词元级可恢复性,显著提升模型在无任务监督下的理解能力,表明互信息可作为理解与优化语言模型能力的基础。

原文摘要 · Abstract (English)

Recent advances in large language models (LLMs) have revolutionized natural language processing, yet evaluating their intrinsic linguistic understanding remains challenging. Moving beyond specialized evaluation tasks, we propose an information-theoretic framework grounded in mutual information (MI) to achieve this. We formalize the understanding as MI between an input sentence and its latent representation (sentence-level MI), measuring how effectively input information is preserved in latent representation. Given that LLMs learn embeddings for individual tokens, we decompose sentence-level MI into token-level MI between tokens and sentence embeddings, establishing theoretical bounds connecting these measures. Based on this foundation, we theoretically derive a computable lower bound for token-level MI using Fano's inequality, which directly relates to token-level recoverability-the ability to predict original tokens from sentence embedding. We implement this recoverability task to comparatively measure MI across different LLMs, revealing that encoder-only models consistently maintain higher information fidelity than their decoder-only counterparts, with the latter exhibiting a distinctive late-layer "forgetting" pattern where mutual information is first enhanced and then discarded. Moreover, fine-tuning to maximize token-level recoverability consistently improves understanding ability of LLMs on tasks without task-specific supervision, demonstrating that mutual information can serve as a foundation for understanding and improving language model capabilities.

大模型理解互信息编码器可恢复性

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