Why did we open-source our inference engine? Read the post

← Catalog

opensearch-project/opensearch-neural-sparse-encoding-doc-v3-distill

Open comparison →

Primitive: /encode · Encode · DistilBERT

The model should be selected considering search relevance, model inference and retrieval efficiency(FLOPS). We benchmark models' performance on a subset of BEIR benchmark: TrecCovid,NFCorpus,NQ,HotpotQA,FiQA,ArguAna,Touche,DBPedia,SCIDOCS,FEVER,Climate FEVER,SciFact,Quora.

Sparse

Overview

Hardware: — drives latency, throughput & cost

Size67M params
Tasks /encode
Licenseapache-2.0
Languagesen
Latency61 ms
Throughput50.1K tok/s
Cost$0.0044 /1M tok

Cost is approximate — computed from list GPU prices; your actual price depends on the provider you deploy SIE with.

Embedding

Output typesSparse
Dimensionssparse: 30,522
Max sequence length512
Inputstext

Benchmarks

CQADupstackPhysicsRetrieval

scientific retrieval en

Duplicate question retrieval from StackExchange Physics

Corpus: 38,314 Queries: 1,039
Quality
map at 10 0.3390
mrr at 10 0.3891
ndcg at 10 0.3895
Performance L4 b1 c16
Corpus 33.5K tok/s
Corpus p50 58.4ms
Query 4.1K tok/s
Query p50 41.1ms
Reference →

CosQA

technology retrieval en

Code search with natural language queries

Corpus: 6,267 Queries: 500
Performance L4 b1 c16
Corpus 15.6K tok/s
Corpus p50 51.5ms
Query 1.9K tok/s
Query p50 47.9ms
Reference →

FiQA2018

finance retrieval en

Financial opinion mining and question answering

Corpus: 57,599 Queries: 648
Quality
map at 10 0.2313
mrr at 10 0.3666
ndcg at 10 0.3022
Performance L4 b1 c16
Corpus 40.5K tok/s
Corpus p50 59.2ms
Query 4.6K tok/s
Query p50 39.5ms
Reference →

LegalBenchConsumerContractsQA

legal retrieval en

Question answering on consumer contracts

Corpus: 153 Queries: 396
Performance L4 b1 c16
Corpus 97.4K tok/s
Corpus p50 76.3ms
Query 6.3K tok/s
Query p50 40.7ms
Reference →

NFCorpus

medical retrieval en

Biomedical literature search from NutritionFacts.org

Corpus: 3,593 Queries: 323
Quality
ndcg at 10 0.3294
Performance L4 b1 c16
Corpus 71.1K tok/s
Corpus p50 65.9ms
Query 1.9K tok/s
Query p50 41.1ms
Reference →

SCIDOCS

scientific retrieval en

Citation prediction, document classification, and recommendation for scientific papers

Corpus: 25,656 Queries: 1,000
Quality
map at 10 0.0856
mrr at 10 0.2585
ndcg at 10 0.1472
Performance L4 b1 c16
Corpus 47.1K tok/s
Corpus p50 59.3ms
Query 4.4K tok/s
Query p50 39.9ms
Reference →

SciFact

scientific retrieval en

Scientific claim verification using research literature

Corpus: 5,183 Queries: 300
Quality
map at 10 0.6618
mrr at 10 0.6730
ndcg at 10 0.7038
Performance L4 b1 c16
Corpus 59.3K tok/s
Corpus p50 67.2ms
Query 6.1K tok/s
Query p50 42.4ms
Reference →

StackOverflowQA

technology retrieval en

Programming question answering from Stack Overflow

Corpus: 19,931 Queries: 1,994
Performance L4 b1 c16
Corpus 53.1K tok/s
Corpus p50 62.1ms
Query 97.0K tok/s
Query p50 46.1ms
Reference →

Open source inference for agents

Open-source inference for the models behind your agents. Run it yourself, or let us run it for you.

Github 2.1K

Contact us

Tell us about your use case and we'll get back to you shortly.

Apply for an inference grant

Free capacity on our hosted cluster for selected projects. Tell us what you run and we reply by email.