Towards Understanding Domain Adapted Sentence Embeddings for Document Retrieval

• Published in: arXiv.org 2024-12
• Main Authors: Roychowdhury, Sujoy, Soman, Sumit, Ranjani, H G, Chhabra, Vansh, Gunda, Neeraj, Gautam, Shashank, Bandyopadhyay, Subhadip, Bala, Sai Krishna
• Format: Article
• Language: English
• Subjects: Adaptation; Confidence intervals; Isotropy; Retrieval; Similarity; Telecommunications; Thresholds
• ISSN: 2331-8422

A plethora of sentence embedding models makes it challenging to choose one, especially for technical domains rich with specialized vocabulary. In this work, we domain adapt embeddings using telecom, health and science datasets for question answering. We evaluate embeddings obtained from publicly available models and their domain-adapted variants, on both point retrieval accuracies, as well as their (95\%) confidence intervals. We establish a systematic method to obtain thresholds for similarity scores for different embeddings. As expected, we observe that fine-tuning improves mean bootstrapped accuracies. We also observe that it results in tighter confidence intervals, which further improve when pre-training is preceded by fine-tuning. We introduce metrics which measure the distributional overlaps of top-\(K\), correct and random document similarities with the question. Further, we show that these metrics are correlated with retrieval accuracy and similarity thresholds. Recent literature shows conflicting effects of isotropy on retrieval accuracies. Our experiments establish that the isotropy of embeddings (as measured by two independent state-of-the-art isotropy metric definitions) is poorly correlated with retrieval performance. We show that embeddings for domain-specific sentences have little overlap with those for domain-agnostic ones, and fine-tuning moves them further apart. Based on our results, we provide recommendations for use of our methodology and metrics by researchers and practitioners.