Beyond Benchmarks: Evaluating Embedding Model Similarity for Retrieval Augmented Generation Systems

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Caspari, Laura, Dastidar, Kanishka Ghosh, Zerhoudi, Saber, Mitrovic, Jelena, Granitzer, Michael
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929417650962432
author Caspari, Laura
Dastidar, Kanishka Ghosh
Zerhoudi, Saber
Mitrovic, Jelena
Granitzer, Michael
author_facet Caspari, Laura
Dastidar, Kanishka Ghosh
Zerhoudi, Saber
Mitrovic, Jelena
Granitzer, Michael
contents The choice of embedding model is a crucial step in the design of Retrieval Augmented Generation (RAG) systems. Given the sheer volume of available options, identifying clusters of similar models streamlines this model selection process. Relying solely on benchmark performance scores only allows for a weak assessment of model similarity. Thus, in this study, we evaluate the similarity of embedding models within the context of RAG systems. Our assessment is two-fold: We use Centered Kernel Alignment to compare embeddings on a pair-wise level. Additionally, as it is especially pertinent to RAG systems, we evaluate the similarity of retrieval results between these models using Jaccard and rank similarity. We compare different families of embedding models, including proprietary ones, across five datasets from the popular Benchmark Information Retrieval (BEIR). Through our experiments we identify clusters of models corresponding to model families, but interestingly, also some inter-family clusters. Furthermore, our analysis of top-k retrieval similarity reveals high-variance at low k values. We also identify possible open-source alternatives to proprietary models, with Mistral exhibiting the highest similarity to OpenAI models.
format Preprint
id arxiv_https___arxiv_org_abs_2407_08275
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Beyond Benchmarks: Evaluating Embedding Model Similarity for Retrieval Augmented Generation Systems
Caspari, Laura
Dastidar, Kanishka Ghosh
Zerhoudi, Saber
Mitrovic, Jelena
Granitzer, Michael
Information Retrieval
The choice of embedding model is a crucial step in the design of Retrieval Augmented Generation (RAG) systems. Given the sheer volume of available options, identifying clusters of similar models streamlines this model selection process. Relying solely on benchmark performance scores only allows for a weak assessment of model similarity. Thus, in this study, we evaluate the similarity of embedding models within the context of RAG systems. Our assessment is two-fold: We use Centered Kernel Alignment to compare embeddings on a pair-wise level. Additionally, as it is especially pertinent to RAG systems, we evaluate the similarity of retrieval results between these models using Jaccard and rank similarity. We compare different families of embedding models, including proprietary ones, across five datasets from the popular Benchmark Information Retrieval (BEIR). Through our experiments we identify clusters of models corresponding to model families, but interestingly, also some inter-family clusters. Furthermore, our analysis of top-k retrieval similarity reveals high-variance at low k values. We also identify possible open-source alternatives to proprietary models, with Mistral exhibiting the highest similarity to OpenAI models.
title Beyond Benchmarks: Evaluating Embedding Model Similarity for Retrieval Augmented Generation Systems
topic Information Retrieval
url https://arxiv.org/abs/2407.08275