lightonai/GTE-ModernColBERT-v1 (Encode)
This is a PyLate model trained on the ms-marco-en-bge-gemma dataset. It maps sentences & paragraphs to sequences of 128-dimensional dense vectors and can be used for semantic textual similarity using the MaxSim operator.
Benchmarks
CQADupstackPhysicsRetrieval
Duplicate question retrieval from StackExchange Physics
Corpus: 38,314 Queries: 1,039
Quality
ndcg at 10 0.3886
map at 10 0.3410
mrr at 10 0.3904
Performance L4 b1 c16
Corpus 21.7K tok/s
Corpus p50 88.1ms
Query 2.5K tok/s
Query p50 68.2ms
CosQA
Code search with natural language queries
Corpus: 6,267 Queries: 500
Quality
ndcg at 10 0.3126
map at 10 0.2347
mrr at 10 0.2366
Performance L4 b1 c16
Corpus 7.4K tok/s
Corpus p50 84.4ms
Query 462 tok/s
Query p50 76.3ms
FiQA2018
Financial opinion mining and question answering
Corpus: 57,599 Queries: 648
Quality
ndcg at 10 0.3838
map at 10 0.3133
mrr at 10 0.4648
Performance L4 b1 c16
Corpus 18.9K tok/s
Corpus p50 106.6ms
Query 2.4K tok/s
Query p50 71.9ms
LegalBenchConsumerContractsQA
Question answering on consumer contracts
Corpus: 153 Queries: 396
Quality
ndcg at 10 0.7773
map at 10 0.7300
mrr at 10 0.7321
Performance L4 b1 c16
Corpus 42.9K tok/s
Corpus p50 192.4ms
Query 3.6K tok/s
Query p50 70.2ms
NFCorpus
Biomedical literature search from NutritionFacts.org
Corpus: 3,593 Queries: 323
Quality
ndcg at 10 0.3616
map at 10 0.1390
mrr at 10 0.5824
Performance L4 b1 c16
Corpus 35.9K tok/s
Corpus p50 101.3ms
Query 1.7K tok/s
Query p50 45.7ms
NanoFiQA2018Retrieval
Smaller subset of the FiQA financial QA dataset
Quality
ndcg at 10 0.5229
map at 10 0.4304
mrr at 10 0.5544
SCIDOCS
Citation prediction, document classification, and recommendation for scientific papers
Corpus: 25,656 Queries: 1,000
Quality
ndcg at 10 0.1607
map at 10 0.0934
mrr at 10 0.2874
Performance L4 b1 c16
Corpus 30.1K tok/s
Corpus p50 96.3ms
Query 2.1K tok/s
Query p50 68.6ms
SciFact
Scientific claim verification using research literature
Corpus: 5,183 Queries: 300
Quality
ndcg at 10 0.7326
map at 10 0.6940
mrr at 10 0.7090
Performance L4 b1 c16
Corpus 31.9K tok/s
Corpus p50 118.1ms
Query 3.4K tok/s
Query p50 75.1ms
StackOverflowQA
Programming question answering from Stack Overflow
Corpus: 19,931 Queries: 1,994
Quality
ndcg at 10 0.5067
map at 10 0.4750
mrr at 10 0.4750
Performance L4 b1 c16
Corpus 26.0K tok/s
Corpus p50 127.7ms
Query 52.9K tok/s
Query p50 91.7ms