tomaarsen HF Staff commited on
Commit
7b6d1cb
·
verified ·
1 Parent(s): 26aea9c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +172 -42
README.md CHANGED
@@ -17,7 +17,7 @@ metrics:
17
  - mrr@10
18
  - ndcg@10
19
  model-index:
20
- - name: Production Stage 1 (68m, MSE distill from mxbai-rerank-large-v2)
21
  results:
22
  - task:
23
  type: cross-encoder-reranking
@@ -245,9 +245,11 @@ model-index:
245
  name: Ndcg@10
246
  ---
247
 
248
- # Production Stage 1 (68m, MSE distill from mxbai-rerank-large-v2)
249
 
250
- This is a [Cross Encoder](https://www.sbert.net/docs/cross_encoder/usage/usage.html) model finetuned from [jhu-clsp/ettin-encoder-68m](https://huggingface.co/jhu-clsp/ettin-encoder-68m) using the [sentence-transformers](https://www.SBERT.net) library. It computes scores for pairs of texts, which can be used for text reranking and semantic search.
 
 
251
 
252
  ## Model Details
253
 
@@ -257,7 +259,7 @@ This is a [Cross Encoder](https://www.sbert.net/docs/cross_encoder/usage/usage.h
257
  - **Maximum Sequence Length:** 7999 tokens
258
  - **Number of Output Labels:** 1 label
259
  - **Supported Modality:** Text
260
- <!-- - **Training Dataset:** Unknown -->
261
  - **Language:** en
262
  - **License:** apache-2.0
263
 
@@ -295,31 +297,27 @@ Then you can load this model and run inference.
295
  from sentence_transformers import CrossEncoder
296
 
297
  # Download from the 🤗 Hub
298
- model = CrossEncoder("cross-encoder/ettin-reranker-68m-v1")
 
 
 
 
299
  # Get scores for pairs of inputs
300
- pairs = [
301
- ['Why do we need binomial distribution?', 'Why is the binomial distribution important?'],
302
- ['I already have Windows 10, can I delete Windows.old?', 'After resetting windows 10, can I safely delete the "old windows" folder?'],
303
- ['How can guys last longer during sex?', 'How do men last longer in bed?'],
304
- ['I feel depressed all the time. What do I do?', 'I feel depressed all the time, what should I do?'],
305
- ['How is Gal Gadot as a woman and person?', 'How is Gal Gadot as a woman?'],
306
  ]
307
- scores = model.predict(pairs)
308
  print(scores)
309
- # [11.875 11. 12.3125 15.5 14.5625]
310
-
311
- # Or rank different texts based on similarity to a single text
312
- ranks = model.rank(
313
- 'Why do we need binomial distribution?',
314
- [
315
- 'Why is the binomial distribution important?',
316
- 'After resetting windows 10, can I safely delete the "old windows" folder?',
317
- 'How do men last longer in bed?',
318
- 'I feel depressed all the time, what should I do?',
319
- 'How is Gal Gadot as a woman?',
320
- ]
321
- )
322
- # [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]
323
  ```
324
 
325
  <!--
@@ -348,6 +346,131 @@ You can finetune this model on your own dataset.
348
 
349
  ## Evaluation
350
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
351
  ### Metrics
352
 
353
  #### Cross Encoder Reranking
@@ -401,6 +524,9 @@ You can finetune this model on your own dataset.
401
  | mrr@10 | 0.7499 (+0.1185) |
402
  | **ndcg@10** | **0.6895 (+0.1150)** |
403
 
 
 
 
404
  <!--
405
  ## Bias, Risks and Limitations
406
 
@@ -417,8 +543,9 @@ You can finetune this model on your own dataset.
417
 
418
  ### Training Dataset
419
 
420
- #### Unnamed Dataset
421
 
 
422
  * Size: 143,393,475 training samples
423
  * Columns: <code>query</code>, <code>document</code>, and <code>label</code>
424
  * Approximate statistics based on the first 1000 samples:
@@ -427,20 +554,11 @@ You can finetune this model on your own dataset.
427
  | type | string | string | float |
428
  | details | <ul><li>min: 26 characters</li><li>mean: 55.52 characters</li><li>max: 249 characters</li></ul> | <ul><li>min: 63 characters</li><li>mean: 659.91 characters</li><li>max: 3975 characters</li></ul> | <ul><li>min: -2.94</li><li>mean: 8.51</li><li>max: 13.88</li></ul> |
429
  * Samples:
430
- | query | document | label |
431
- |:----------------------------------------------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------|
432
- | <code>Help me with my Reborn performance</code> | <code>I was reading the comment section for Dotacinema's world of dota video, and a bunch of people were complaining how there were a lot of bugs and some talked about PERFORMANCE ISSUES. But there were also people saying that reborn has actually IMPROVED their gameplay?
433
-
434
-
435
- I am one of those people who is running into performance issues and would desperately like to know how some are getting BETTER performance while others like me are getting worse. I'm not complaining about bugs, I'm complaing about framerate, I use to get 60 fps solid in source 1 but I now have 40 or at worst 30 fps in source 2.
436
- I have an i3 processor/gtx560ti/16gb RAM
437
-
438
- i dont think it's a potato pc, so I dont know what's happening, I cleaned my computer recently so dust isnt affecting anything in anyway.
439
- So if you gained or had IMPROVED performance in source 2 please list the settings you are enabling, so I can see where I am at fault. (v sync is off btw)
440
-
441
- TLDR: Have bad performance now from source 2, if you have good p...</code> | <code>9.5</code> |
442
- | <code>Really wanna try out the game and expansion, ~$60 is hefty. Likelihood of sales?</code> | <code>As per title, steam sells the game and its expansions for $60 total. Heavy price to drop. Are there sales on any other website? This game looks fantastic to immerse in otherwise and I'm pleased that this subreddit has at least some attention to help out new folks!</code> | <code>9.25</code> |
443
- | <code>Your Avatar. [MGSV Spoilers]</code> | <code>Was anyone else suprised he actually replaces the snake model in some cutscenes. I've only tried the first Quiet cutscenes, i was just amazed I haven't seen anybody else say this yet.<br>Sorry if repost.</code> | <code>5.25</code> |
444
  * Loss: [<code>MSELoss</code>](https://sbert.net/docs/package_reference/cross_encoder/losses.html#mseloss) with these parameters:
445
  ```json
446
  {
@@ -450,8 +568,9 @@ You can finetune this model on your own dataset.
450
 
451
  ### Evaluation Dataset
452
 
453
- #### Unnamed Dataset
454
 
 
455
  * Size: 5,000 evaluation samples
456
  * Columns: <code>query</code>, <code>document</code>, and <code>label</code>
457
  * Approximate statistics based on the first 1000 samples:
@@ -667,6 +786,17 @@ You can finetune this model on your own dataset.
667
 
668
  ### BibTeX
669
 
 
 
 
 
 
 
 
 
 
 
 
670
  #### Sentence Transformers
671
  ```bibtex
672
  @inproceedings{reimers-2019-sentence-bert,
 
17
  - mrr@10
18
  - ndcg@10
19
  model-index:
20
+ - name: ettin-reranker-68m-v1
21
  results:
22
  - task:
23
  type: cross-encoder-reranking
 
245
  name: Ndcg@10
246
  ---
247
 
248
+ # ettin-reranker-68m-v1
249
 
250
+ This is a [Cross Encoder](https://www.sbert.net/docs/cross_encoder/usage/usage.html) model finetuned from [jhu-clsp/ettin-encoder-68m](https://huggingface.co/jhu-clsp/ettin-encoder-68m) on the [cross-encoder/ettin-reranker-v1-data](https://huggingface.co/datasets/cross-encoder/ettin-reranker-v1-data) dataset using the [sentence-transformers](https://www.SBERT.net) library. It computes scores for pairs of texts, which can be used for text reranking and semantic search.
251
+
252
+ See the [release blogpost](https://huggingface.co/blog/ettin-reranker) for details on the training recipe, evaluation results, and speed benchmarks against other public rerankers. The [Evaluation](#evaluation) section below also has the headline numbers.
253
 
254
  ## Model Details
255
 
 
259
  - **Maximum Sequence Length:** 7999 tokens
260
  - **Number of Output Labels:** 1 label
261
  - **Supported Modality:** Text
262
+ - **Training Dataset:** [cross-encoder/ettin-reranker-v1-data](https://huggingface.co/datasets/cross-encoder/ettin-reranker-v1-data)
263
  - **Language:** en
264
  - **License:** apache-2.0
265
 
 
297
  from sentence_transformers import CrossEncoder
298
 
299
  # Download from the 🤗 Hub
300
+ model = CrossEncoder(
301
+ "cross-encoder/ettin-reranker-68m-v1",
302
+ model_kwargs={"dtype": "bfloat16", "attn_implementation": "flash_attention_2"}, # Optional: pip install kernels
303
+ )
304
+
305
  # Get scores for pairs of inputs
306
+ query = "Which planet is known as the Red Planet?"
307
+ passages = [
308
+ "Venus is often called Earth's twin because of its similar size and proximity.",
309
+ "Mars, known for its reddish appearance, is often referred to as the Red Planet.",
310
+ "Jupiter, the largest planet in our solar system, has a prominent red spot.",
311
+ "Saturn, famous for its rings, is sometimes mistaken for the Red Planet.",
312
  ]
313
+ scores = model.predict([(query, passage) for passage in passages])
314
  print(scores)
315
+ # [ 6.375 11.5 7.625 10.4375]
316
+
317
+ # Or rank passages by relevance to a single query
318
+ ranked = model.rank(query, passages)
319
+ print(ranked)
320
+ # [{'corpus_id': 1, 'score': np.float32(11.5)}, ...]
 
 
 
 
 
 
 
 
321
  ```
322
 
323
  <!--
 
346
 
347
  ## Evaluation
348
 
349
+ ### MTEB(eng, v2) Retrieval
350
+
351
+ Each model in the ettin-reranker-v1 family was evaluated on the full [`MTEB(eng, v2)` Retrieval benchmark](https://github.com/embeddings-benchmark/mteb) (10 tasks, top-100 reranked) using MTEB's [two-stage reranking flow](https://embeddings-benchmark.github.io/mteb/get_started/advanced_usage/two_stage_reranking/), pairing each reranker with six embedding models that span the speed/quality spectrum.
352
+ The dashed retriever-only line in each chart below is the headline number to beat. Anything below it means the reranker actively hurts the pipeline on average:
353
+
354
+ | | |
355
+ |-|-|
356
+ | ![MTEB(eng, v2) Retrieval with static-retrieval-mrl-en-v1 + reranker](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/ettin-reranker/mteb_ndcg10_static-retrieval-mrl-en-v1.png) | ![MTEB(eng, v2) Retrieval with all-MiniLM-L6-v2 + reranker](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/ettin-reranker/mteb_ndcg10_all-MiniLM-L6-v2.png) |
357
+ | ![MTEB(eng, v2) Retrieval with bge-small-en-v1.5 + reranker](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/ettin-reranker/mteb_ndcg10_bge-small-en-v1.5.png) | ![MTEB(eng, v2) Retrieval with nomic-embed-text-v1.5 + reranker](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/ettin-reranker/mteb_ndcg10_nomic-embed-text-v1.5.png) |
358
+ | ![MTEB(eng, v2) Retrieval with embeddinggemma-300m + reranker](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/ettin-reranker/mteb_ndcg10_embeddinggemma-300m.png) | ![MTEB(eng, v2) Retrieval with jina-embeddings-v5-text-small-retrieval + reranker](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/blog/ettin-reranker/mteb_ndcg10_jina-embeddings-v5-text-small-retrieval.png) |
359
+
360
+ <details><summary>Full table of results (click to expand)</summary>
361
+
362
+ Mean NDCG@10 over the 6 embedder pairings, sorted by MTEB. The released ettin-reranker-v1 family is in **bold**, and the teacher [`mixedbread-ai/mxbai-rerank-large-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v2) is <u>underlined</u>.
363
+
364
+ | Reranker | Params | MTEB(eng, v2) Retrieval NDCG@10 |
365
+ | --- | ---: | ---: |
366
+ | [`Qwen/Qwen3-Reranker-4B`](https://huggingface.co/Qwen/Qwen3-Reranker-4B)<sup>†</sup> | 4.02B | 0.6367 |
367
+ | <u>[`mixedbread-ai/mxbai-rerank-large-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v2)</u> | <u>1.54B</u> | <u>0.6115</u> |
368
+ | **[`cross-encoder/ettin-reranker-1b-v1`](https://huggingface.co/cross-encoder/ettin-reranker-1b-v1)** | **1.00B** | **0.6114** |
369
+ | **[`cross-encoder/ettin-reranker-400m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-400m-v1)** | **401M** | **0.6091** |
370
+ | **[`cross-encoder/ettin-reranker-150m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-150m-v1)** | **151M** | **0.5994** |
371
+ | [`Qwen/Qwen3-Reranker-0.6B`](https://huggingface.co/Qwen/Qwen3-Reranker-0.6B) | 596M | 0.5940 |
372
+ | [`mixedbread-ai/mxbai-rerank-base-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v2) | 494M | 0.5920 |
373
+ | **[`cross-encoder/ettin-reranker-68m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-68m-v1)** | **68.6M** | **0.5915** |
374
+ | [`jinaai/jina-reranker-m0`](https://huggingface.co/jinaai/jina-reranker-m0) | 2.44B | 0.5856 |
375
+ | [`Alibaba-NLP/gte-reranker-modernbert-base`](https://huggingface.co/Alibaba-NLP/gte-reranker-modernbert-base) | 150M | 0.5843 |
376
+ | **[`cross-encoder/ettin-reranker-32m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-32m-v1)** | **32.8M** | **0.5779** |
377
+ | [`ibm-granite/granite-embedding-reranker-english-r2`](https://huggingface.co/ibm-granite/granite-embedding-reranker-english-r2) | 150M | 0.5656 |
378
+ | **[`cross-encoder/ettin-reranker-17m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-17m-v1)** | **17.6M** | **0.5576** |
379
+ | [`BAAI/bge-reranker-v2-m3`](https://huggingface.co/BAAI/bge-reranker-v2-m3) | 568M | 0.5526 |
380
+ | [`zeroentropy/zerank-2-reranker`](https://huggingface.co/zeroentropy/zerank-2-reranker)<sup>†</sup> | 4.02B | 0.5300 |
381
+ | [`BAAI/bge-reranker-large`](https://huggingface.co/BAAI/bge-reranker-large) | 560M | 0.5098 |
382
+ | [`cross-encoder/ms-marco-MiniLM-L6-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L6-v2) | 22.7M | 0.5082 |
383
+ | [`cross-encoder/ms-marco-MiniLM-L12-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L12-v2) | 33.4M | 0.5066 |
384
+ | [`mixedbread-ai/mxbai-rerank-large-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v1) | 435M | 0.5063 |
385
+ | [`cross-encoder/ms-marco-MiniLM-L4-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L4-v2) | 19.2M | 0.4979 |
386
+ | [`mixedbread-ai/mxbai-rerank-xsmall-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-xsmall-v1) | 70.8M | 0.4968 |
387
+ | [`BAAI/bge-reranker-base`](https://huggingface.co/BAAI/bge-reranker-base) | 278M | 0.4890 |
388
+ | [`mixedbread-ai/mxbai-rerank-base-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v1) | 184M | 0.4865 |
389
+
390
+ <sup>†</sup> Capped to `max_seq_length=8192` (the 4B Qwen3-based rerankers don't fit on a single H100 80GB at native context). Native-context evaluation is likely higher.
391
+
392
+ </details>
393
+
394
+ See the [release blogpost](https://huggingface.co/blog/ettin-reranker) for the full analysis and per-model commentary.
395
+
396
+ ### Speed
397
+
398
+ All six released models were benchmarked against thirteen public rerankers on three hardware tiers, using [`sentence-transformers/natural-questions`](https://huggingface.co/datasets/sentence-transformers/natural-questions) at `max_length=512` with each model's best supported attention implementation. The full sweep over `fp32+SDPA`, `bf16+SDPA`, padded `bf16+FA2`, and unpadded `bf16+FA2` (showing why the ettin-reranker-v1 family is faster than other ModernBERT-based rerankers) is in the [release blogpost](https://huggingface.co/blog/ettin-reranker#speed). This table shows the throughput in pairs per second on a NVIDIA H100 80GB, all in `bfloat16`:
399
+
400
+ | Model | Params | Attn | pairs / second |
401
+ |---|---:|---|---|
402
+ | **[`cross-encoder/ettin-reranker-17m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-17m-v1)** | **17M** | FA2 | **7517** |
403
+ | **[`cross-encoder/ettin-reranker-32m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-32m-v1)** | **32M** | FA2 | **6602** |
404
+ | **[`cross-encoder/ettin-reranker-68m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-68m-v1)** | **68M** | FA2 | **4913** |
405
+ | [`cross-encoder/ms-marco-MiniLM-L4-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L4-v2) | 19M | FA2 | 4029 |
406
+ | [`cross-encoder/ms-marco-MiniLM-L6-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L6-v2) | 22M | FA2 | 3817 |
407
+ | [`cross-encoder/ms-marco-MiniLM-L12-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L12-v2) | 33M | FA2 | 3311 |
408
+ | **[`cross-encoder/ettin-reranker-150m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-150m-v1)** | **150M** | FA2 | **3237** |
409
+ | [`BAAI/bge-reranker-base`](https://huggingface.co/BAAI/bge-reranker-base) | 278M | FA2 | 2858 |
410
+ | [`mixedbread-ai/mxbai-rerank-xsmall-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-xsmall-v1) | 70M | eager | 2636 |
411
+ | [`mixedbread-ai/mxbai-rerank-base-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v1) | 184M | eager | 1953 |
412
+ | **[`cross-encoder/ettin-reranker-400m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-400m-v1)** | **400M** | FA2 | **1738** |
413
+ | [`BAAI/bge-reranker-large`](https://huggingface.co/BAAI/bge-reranker-large) | 560M | FA2 | 1659 |
414
+ | [`BAAI/bge-reranker-v2-m3`](https://huggingface.co/BAAI/bge-reranker-v2-m3) | 568M | FA2 | 1569 |
415
+ | [`Alibaba-NLP/gte-reranker-modernbert-base`](https://huggingface.co/Alibaba-NLP/gte-reranker-modernbert-base) | 150M | FA2 | 1418 |
416
+ | [`ibm-granite/granite-embedding-reranker-english-r2`](https://huggingface.co/ibm-granite/granite-embedding-reranker-english-r2) | 150M | FA2 | 1404 |
417
+ | **[`cross-encoder/ettin-reranker-1b-v1`](https://huggingface.co/cross-encoder/ettin-reranker-1b-v1)** | **1B** | FA2 | **928** |
418
+ | [`mixedbread-ai/mxbai-rerank-large-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v1) | 435M | eager | 867 |
419
+ | [`mixedbread-ai/mxbai-rerank-base-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v2) | 494M | FA2 | 809 |
420
+ | <u>[`mixedbread-ai/mxbai-rerank-large-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v2)</u> | <u>1.5B</u> | FA2 | <u>387</u> |
421
+
422
+ <details><summary>Same benchmark on a consumer GPU (RTX 3090, 24 GB)</summary>
423
+
424
+ | Model | Params | Best attn | pairs / second |
425
+ |---|---:|---|---:|
426
+ | **[`cross-encoder/ettin-reranker-17m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-17m-v1)** | **17M** | FA2 | **9008** |
427
+ | [`cross-encoder/ms-marco-MiniLM-L4-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L4-v2) | 19M | FA2 | 5071 |
428
+ | **[`cross-encoder/ettin-reranker-32m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-32m-v1)** | **32M** | FA2 | **4497** |
429
+ | [`cross-encoder/ms-marco-MiniLM-L6-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L6-v2) | 22M | FA2 | 4234 |
430
+ | [`cross-encoder/ms-marco-MiniLM-L12-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L12-v2) | 33M | FA2 | 2847 |
431
+ | **[`cross-encoder/ettin-reranker-68m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-68m-v1)** | **68M** | FA2 | **1916** |
432
+ | [`mixedbread-ai/mxbai-rerank-xsmall-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-xsmall-v1) | 70M | eager | 1677 |
433
+ | [`BAAI/bge-reranker-base`](https://huggingface.co/BAAI/bge-reranker-base) | 278M | FA2 | 1329 |
434
+ | **[`cross-encoder/ettin-reranker-150m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-150m-v1)** | **150M** | FA2 | **982** |
435
+ | [`mixedbread-ai/mxbai-rerank-base-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v1) | 184M | eager | 772 |
436
+ | [`ibm-granite/granite-embedding-reranker-english-r2`](https://huggingface.co/ibm-granite/granite-embedding-reranker-english-r2) | 150M | FA2 | 598 |
437
+ | [`Alibaba-NLP/gte-reranker-modernbert-base`](https://huggingface.co/Alibaba-NLP/gte-reranker-modernbert-base) | 150M | FA2 | 586 |
438
+ | [`BAAI/bge-reranker-large`](https://huggingface.co/BAAI/bge-reranker-large) | 560M | FA2 | 448 |
439
+ | [`BAAI/bge-reranker-v2-m3`](https://huggingface.co/BAAI/bge-reranker-v2-m3) | 568M | FA2 | 436 |
440
+ | **[`cross-encoder/ettin-reranker-400m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-400m-v1)** | **400M** | FA2 | **429** |
441
+ | [`mixedbread-ai/mxbai-rerank-large-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v1) | 435M | eager | 266 |
442
+ | [`mixedbread-ai/mxbai-rerank-base-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v2) | 494M | FA2 | 221 |
443
+ | **[`cross-encoder/ettin-reranker-1b-v1`](https://huggingface.co/cross-encoder/ettin-reranker-1b-v1)** | **1B** | FA2 | **189** |
444
+ | <u>[`mixedbread-ai/mxbai-rerank-large-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v2)</u> | <u>1.5B</u> | FA2 | <u>69</u> |
445
+
446
+ </details>
447
+
448
+ <details><summary>Same benchmark on CPU (Intel Core i7-13700K)</summary>
449
+
450
+ | Model | Params | Best attn | pairs / second |
451
+ |---|---:|---|---:|
452
+ | **[`cross-encoder/ettin-reranker-17m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-17m-v1)** | **17M** | SDPA | **76.1** |
453
+ | [`cross-encoder/ms-marco-MiniLM-L4-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L4-v2) | 19M | SDPA | 53.0 |
454
+ | [`cross-encoder/ms-marco-MiniLM-L6-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L6-v2) | 22M | SDPA | 29.4 |
455
+ | **[`cross-encoder/ettin-reranker-32m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-32m-v1)** | **32M** | SDPA | **28.5** |
456
+ | [`cross-encoder/ms-marco-MiniLM-L12-v2`](https://huggingface.co/cross-encoder/ms-marco-MiniLM-L12-v2) | 33M | SDPA | 17.3 |
457
+ | **[`cross-encoder/ettin-reranker-68m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-68m-v1)** | **68M** | SDPA | **8.5** |
458
+ | [`mixedbread-ai/mxbai-rerank-xsmall-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-xsmall-v1) | 70M | eager | 6.0 |
459
+ | [`BAAI/bge-reranker-base`](https://huggingface.co/BAAI/bge-reranker-base) | 278M | SDPA | 4.7 |
460
+ | [`Alibaba-NLP/gte-reranker-modernbert-base`](https://huggingface.co/Alibaba-NLP/gte-reranker-modernbert-base) | 150M | SDPA | 3.7 |
461
+ | **[`cross-encoder/ettin-reranker-150m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-150m-v1)** | **150M** | SDPA | **3.6** |
462
+ | [`ibm-granite/granite-embedding-reranker-english-r2`](https://huggingface.co/ibm-granite/granite-embedding-reranker-english-r2) | 150M | SDPA | 3.6 |
463
+ | [`mixedbread-ai/mxbai-rerank-base-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v1) | 184M | eager | 2.4 |
464
+ | **[`cross-encoder/ettin-reranker-400m-v1`](https://huggingface.co/cross-encoder/ettin-reranker-400m-v1)** | **400M** | SDPA | **1.3** |
465
+ | [`BAAI/bge-reranker-large`](https://huggingface.co/BAAI/bge-reranker-large) | 560M | SDPA | 1.2 |
466
+ | [`BAAI/bge-reranker-v2-m3`](https://huggingface.co/BAAI/bge-reranker-v2-m3) | 568M | SDPA | 1.2 |
467
+ | [`mixedbread-ai/mxbai-rerank-base-v2`](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v2) | 494M | SDPA | 0.8 |
468
+ | [`mixedbread-ai/mxbai-rerank-large-v1`](https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v1) | 435M | eager | 0.8 |
469
+ | **[`cross-encoder/ettin-reranker-1b-v1`](https://huggingface.co/cross-encoder/ettin-reranker-1b-v1)** | **1B** | SDPA | **0.5** |
470
+
471
+ </details>
472
+
473
+
474
  ### Metrics
475
 
476
  #### Cross Encoder Reranking
 
524
  | mrr@10 | 0.7499 (+0.1185) |
525
  | **ndcg@10** | **0.6895 (+0.1150)** |
526
 
527
+ > [!NOTE]
528
+ > The [release blogpost](https://huggingface.co/blog/ettin-reranker) quotes a slightly higher NanoBEIR mean NDCG@10 of `0.6915` for this model, computed in `fp32` rather than the `bfloat16` used by the training-time evaluation above. Both numbers are valid.
529
+
530
  <!--
531
  ## Bias, Risks and Limitations
532
 
 
543
 
544
  ### Training Dataset
545
 
546
+ #### ettin-reranker-v1-data
547
 
548
+ * Dataset: [cross-encoder/ettin-reranker-v1-data](https://huggingface.co/datasets/cross-encoder/ettin-reranker-v1-data)
549
  * Size: 143,393,475 training samples
550
  * Columns: <code>query</code>, <code>document</code>, and <code>label</code>
551
  * Approximate statistics based on the first 1000 samples:
 
554
  | type | string | string | float |
555
  | details | <ul><li>min: 26 characters</li><li>mean: 55.52 characters</li><li>max: 249 characters</li></ul> | <ul><li>min: 63 characters</li><li>mean: 659.91 characters</li><li>max: 3975 characters</li></ul> | <ul><li>min: -2.94</li><li>mean: 8.51</li><li>max: 13.88</li></ul> |
556
  * Samples:
557
+ | query | document | label |
558
+ |:----------------------------------------------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------|
559
+ | <code>Help me with my Reborn performance</code> | <code>I was reading the comment section for Dotacinema's world of dota video, and a bunch of people were complaining how there were a lot of bugs and some talked about PERFORMANCE ISSUES. But there were also people saying that reborn has actually IMPROVED their gameplay?<br><br><br>I am one of those people who is running into performance issues and would desperately like to know how some are getting BETTER performance while others like me are getting worse. I'm not complaining about bugs, I'm complaing about framerate, I use to get 60 fps solid in source 1 but I now have 40 or at worst 30 fps in source 2.<br>I have an i3 processor/gtx560ti/16gb RAM<br><br>i dont think it's a potato pc, so I dont know what's happening, I cleaned my computer recently so dust isnt affecting anything in anyway.<br>So if you gained or had IMPROVED performance in source 2 please list the settings you are enabling, so I can see where I am at fault. (v sync is off btw)<br><br>TLDR: Have bad performance now from source 2, if you have good p...</code> | <code>9.5</code> |
560
+ | <code>Really wanna try out the game and expansion, ~$60 is hefty. Likelihood of sales?</code> | <code>As per title, steam sells the game and its expansions for $60 total. Heavy price to drop. Are there sales on any other website? This game looks fantastic to immerse in otherwise and I'm pleased that this subreddit has at least some attention to help out new folks!</code> | <code>9.25</code> |
561
+ | <code>Your Avatar. [MGSV Spoilers]</code> | <code>Was anyone else suprised he actually replaces the snake model in some cutscenes. I've only tried the first Quiet cutscenes, i was just amazed I haven't seen anybody else say this yet.<br>Sorry if repost.</code> | <code>5.25</code> |
 
 
 
 
 
 
 
 
 
562
  * Loss: [<code>MSELoss</code>](https://sbert.net/docs/package_reference/cross_encoder/losses.html#mseloss) with these parameters:
563
  ```json
564
  {
 
568
 
569
  ### Evaluation Dataset
570
 
571
+ #### ettin-reranker-v1-data
572
 
573
+ * Dataset: [cross-encoder/ettin-reranker-v1-data](https://huggingface.co/datasets/cross-encoder/ettin-reranker-v1-data)
574
  * Size: 5,000 evaluation samples
575
  * Columns: <code>query</code>, <code>document</code>, and <code>label</code>
576
  * Approximate statistics based on the first 1000 samples:
 
786
 
787
  ### BibTeX
788
 
789
+ #### Ettin Reranker Blogpost
790
+ ```bibtex
791
+ @misc{aarsen2026ettin-reranker,
792
+ title = "Introducing the Ettin Reranker Family",
793
+ author = "Aarsen, Tom",
794
+ year = "2026",
795
+ publisher = "Hugging Face",
796
+ url = "https://huggingface.co/blog/ettin-reranker",
797
+ }
798
+ ```
799
+
800
  #### Sentence Transformers
801
  ```bibtex
802
  @inproceedings{reimers-2019-sentence-bert,