Too Much in Common: Shifting of Embeddings in Transformer Language Models and its Implications

Daniel Biś, Maksim Podkorytov, Xiuwen Liu
2021 Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies   unpublished
The success of language models based on the Transformer architecture appears to be inconsistent with observed anisotropic properties of representations learned by such models. We resolve this by showing, contrary to previous studies, that the representations do not occupy a narrow cone, but rather drift in common directions. At any training step, all of the embeddings except for the ground-truth target embedding are updated with gradient in the same direction. Compounded over the training set,
more » ... he embeddings drift and share common components, manifested in their shape in all the models we have empirically tested. Our experiments show that isotropy can be restored using a simple transformation. 1
doi:10.18653/v1/2021.naacl-main.403 fatcat:tqb6zdi3e5eb7fe5fb4itgorbi