LLMs预测结果易受变量名等无关信息干扰,缺乏基本鲁棒性。
Robustness is Important: Limitations of LLMs for Data Fitting
- 通过改变变量名等无关信息,可使预测误差变化高达82%。
- 无论上下文学习还是微调,通用大模型均存在敏感性问题。
- 即使专为数据拟合设计的TabPFN也未能免疫此类干扰,适合关注鲁棒性的研究者参考。
大型语言模型(LLMs)正被广泛应用于数据拟合与预测任务,超越传统语言处理场景。已有研究表明,通过上下文学习或监督微调,LLMs在预测性能上可媲美多种表格监督学习方法。然而,我们发现使用LLMs进行数据拟合存在关键缺陷:对任务无关的数据表示变化极度敏感。例如,仅更改变量名称即可导致预测误差变化高达82%。这种敏感性在上下文学习和监督微调下均存在,且在闭源与开源通用大模型中均出现。通过分析开源模型的注意力分布,发现训练样本和变量名/值在提示中的位置影响其关注度,呈现非均匀注意力模式,解释了该敏感性成因。此外,针对数据拟合设计的先进基础模型TabPFN虽强调鲁棒性,仍无法完全抵御任务无关扰动。总体而言,尽管LLMs具备强大预测能力,但目前仍缺乏作为可靠数据拟合工具所必需的基本鲁棒性。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are being applied in a wide array of settings, well beyond the typical language-oriented use cases. In particular, LLMs are increasingly used as a plug-and-play method for fitting data and generating predictions. Prior work has shown that LLMs, via in-context learning or supervised fine-tuning, can perform competitively with many tabular supervised learning techniques in terms of predictive performance. However, we identify a critical vulnerability of using LLMs for data fitting -- making changes to data representation that are completely irrelevant to the underlying learning task can drastically alter LLMs' predictions on the same data. For example, simply changing variable names can sway the size of prediction error by as much as 82% in certain settings. Such prediction sensitivity with respect to task-irrelevant variations manifests under both in-context learning and supervised fine-tuning, for both close-weight and open-weight general-purpose LLMs. Moreover, by examining the attention scores of an open-weight LLM, we discover a non-uniform attention pattern: training examples and variable names/values which happen to occupy certain positions in the prompt receive more attention when output tokens are generated, even though different positions are expected to receive roughly the same attention. This partially explains the sensitivity in the presence of task-irrelevant variations. We also consider a state-of-the-art tabular foundation model (TabPFN) trained specifically for data fitting. Despite being explicitly designed to achieve prediction robustness, TabPFN is still not immune to task-irrelevant variations. Overall, despite LLMs' impressive predictive capabilities, currently they lack even the basic level of robustness to be used as a principled data-fitting tool.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。