AnswerDotAI / AnswerDotAI/RAGatouille

Discrepancy in CPU Inference Latency: Cross-Encoder MiniLM Models vs. ColBERT

Open
#190 1 comment 0 reactions 0 assignees View on GitHub
question
Dominant language
Python
Stars
4k
Forks
276
PR merge metrics
No merged PRs in 30d

Description

Greetings :wave:

I've been benchmarking the CPU inference latency for various models and observed some significant differences. Specifically, I'm comparing the performance of the sentence_transformers 'cross-encoder/ms-marco-MiniLM-L-12-v2' with other models. The latency for the top 10 re-ranking seems to vary quite a bit, and I'm trying to understand if this is an expected behavior or if there might be an issue with my setup.
For clarity, here's a quick summary of the latencies I've recorded:
- ColBERT: 620 ms
- Cross-encoder MiniLM-L-12-v2: 309 ms
- Cross-encoder MiniLM-L-6-v2: 150 ms
I've attached a visual representation of these findings for reference:

![image (5)](https://github.com/bclavie/RAGatouille/assets/7850682/37dcabc2-0e4c-4750-a0b3-f6e8657288bf)

Could someone please shed some light on this? Is there a particular reason why the ColBERT model has over double the latency of MiniLM-L-12-v2? Any insights or suggestions for improving the inference speed of ColBert on CPU would be greatly appreciated.

Contributor guide

No contributing guide indexed for this repository

Research direction

No repository files, tests, or entry points are named. Start by reviewing the reported CPU latency comparison for ColBERT and the MiniLM cross-encoders, then determine whether the difference is expected or caused by the setup; done means documenting the explanation or a reproducible issue.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning, performance
Issue type
Bug
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.