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

← Catalog

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

Open comparison →

Primitive: /encode · Encode · DistilBERT

The model should be selected considering search relevance, model inference and retrieval efficiency(FLOPS). We benchmark models' zero-shot 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
Latency63 ms
Throughput49.1K tok/s
Cost$0.0045 /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.3466
mrr at 10 0.3999
ndcg at 10 0.4002
Performance L4 b1 c16
Corpus 31.0K tok/s
Corpus p50 60.2ms
Query 4.1K tok/s
Query p50 41.5ms
Reference →

CosQA

technology retrieval en

Code search with natural language queries

Corpus: 6,267 Queries: 500
Performance L4 b1 c16
Corpus 17.8K tok/s
Corpus p50 46.4ms
Query 2.6K tok/s
Query p50 37.3ms
Reference →

FiQA2018

finance retrieval en

Financial opinion mining and question answering

Corpus: 57,599 Queries: 648
Quality
map at 10 0.2270
mrr at 10 0.3632
ndcg at 10 0.2949
Performance L4 b1 c16
Corpus 43.5K tok/s
Corpus p50 56.2ms
Query 4.2K tok/s
Query p50 42.8ms
Reference →

LegalBenchConsumerContractsQA

legal retrieval en

Question answering on consumer contracts

Corpus: 153 Queries: 396
Performance L4 b1 c16
Corpus 104.9K tok/s
Corpus p50 72.6ms
Query 6.9K tok/s
Query p50 37.3ms
Reference →

NFCorpus

medical retrieval en

Biomedical literature search from NutritionFacts.org

Corpus: 3,593 Queries: 323
Quality
ndcg at 10 0.3396
Performance L4 b1 c16
Corpus 56.8K tok/s
Corpus p50 70.8ms
Query 1.8K tok/s
Query p50 41.7ms
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.0804
mrr at 10 0.2469
ndcg at 10 0.1394
Performance L4 b1 c16
Corpus 48.7K tok/s
Corpus p50 58.5ms
Query 4.7K tok/s
Query p50 37.7ms
Reference →

SciFact

scientific retrieval en

Scientific claim verification using research literature

Corpus: 5,183 Queries: 300
Quality
map at 10 0.6635
mrr at 10 0.6756
ndcg at 10 0.7042
Performance L4 b1 c16
Corpus 60.2K tok/s
Corpus p50 66.4ms
Query 5.8K tok/s
Query p50 44.1ms
Reference →

StackOverflowQA

technology retrieval en

Programming question answering from Stack Overflow

Corpus: 19,931 Queries: 1,994
Performance L4 b1 c16
Corpus 49.6K tok/s
Corpus p50 70.8ms
Query 77.6K tok/s
Query p50 52.3ms
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.