We release two new best-in-class multilingual retrieval models:
LFM2.5-Embedding-350M — A dense bi-encoder, one vector per document. Smallest, fastest index.
LFM2.5-ColBERT-350M — A late-interaction model. One vector per token, matched via MaxSim. Higher accuracy and better generalization at the cost of index size.
Both models are 350M params and the first bidirectional members of the LFM family, built on LFM2.5-350M-Base. They can be used as a drop-in replacement for your current RAG pipeline and target fast, cheap, and reliable multilingual / cross-lingual search across 11 languages.
Find more details about the bidirectional architecture and training recipe in our blog post.
This model can be used as a multi-vector (ColBERT-style late interaction) retriever directly with Sentence Transformers via the MultiVectorEncoder.
pip install "sentence-transformers>=6.0.0"
python
1from sentence_transformers import MultiVectorEncoder
23model = MultiVectorEncoder("LiquidAI/LFM2.5-ColBERT-350M", trust_remote_code=True)45query ="Which planet is known as the Red Planet?"6documents =[7"Venus is often called Earth's twin because of its similar size and proximity.",8"Mars, known for its reddish appearance, is often referred to as the Red Planet.",9"Jupiter, the largest planet in our solar system, has a prominent red spot.",10"Saturn, famous for its rings, is sometimes mistaken for the Red Planet.",11]1213query_embeddings = model.encode_query([query])14document_embeddings = model.encode_document(documents)15print(query_embeddings[0].shape, document_embeddings[0].shape)16# (32, 128) (17, 128)1718# MaxSim late-interaction scoring (the Mars document ranks highest)19scores = model.similarity(query_embeddings, document_embeddings)20print(scores)21# tensor([[27.1621, 28.2578, 27.7266, 28.1992]])
Using PyLate
Use this model with PyLate to index and retrieve documents. The index uses FastPLAID for efficient similarity search. First, install PyLate and transformers:
pip install -U pylate
Indexing documents
Load LFM2.5-ColBERT-350M and initialize the PLAID index, then encode and index your documents:
python
1from pylate import indexes, models, retrieve
23# Step 1: Load the ColBERT model (trust_remote_code applies the bidirectional patches)4model = models.ColBERT(5 model_name_or_path="LiquidAI/LFM2.5-ColBERT-350M",6 trust_remote_code=True,7)8model.tokenizer.pad_token = model.tokenizer.eos_token
910# Step 2: Initialize the PLAID index11index = indexes.PLAID(12 index_folder="pylate-index",13 index_name="index",14 override=True,# This overwrites the existing index if any15)1617# Step 3: Encode the documents18documents_ids =["1","2","3"]19documents =["document 1 text","document 2 text","document 3 text"]2021documents_embeddings = model.encode(22 documents,23 batch_size=32,24 is_query=False,# Ensure that it is set to False to indicate that these are documents, not queries25 show_progress_bar=True,26)2728# Step 4: Add document embeddings to the index by providing embeddings and corresponding ids29index.add_documents(30 documents_ids=documents_ids,31 documents_embeddings=documents_embeddings,32)
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:
python
1# To load an index, simply instantiate it with the correct folder/name and without overriding it2index = indexes.PLAID(3 index_folder="pylate-index",4 index_name="index",5)
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. To do so, initialize the ColBERT retriever with the index you want to search in, encode the queries, and then retrieve the top-k documents to get the top matches ids and relevance scores:
python
1# Step 1: Initialize the ColBERT retriever2retriever = retrieve.ColBERT(index=index)34# Step 2: Encode the queries5queries_embeddings = model.encode(6["query for document 3","query for document 1"],7 batch_size=32,8 is_query=True,# Ensure that it is set to True to indicate that these are queries9 show_progress_bar=True,10)1112# Step 3: Retrieve top-k documents13scores = retriever.retrieve(14 queries_embeddings=queries_embeddings,15 k=10,# Retrieve the top 10 matches for each query16)
Reranking
If you only want to use LFM2.5-ColBERT-350M 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:
MKQA. Cross-lingual capabilities (subset of the 11 languages we target).
Model
Type
AVG
ar
de
en
es
fr
it
ja
ko
no
pt
sv
LiquidAI/LFM2.5-ColBERT-350M
late
0.694
0.608
0.709
0.748
0.711
0.715
0.707
0.703
0.640
0.689
0.703
0.700
LiquidAI/LFM2.5-Embedding-350M
dense
0.691
0.610
0.709
0.738
0.708
0.715
0.703
0.685
0.630
0.691
0.710
0.708
Alibaba-NLP/gte-multilingual-base
dense
0.675
0.567
0.692
0.741
0.705
0.703
0.697
0.655
0.563
0.698
0.700
0.699
LiquidAI/LFM2-ColBERT-350M
late
0.646
0.554
0.696
0.754
0.711
0.710
0.667
0.658
0.558
0.541
0.669
0.589
Qwen/Qwen3-Embedding-0.6B
dense
0.638
0.520
0.671
0.723
0.678
0.672
0.671
0.635
0.543
0.620
0.667
0.620
lightonai/GTE-ModernColBERT-v1
late
0.459
0.092
0.532
0.754
0.552
0.615
0.510
0.275
0.166
0.503
0.524
0.524
lightonai/LateOn
late
0.454
0.157
0.492
0.755
0.537
0.577
0.481
0.316
0.209
0.472
0.502
0.501
lightonai/DenseOn
dense
0.435
0.165
0.482
0.751
0.491
0.553
0.457
0.325
0.222
0.438
0.443
0.453
BAAI/bge-large-en-v1.5
dense
0.413
0.133
0.471
0.748
0.450
0.531
0.461
0.208
0.172
0.456
0.443
0.467
Alibaba-NLP/gte-modernbert-base
dense
0.295
0.060
0.333
0.736
0.273
0.417
0.291
0.100
0.052
0.332
0.326
0.330
Inference speed - llama.cpp
End-to-end latency on MacBook Pro M4 Max via llama.cpp at fp16, measured at 32-token queries and 256-token documents. Docs cached means that the document embeddings are pre-computed and looked up (from an index).
For large-scale production-grade enterprise deployments, we also experiment with an internal GPU stack to deliver extremely low-latency serving under high inbound load. We observe latencies as low as 1 ms.
@misc{PyLate,
title={PyLate: Flexible Training and Retrieval for Late Interaction Models},
author={Chaffin, Antoine and Sourty, Raphaël},
url={https://github.com/lightonai/pylate},
year={2024}
}