发现语言模型性能与权重几何关系不大,更多反映训练设置。
Disentangling Geometry, Performance, and Training in Language Models
- 分析108个可控训练的模型,考察解码矩阵有效秩与性能关系。
- 高有效秩常伴随高性能,但非普遍规律,且低秩不必然导致性能下降。
- 有效秩受训练超参影响大,无法可靠预测下游任务表现。
Transformer权重的几何特性,尤其是解码矩阵,在语言模型可解释性研究中被广泛使用。然而其对下游性能的预测能力尚不明确。本文系统研究了模型性能与解码矩阵几何特性(特别是有效秩)的关系。通过在108个类似OLMo的语言模型上进行受控训练实验,发现最佳模型常具有高有效秩,但这一趋势并非在所有任务和训练设置下都成立。与先前研究相反,我们发现低有效秩不会导致小模型后期性能退化,而是与之共现;存在反例表明低秩模型并未出现饱和现象。此外,有效秩显著受预训练超参数(如批量大小、权重衰减)影响,进而影响模型性能。扩展到其他几何度量和最后一层表示后,我们发现这些度量整体一致,但均无法可靠预测下游性能。总体而言,现有几何度量所捕捉的模型几何特性主要反映训练选择,而非性能本身。
原文摘要 · Abstract (English)
Geometric properties of Transformer weights, particularly the unembedding matrix, have been widely useful in language model interpretability research. Yet, their utility for estimating downstream performance remains unclear. In this work, we systematically investigate the relationship between model performance and the unembedding matrix geometry, particularly its effective rank. Our experiments, involving a suite of 108 OLMo-style language models trained under controlled variation, reveal several key findings. While the best-performing models often exhibit a high effective rank, this trend is not universal across tasks and training setups. Contrary to prior work, we find that low effective rank does not cause late-stage performance degradation in small models, but instead co-occurs with it; we find adversarial cases where low-rank models do not exhibit saturation. Moreover, we show that effective rank is strongly influenced by pre-training hyperparameters, such as batch size and weight decay, which in-turn affect the model's performance. Lastly, extending our analysis to other geometric metrics and final-layer representation, we find that these metrics are largely aligned, but none can reliably predict downstream performance. Overall, our findings suggest that the model's geometry, as captured by existing metrics, primarily reflects training choices rather than performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。