Abstract:The recent advancement in language representation modeling has broadly affected the design of dense retrieval models. In particular, many of the high-performing dense retrieval models evaluate representations of query and document using BERT, and subsequently apply a cosine-similarity based scoring to determine the relevance. BERT representations, however, are known to follow an anisotropic distribution of a narrow cone shape and such an anisotropic distribution can be undesirable for the cosine-similarity based scoring. In this work, we first show that BERT-based DR also follows an anisotropic distribution. To cope with the problem, we introduce unsupervised post-processing methods of Normalizing Flow and whitening, and develop token-wise method in addition to the sequence-wise method for applying the post-processing methods to the representations of dense retrieval models. We show that the proposed methods can effectively enhance the representations to be isotropic, then we perform experiments with ColBERT and RepBERT to show that the performance (NDCG at 10) of document re-ranking can be improved by 5.17\%$\sim$8.09\% for ColBERT and 6.88\%$\sim$22.81\% for RepBERT. To examine the potential of isotropic representation for improving the robustness of DR models, we investigate out-of-distribution tasks where the test dataset differs from the training dataset. The results show that isotropic representation can achieve a generally improved performance. For instance, when training dataset is MS-MARCO and test dataset is Robust04, isotropy post-processing can improve the baseline performance by up to 24.98\%. Furthermore, we show that an isotropic model trained with an out-of-distribution dataset can even outperform a baseline model trained with the in-distribution dataset.

Statistical Characteristics of Deep Representations: An Empirical Investigation

Subdomain contraction in deep networks for robust representation learning

VNE: An Effective Method for Improving Deep Representation by Manipulating Eigenvalue Distribution

Impact of Regularization on Calibration and Robustness: from the Representation Space Perspective

Frustratingly Easy Regularization on Representation Can Boost Deep Reinforcement Learning

Improving Generalization of Double Low-Rank Representation Using Schatten-p Norm

Adaptive Regularization of Representation Rank as an Implicit Constraint of Bellman Equation

Deep Reinforcement Learning with Decorrelation

Learning Representations from Imperfect Time Series Data via Tensor Rank Regularization

Robust low rank representation via feature and sample scaling

Sparse fixed-rank representation for robust visual analysis.

Collaborative Representation for Classification, Sparse or Non-sparse?

Causally Inspired Regularization Enables Domain General Representations

WERank: Towards Rank Degradation Prevention for Self-Supervised Learning Using Weight Regularization

Unveiling Key Aspects of Fine-Tuning in Sentence Embeddings: A Representation Rank Analysis

Rank-N-Contrast: Learning Continuous Representations for Regression

Effects of Common Regularization Techniques on Open-Set Recognition

Isotropic Representation Can Improve Dense Retrieval

Regularized Deep Linear Discriminant Analysis

Representer Point Selection for Explaining Regularized High-dimensional Models

Laplacian Regularized Low-Rank Representation and Its Applications