mLateOn-unsupervised
State-of-the-Art Multilingual ColBERT Retrieval Model by LightOn
mDenseOn |
mLateOn |
DenseOn |
LateOn |
PyLate |
FastPlaid
π― TL;DR: The intermediate multilingual ColBERT checkpoint produced by Stage 1 only (unsupervised contrastive pre-training) of the mLateOn pipeline, trained on a multilingual dataset with 2.8B queryβdocument pairs across nine languages (including 25% cross-lingual pairs). Released as a strong starting point for your own supervised fine-tuning, knowledge distillation, or downstream adaptation.
About the mDenseOn / mLateOn Family
With DenseOn and LateOn, we demonstrated that an open, carefully curated data recipe can match closed-data retrieval models on English. mDenseOn and mLateOn extend this recipe to multilingual, long-context, and code retrieval.
Rather than independently collecting multilingual corpora from scratch (which would be expensive, uneven across languages, and hard to curate at the same quality), we applied the translate-train approach: machine-translating our validated English data into eight target languages (French, German, Italian, Spanish, Portuguese, Swedish, Norwegian, and Arabic) and adding cross-lingual pairs for cross-lingual alignment.
For more information, please read our multilingual models blog post, our English models blog post and our paper.
mLateOn-unsupervised
mLateOn-unsupervised is the output of the first stage of the mLateOn training pipeline. It has been pre-trained on a large, filtered multilingual corpus of approximately 2.8B queryβdocument pairs across nine languages (including 25% cross-lingual pairs) using in-batch contrastive learning, but has not yet been fine-tuned with mined hard negatives.
For most production use cases, you should use the fully-trained mLateOn instead. This unsupervised checkpoint is intended for:
- Researchers studying what each pipeline stage contributes in a multilingual setting
- Practitioners who want to fine-tune on their own domain-specific or language-specific data
- Distillation experiments where you want to start from a strong but un-aligned multilingual base
- Anyone running their own ablations on hard-negative mining strategies or translate-train recipes
Related Checkpoints
Model Description
- Model Type: PyLate ColBERT model
- Base model: mmBERT-base
- Query/Document Length: 8,192 tokens
- Output Dimensionality: 128 tokens
- Similarity Function: MaxSim
- Languages: English, French, German, Italian, Spanish, Portuguese, Swedish, Norwegian, Arabic
- License: Apache 2.0
Model Sources
Full Model Architecture
Usage
First install the PyLate library:
Retrieval
Use this model with PyLate to index and retrieve documents. The index uses FastPlaid for efficient similarity search.
Indexing documents
Load the ColBERT model and initialize the PLAID index, then encode and index your documents:
Note that you do not have to recreate the index and encode the documents every time. Once you have created an index and added the documents, you can re-use the index later by loading it:
Retrieving top-k documents for queries
Once the documents are indexed, you can retrieve the top-k most relevant documents for a given set of queries β including cross-lingual retrieval:
Reranking
If you only want to use the ColBERT model to perform reranking on top of your first-stage retrieval pipeline without building an index, you can simply use the rank function and pass the queries and documents to rerank:
Framework Versions
- Python: 3.11.10
- Sentence Transformers: 5.1.1
- PyLate: 1.3.4
- Transformers: 4.57.5
- PyTorch: 2.9.0+cu128
- Accelerate: 1.12.0
- Datasets: 3.6.0
- Tokenizers: 0.22.1
Citation
BibTeX
mDenseOn and mLateOn
DenseOn and LateOn
PyLate
Sentence Transformers
Acknowledgements
We thank Eugene Yang for his feedback on adapting our English study to multilinguality through translate-train. We again 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 on the original English study. We thank Orion Weller for building the FineWeb-derived Common Crawl split as well as for his feedback and help. We are grateful to the teams behind Sentence Transformers, BEIR, and MIRACL, and to the open-source retrieval community, in particular the authors of Nomic Embed.
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. This project is also supported by the OpenEuroLLM project, co-funded by the Digital Europe Programme under GA no. 101195233.