按词频分布做白化,让低频词更突出,提升模型性能。
Zipfian Whitening
- 基于词频的加权PCA白化,纠正嵌入空间偏斜。
- 在多个任务上超越现有基线,尤其提升低频词表现。
- 适合关注词频偏差与不平衡分类的研究者。
神经网络中的词向量空间存在分布偏斜,纠正此问题可提升任务表现。现有大多数建模、修正和测量嵌入空间对称性的方法,隐含假设词频均匀分布;但现实中词频遵循高度非均匀的齐普夫定律。令人惊讶的是,仅对符合齐普夫定律的词频进行加权主成分分析(PCA)白化,就能显著提升任务性能,超越已有基线。从理论角度看,我们的方法与现有方法可明确归类:词表示服从具有均匀或齐普夫型基测度的指数族分布。采用后者能自然增强低频词的向量范数,这在信息几何视角及不平衡分类损失函数中均显现出优势。此外,该理论也解释了为何主流NLP方法如skip-gram负采样、WhiteningBERT和无头语言模型效果良好——因其词向量编码了经验词频至底层概率模型中。
原文摘要 · Abstract (English)
The word embedding space in neural models is skewed, and correcting this can improve task performance. We point out that most approaches for modeling, correcting, and measuring the symmetry of an embedding space implicitly assume that the word frequencies are uniform; in reality, word frequencies follow a highly non-uniform distribution, known as Zipf's law. Surprisingly, simply performing PCA whitening weighted by the empirical word frequency that follows Zipf's law significantly improves task performance, surpassing established baselines. From a theoretical perspective, both our approach and existing methods can be clearly categorized: word representations are distributed according to an exponential family with either uniform or Zipfian base measures. By adopting the latter approach, we can naturally emphasize informative low-frequency words in terms of their vector norm, which becomes evident from the information-geometric perspective, and in terms of the loss functions for imbalanced classification. Additionally, our theory corroborates that popular natural language processing methods, such as skip-gram negative sampling, WhiteningBERT, and headless language models, work well just because their word embeddings encode the empirical word frequency into the underlying probabilistic model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。