NohTow commited on
Commit
0edbd55
·
verified ·
1 Parent(s): 36e5b20

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +120 -7
README.md CHANGED
@@ -4,25 +4,113 @@ tags:
4
  - sentence-similarity
5
  - feature-extraction
6
  - dense
 
 
 
7
  pipeline_tag: sentence-similarity
8
  library_name: sentence-transformers
9
  ---
 
 
10
 
11
- # SentenceTransformer
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
- This is a [sentence-transformers](https://www.SBERT.net) model trained. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
14
 
15
  ## Model Details
16
 
17
  ### Model Description
18
  - **Model Type:** Sentence Transformer
19
- <!-- - **Base model:** [Unknown](https://huggingface.co/unknown) -->
20
  - **Maximum Sequence Length:** 512 tokens
21
  - **Output Dimensionality:** 768 dimensions
22
  - **Similarity Function:** Cosine Similarity
23
- <!-- - **Training Dataset:** Unknown -->
24
- <!-- - **Language:** Unknown -->
25
- <!-- - **License:** Unknown -->
26
 
27
  ### Model Sources
28
 
@@ -54,7 +142,7 @@ Then you can load this model and run inference.
54
  from sentence_transformers import SentenceTransformer
55
 
56
  # Download from the 🤗 Hub
57
- model = SentenceTransformer("lightonai/LateOn-unsupervised")
58
  # Run inference
59
  queries = [
60
  "Which planet is known as the Red Planet?",
@@ -126,6 +214,8 @@ You can finetune this model on your own dataset.
126
 
127
  ### BibTeX
128
 
 
 
129
  ```bibtex
130
  @misc{sourty2026denseonlateon,
131
  title={DenseOn with the LateOn: Open State-of-the-Art Single and Multi-Vector Models},
@@ -135,6 +225,8 @@ You can finetune this model on your own dataset.
135
  }
136
  ```
137
 
 
 
138
  ```bibtex
139
  @inproceedings{DBLP:conf/cikm/ChaffinS25,
140
  author = {Antoine Chaffin and
@@ -161,6 +253,27 @@ You can finetune this model on your own dataset.
161
  }
162
  ```
163
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
164
  <!--
165
  ## Glossary
166
 
 
4
  - sentence-similarity
5
  - feature-extraction
6
  - dense
7
+ license: apache-2.0
8
+ language:
9
+ - en
10
  pipeline_tag: sentence-similarity
11
  library_name: sentence-transformers
12
  ---
13
+ <div align="center">
14
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/609bbe2f4932693ca2009d6a/kbQOAarw0eaApow3M9HIl.png" alt="LightOn" width="512">
15
 
16
+ [![Website](https://img.shields.io/badge/LightOn-Website-blue?logo=google-chrome)](https://lighton.ai)
17
+ [![LinkedIn](https://img.shields.io/badge/LightOn-LinkedIn-0A66C2?logo=linkedin)](https://www.linkedin.com/company/lighton/)
18
+ [![X](https://img.shields.io/badge/@LightOnIO-X-black?logo=x)](https://x.com/LightOnIO)
19
+
20
+ 📚 [Collection](https://huggingface.co/collections/lightonai/denseon-and-lateon) | 📝 [Blog](https://huggingface.co/blog/lightonai/denseon-lateon)
21
+
22
+ </div>
23
+
24
+ <h1 align="center">DenseOn-unsupervised</h1>
25
+
26
+ <h3 align="center">State-of-the-Art Dense Retrieval Model by LightOn</h3>
27
+
28
+ <p align="center">
29
+ <a href="https://huggingface.co/lightonai/DenseOn">DenseOn</a> |
30
+ <a href="https://huggingface.co/lightonai/LateOn">LateOn</a> |
31
+ <a href="https://github.com/lightonai/pylate">PyLate</a> |
32
+ <a href="https://github.com/lightonai/fast-plaid">FastPLAID</a>
33
+ </p>
34
+
35
+ > 🎯 **TL;DR**: The intermediate dense checkpoint produced by **Stage 1 only** of the DenseOn pipeline: large-scale unsupervised contrastive pre-training on filtered query-document pairs. Released as a **strong starting point for your own supervised fine-tuning, knowledge distillation, or downstream adaptation**.
36
+
37
+ ## About the LateOn / DenseOn Family
38
+
39
+ State-of-the-art retrieval is increasingly dominated by closed models, either hidden behind APIs or trained on undisclosed data. This blocks reproducibility, prevents study of possible data leakage, and gatekeeps progress to a handful of private labs. We thus decided to gather and curate a large amount of data and explore various mixtures. We release all the data used in our explorations:
40
+ - [Gathered pre-training data](https://huggingface.co/datasets/lightonai/embeddings-pre-training), 1.4B query-documents pairs alongside annotations used for non-destructive filtering (structural filtering, deduplication, cross-encoder pair relevancy)
41
+ - [Best pre-training mixture](https://huggingface.co/datasets/lightonai/embeddings-pre-training-curated) found with already applied filters
42
+ - [Fine-tuning datasets](https://huggingface.co/datasets/lightonai/embeddings-fine-tuning) with query, positive and 2048 mined documents alongside their scores for 1.88M samples.
43
+
44
+ Based on our findings, we trained LateOn (multi-vector/ColBERT) and DenseOn (single vector/dense) models on a proprietary Apache 2.0-compatible training dataset and release those models as well.
45
+ Both are built on the [ModernBERT](https://huggingface.co/answerdotai/ModernBERT-base) backbone at 149M parameters, a size we believe sits at the sweet spot: large enough to handle real-world queries and documents, small enough to serve at high throughput in latency-sensitive production systems.
46
+ For more information, please read our [blogpost](https://huggingface.co/blog/lightonai/denseon-lateon).
47
+
48
+ ## DenseOn-unsupervised
49
+
50
+ DenseOn-unsupervised is the output of the **first stage** of the [DenseOn](https://huggingface.co/lightonai/DenseOn) training pipeline. It has been pre-trained on a large, filtered corpus of query-document pairs using in-batch contrastive learning, but has **not** yet been fine-tuned with mined hard negatives.
51
+
52
+ For most production use cases, you should use the **fully-trained [DenseOn](https://huggingface.co/lightonai/DenseOn)** instead. This unsupervised checkpoint is intended for:
53
+
54
+ - Researchers studying what each pipeline stage contributes
55
+ - Practitioners who want to fine-tune on their own domain-specific data
56
+ - Distillation experiments where you want to start from a strong but un-aligned base
57
+ - Anyone running their own ablations on hard-negative mining strategies
58
+
59
+
60
+ ## Results
61
+
62
+ ### BEIR (14 datasets, NDCG@10)
63
+
64
+ | Model | Average | Size | Emb dim | ArguAna | CQADupstackRetrieval | ClimateFEVER | DBPedia | FEVER | FiQA2018 | HotpotQA | MSMARCO | NFCorpus | NQ | QuoraRetrieval | SCIDOCS | SciFact | TRECCOVID | Touche2020 |
65
+ |---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
66
+ | [modernbert-embed-base](https://huggingface.co/nomic-ai/modernbert-embed-base) | 52.89 | 149 | 768 | 48.96 | 42.08 | 35.67 | 41.50 | 87.35 | 40.59 | 67.11 | 41.47 | 33.40 | 62.15 | 88.85 | 18.59 | 69.63 | 84.15 | 31.91 |
67
+ | [bge-large-en-v1.5](https://huggingface.co/BAAI/bge-large-en-v1.5) | 54.34 | 335 | 1024 | 64.52 | 42.23 | 36.57 | 44.11 | 87.18 | 45.02 | 74.10 | 42.49 | 38.06 | 55.03 | 89.07 | 22.63 | 74.64 | 74.70 | 24.81 |
68
+ | [gte-modernbert-base](https://huggingface.co/Alibaba-NLP/gte-modernbert-base) | 55.19 | 149 | 768 | **74.56** | 42.64 | **45.90** | 41.39 | **93.98** | 49.54 | 70.39 | 39.93 | 34.32 | 56.10 | 88.57 | 20.44 | **76.41** | 75.75 | 17.97 |
69
+ | [snowflake-arctic-embed-l-v2.0](https://huggingface.co/Snowflake/snowflake-arctic-embed-l-v2.0) | 55.22 | 568 | 1024 | 59.11 | 45.88 | 41.82 | 43.40 | 91.54 | 45.35 | 68.15 | **44.86** | 35.08 | **63.67** | 88.75 | 20.28 | 70.90 | 83.63 | 25.89 |
70
+ | [jina-embeddings-v5-text-nano](https://huggingface.co/jinaai/jina-embeddings-v5-text-nano) | 56.06 | 239 | 768 | 65.70 | *44.66* | 39.60 | **45.26** | 89.51 | 47.85 | 69.07 | 41.64 | 38.69 | 63.38 | 88.87 | 22.60 | 75.78 | 77.60 | 30.70 |
71
+ | [Qwen3-Embedding-0.6B](https://huggingface.co/Qwen/Qwen3-Embedding-0.6B) | 55.52 | 600 | 1024 | 70.97 | 46.03 | 42.11 | 39.48 | 88.15 | 46.61 | 65.74 | 37.99 | 36.71 | 53.46 | 87.78 | **24.41** | 69.72 | **90.52** | **33.18** |
72
+ | [pplx-embed-v1-0.6b](https://huggingface.co/perplexity-ai/pplx-embed-v1-0.6b) | **56.70** | 600 | 1024 | 60.45 | 45.96 | 39.82 | 44.30 | 90.66 | 52.05 | 74.41 | 43.86 | 35.80 | 62.04 | 88.96 | 22.84 | 74.78 | 85.63 | 28.98 |
73
+ | [DenseOn-unsupervised](https://huggingface.co/lightonai/DenseOn-unsupervised) | 49.05 | 149 | 768 | 54.94 | 46.28 | 18.20 | 37.39 | 70.68 | 52.34 | 59.77 | 29.30 | 37.92 | 50.62 | 88.98 | 23.05 | 76.35 | 68.12 | 21.87 |
74
+ | [DenseOn](https://huggingface.co/lightonai/DenseOn) | 56.20 | 149 | 768 | 54.65 | **46.89** | 37.49 | **44.65** | 90.69 | **53.86** | **74.51** | 43.58 | **39.03** | 59.25 | **89.31** | 22.35 | 75.95 | 82.33 | 28.43 |
75
+
76
+ DenseOn reaches 56.20 NDCG@10 on BEIR, making it the top base-size dense retriever and the first sub-150M model to clear the 56 bar. At 149M parameters it decisively beats `GTE-ModernBERT` (55.19) at the same size, and more tellingly outperforms `snowflake-arctic-embed-l-v2.0` (55.22, 568M) and `Qwen3-Embedding-0.6B` (55.52, 595M) despite being roughly 4× smaller. DenseOn also stays within half a point of the strongest current-generation dense baselines, `pplx-embed-v1-0.6B` (56.70, 596M) and `jina-embeddings-v5-text-nano` (56.08, 239M), both substantially larger.
77
+
78
+ ### Decontaminated BEIR (12 datasets, NDCG@10)
79
+
80
+ Standard benchmarks risk overestimating model quality when training data overlaps with evaluation corpora. This is a non-negligible risk in our case, as our mixture explorations are mostly built on BEIR evaluation. To quantify this and ensure that our model has not memorized possible leakage, we built decontaminated versions of the BEIR datasets by removing samples found in both the mGTE training dataset and in our internal training datasets.
81
+ Since many retrieval models draw from similar public sources (Wikipedia, MS MARCO, Common Crawl, academic corpora), we expect significant overlap across models and believe the decontaminated benchmarks provide a meaningful, if imperfect, stress test. The decontaminated datasets are publicly available on HuggingFace.
82
+
83
+
84
+ Despite being in the toughest position (as the decontamination is based on our data), LateOn and DenseOn stay consistent under decontamination.
85
+ LateOn keeps its #1 position and DenseOn stays in the top four (only falling behind our other strong multi-vector model, ColBERT-Zero). Neither model flinches, which is direct evidence of generalization rather than overfitting.
86
+
87
+ More broadly, ColBERT models seems to generalize better under decontamination: all three ColBERT models hold or improve their ranking, and they take 2 of the top 3 decontaminated positions. Although DenseOn holds strong, some dense models are hit harder, for example `GTE-ModernBERT`, dropping from 8th to last, which is particularly interesting considering our base mixture is derived from theirs. This highlights the strength of our curation methodology. While other models such as `Qwen3-Embedding-0.6B` also drop some ranks, hinting at an overlap with the BEIR evaluation, it is worth noting that newer models, such as the new `jina-embeddings-v5` and `pplx-embed-v1-0.6b` seems to exhibit stronger evidence of generalization rather than overfitting.
88
+
89
+ <div align="center">
90
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/61dd91afdb45389905634d35/9kMmY66jIZsLxrQ6a7v3L.png" alt="LightOn" width="1024">
91
+ </div>
92
+
93
+
94
+ ## Related Checkpoints
95
+
96
+ | Model | Stage | Link |
97
+ |:---|:---|:---|
98
+ | **DenseOn-unsupervised** *(this card)* | Pre-training only | [`lightonai/DenseOn-unsupervised`](https://huggingface.co/lightonai/DenseOn-unsupervised) |
99
+ | DenseOn | Pre-training + fine-tuning | [`lightonai/DenseOn`](https://huggingface.co/lightonai/DenseOn) |
100
+ | LateOn-unsupervised | Multi-vector counterpart, pre-training only | [`lightonai/LateOn-unsupervised`](https://huggingface.co/lightonai/LateOn-unsupervised) |
101
+ | LateOn | Multi-vector counterpart, full pipeline | [`lightonai/LateOn`](https://huggingface.co/lightonai/LateOn) |
102
 
 
103
 
104
  ## Model Details
105
 
106
  ### Model Description
107
  - **Model Type:** Sentence Transformer
108
+ - **Base model:** [ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base)
109
  - **Maximum Sequence Length:** 512 tokens
110
  - **Output Dimensionality:** 768 dimensions
111
  - **Similarity Function:** Cosine Similarity
112
+ - **Language:** English
113
+ - **License:** Apache 2.0
 
114
 
115
  ### Model Sources
116
 
 
142
  from sentence_transformers import SentenceTransformer
143
 
144
  # Download from the 🤗 Hub
145
+ model = SentenceTransformer("lightonai/DenseOn-unsupervised")
146
  # Run inference
147
  queries = [
148
  "Which planet is known as the Red Planet?",
 
214
 
215
  ### BibTeX
216
 
217
+ #### DenseOn and LateOn
218
+
219
  ```bibtex
220
  @misc{sourty2026denseonlateon,
221
  title={DenseOn with the LateOn: Open State-of-the-Art Single and Multi-Vector Models},
 
225
  }
226
  ```
227
 
228
+ #### PyLate
229
+
230
  ```bibtex
231
  @inproceedings{DBLP:conf/cikm/ChaffinS25,
232
  author = {Antoine Chaffin and
 
253
  }
254
  ```
255
 
256
+ #### Sentence Transformers
257
+
258
+ ```bibtex
259
+ @inproceedings{reimers-2019-sentence-bert,
260
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
261
+ author = "Reimers, Nils and Gurevych, Iryna",
262
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
263
+ month = "11",
264
+ year = "2019",
265
+ publisher = "Association for Computational Linguistics",
266
+ url = "https://arxiv.org/abs/1908.10084"
267
+ }
268
+ ```
269
+
270
+ ## Acknowledgements
271
+
272
+ We thank Xin Zhang, Zach Nussbaum, Tom Aarsen, Bo Wang, Eugene Yang, Benjamin Clavié, Nandan Thakur, Oskar Hallström and Iacopo Poli for their valuable contributions and feedback. We are grateful to the teams behind Sentence Transformers and the BEIR benchmark, and to the open-source retrieval community, in particular the authors of Nomic Embed.
273
+
274
+ This work was granted access to the HPC resources of IDRIS under GENCI allocations AS011016449, A0181016214, and A0171015706 (Jean Zay supercomputer). We also acknowledge the Barcelona Supercomputing Center (BSC-CNS) for providing access to MareNostrum 5 under EuroHPC AI Factory Fast Lane project EHPC-AIF-2025FL01-445.
275
+
276
+
277
  <!--
278
  ## Glossary
279