a9e9a065aa95dd69.tex
1: \begin{abstract}
2: Fine-tuning pre-trained language models (PTLMs), such as BERT and its better variant RoBERTa, has been a common practice for advancing performance in natural language understanding (NLU) tasks.
3: Recent advance in representation learning shows that \textit{isotropic} (i.e., unit-variance and uncorrelated) embeddings can significantly improve performance on downstream tasks with faster convergence and better generalization.
4: The isotropy of the pre-trained embeddings in PTLMs, however, is relatively under-explored.
5: In this paper, we analyze the isotropy of the pre-trained [CLS] embeddings of PTLMs with straightforward visualization, and point out two major issues: high variance in their standard deviation, and high correlation between different dimensions.
6: We also propose a new network regularization method, isotropic batch normalization (IsoBN) to address the issues, towards learning more isotropic representations in fine-tuning by dynamically penalizing dominating principal components.
7: This simple yet effective fine-tuning method yields about 1.0 absolute increment on the average of seven NLU tasks.
8: \end{abstract}
9: