发现语言模型在分布外时因遗漏变量导致性能下降,提出新框架量化影响。
Omitted Variable Bias in Language Models Under Distribution Shift
- 区分可观测与不可观测的分布偏移,指出后者引发遗漏变量偏差。
- 构建理论边界,评估模型在分布偏移下的最差泛化性能。
- 实验证明该方法更可靠评估分布外表现,适合关注鲁棒性的研究者。
尽管现代语言模型在多种任务上表现优异,但在分布外数据上仍易出现脆弱行为。本文将分布偏移分为可观测与不可观测两部分,指出现有方法仅处理前者,而后者导致的遗漏变量偏差会损害模型评估与优化。为此,我们提出一个框架,将遗漏变量强度映射为分布偏移下语言模型最差泛化性能的边界。实验表明,直接使用这些边界进行评估与优化,可提供更严谨的分布外性能度量,相比传统调整方法显著提升真实分布外表现,并在目标分布标签可用时推断遗漏变量强度。
原文摘要 · Abstract (English)
Despite their impressive performance on a wide variety of tasks, modern language models remain susceptible to distribution shifts, exhibiting brittle behavior when evaluated on data that differs in distribution from their training data. In this paper, we describe how distribution shifts in language models can be separated into observable and unobservable components, and we discuss how established approaches for dealing with distribution shift address only the former. Importantly, we identify that the resulting omitted variable bias from unobserved variables can compromise both evaluation and optimization in language models. To address this challenge, we introduce a framework that maps the strength of the omitted variables to bounds on the worst-case generalization performance of language models under distribution shift. In empirical experiments, we show that using these bounds directly in language model evaluation and optimization provides more principled measures of out-of-distribution performance, improves true out-of-distribution performance relative to standard distribution shift adjustment methods, and further enables inference about the strength of the omitted variables when target distribution labels are available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。