English

Effect of dimensionality change on the bias of word embeddings

Computation and Language 2024-01-01 v1

Abstract

Word embedding methods (WEMs) are extensively used for representing text data. The dimensionality of these embeddings varies across various tasks and implementations. The effect of dimensionality change on the accuracy of the downstream task is a well-explored question. However, how the dimensionality change affects the bias of word embeddings needs to be investigated. Using the English Wikipedia corpus, we study this effect for two static (Word2Vec and fastText) and two context-sensitive (ElMo and BERT) WEMs. We have two observations. First, there is a significant variation in the bias of word embeddings with the dimensionality change. Second, there is no uniformity in how the dimensionality change affects the bias of word embeddings. These factors should be considered while selecting the dimensionality of word embeddings.

Keywords

Cite

@article{arxiv.2312.17292,
  title  = {Effect of dimensionality change on the bias of word embeddings},
  author = {Rohit Raj Rai and Amit Awekar},
  journal= {arXiv preprint arXiv:2312.17292},
  year   = {2024}
}

Comments

Accepted for publication in the Young Research Symposium Track of ACM CODS-COMADS 2024. 2 pages