Implicit Statistical Inference in Transformers: Approximating Likelihood-Ratio Tests In-Context
AI 摘要
研究Transformer在上下文学习中如何进行隐式统计推断,并发现其近似似然比检验。
主要贡献
- 揭示了Transformer在上下文学习中进行隐式统计推断的机制
- 证明了Transformer能够从上下文中逼近贝叶斯最优充分统计量
- 发现模型会根据任务类型调整决策方式,而非仅依赖固定核平滑
方法论
通过在具有不同几何结构的二元假设检验任务上训练Transformer,并进行logit lens和电路对齐的机制分析。
原文摘要
In-context learning (ICL) allows Transformers to adapt to novel tasks without weight updates, yet the underlying algorithms remain poorly understood. We adopt a statistical decision-theoretic perspective by investigating simple binary hypothesis testing, where the optimal policy is determined by the likelihood-ratio test. Notably, this setup provides a mathematically rigorous setting for mechanistic interpretability where the target algorithmic ground truth is known. By training Transformers on tasks requiring distinct geometries (linear shifted means vs. nonlinear variance estimation), we demonstrate that the models approximate the Bayes-optimal sufficient statistics from context up to some monotonic transformation, matching the performance of an ideal oracle estimator in nonlinear regimes. Leveraging this analytical ground truth, mechanistic analysis via logit lens and circuit alignment suggests that the model does not rely on a fixed kernel smoothing heuristic. Instead, it appears to adapt the point at which decisions become linearly decodable: exhibiting patterns consistent with a voting-style ensemble for linear tasks while utilizing a deeper sequential computation for nonlinear tasks. These findings suggest that ICL emerges from the construction of task-adaptive statistical estimators rather than simple similarity matching.