剖析大模型排序机制,揭示其内部编码的检索特征与盲区
Probing Ranking LLMs: A Mechanistic Analysis for Information Retrieval
- 通过探测神经元激活分析大模型内部特征表示
- 发现模型显著编码词法、语义等关键检索特征
- 适合关注检索可解释性与模型泛化能力的研究者
Transformer 网络,尤其是性能接近 GPT 的大型语言模型,以强大的特征提取能力著称,但其提取特征的本质及其与人工设计特征的对齐关系尚不明确。本文针对最先进的微调排序大模型,开展基于探测的机制分析,研究其神经元激活中是否包含已知的人工特征与语义特征。分析覆盖词法信号、文档结构、查询-文档交互以及复杂语义表示等多个维度,旨在揭示影响排序决策的潜在模式。在四个不同排序大模型上的实验表明,部分统计信息检索特征在模型激活中被显著编码,而另一些特征则明显缺失。此外,我们分析了模型对分布外查询与文档的响应,揭示了不同的泛化行为。通过解构大模型激活中的潜在表征,本研究旨在提升排序模型的可解释性与有效性。研究结果为构建更透明、可靠的检索系统提供了关键洞察,并公开所有必要代码与脚本以支持后续探索。
原文摘要 · Abstract (English)
Transformer networks, particularly those achieving performance comparable to GPT models, are well known for their robust feature extraction abilities. However, the nature of these extracted features and their alignment with human-engineered ones remain unexplored. In this work, we investigate the internal mechanisms of state-of-the-art, fine-tuned LLMs for passage reranking. We employ a probing-based analysis to examine neuron activations in ranking LLMs, identifying the presence of known human-engineered and semantic features. Our study spans a broad range of feature categories, including lexical signals, document structure, query-document interactions, and complex semantic representations, to uncover underlying patterns influencing ranking decisions. Through experiments on four different ranking LLMs, we identify statistical IR features that are prominently encoded in LLM activations, as well as others that are notably missing. Furthermore, we analyze how these models respond to out-of-distribution queries and documents, revealing distinct generalization behaviors. By dissecting the latent representations within LLM activations, we aim to improve both the interpretability and effectiveness of ranking models. Our findings offer crucial insights for developing more transparent and reliable retrieval systems, and we release all necessary scripts and code to support further exploration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。