NeoMME-260M-Retriever

<p align="left"> <img src="https://github.com/tonywu71/colpali-cookbooks/blob/6ef1332da6bcb48c7ef1f19b25bfa555be7031a8/assets/neomme/neomme_logo.webp?raw=true" alt="NeoMME logo" style="max-height: 140px;"> </p>

NeoMME-Retriever (260M): Single-Tower Multimodal-Native Multilingual Foundation Encoder 🔎

[!IMPORTANT] NeoMME-Retriever (260M) variants:

• Default (transformers) [current]: Returns dense and multi-vector embeddings together with a single forward pass. Recommended for most use cases and inference.
• ST dense: Supports independent dense fine-tuning with Sentence Transformers.
• ST late-interaction: Supports independent multi-vector fine-tuning with Sentence Transformers.

Model summary

NeoMME-260M-Retriever is a model for multimodal document retrieval. Fine-tuned from NeoMME-260M, it encodes text queries and documents (text or page screenshots) using one shared bidirectional Transformer encoder.

A single forward pass returns both multi-vector and dense representations. Multi-vector embeddings use MeanMaxSim scoring, while dense embeddings use cosine similarity.

<table> <thead> <tr style="background-color: rgba(146, 81, 247, 0.20);"><th>Specification</th><th>Value</th></tr> </thead> <tbody> <tr><td>Parameters</td><td>263M</td></tr> <tr><td>Vocabulary</td><td>131,072 tokens</td></tr> <tr><td>Context length</td><td>16,384 tokens</td></tr> <tr><td>Hidden size</td><td>1,024</td></tr> <tr><td>Image patches</td><td>32 × 32 pixels, up to 2,048 pixels on the longest side (default)</td></tr> <tr><td>Multi-vector embeddings</td><td>128 dimensions per text token or image patch</td></tr> <tr><td>Dense embeddings</td><td>1,024 dimensions (Matryoshka: [128, 256, 512, 1,024])</td></tr> <tr><td>Dense pooling strategy</td><td>Mean</td></tr> </tbody> </table>

Performance

All scores use the metric shown at the full trained dimensions. Higher is better. ViDoRe v3, v2, and v1 measure visual document retrieval, while BEIR-15 measures text retrieval.

<table> <thead> <tr><th rowspan="2">Benchmark</th><th rowspan="2">Metric</th><th colspan="2" style="background-color: rgba(146, 81, 247, 0.20);">NeoMME-260M [current]</th><th colspan="2">NeoMME-800M</th></tr> <tr><th style="background-color: rgba(146, 81, 247, 0.20);">Late interaction</th><th style="background-color: rgba(146, 81, 247, 0.20);">Dense</th><th>Late interaction</th><th>Dense</th></tr> </thead> <tbody> <tr><td>ViDoRe v3</td><td>nDCG@10</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.5226</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.3907</td><td>0.5560</td><td>0.4391</td></tr> <tr><td>ViDoRe v2</td><td>nDCG@5</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.5218</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.4075</td><td>0.5591</td><td>0.4475</td></tr> <tr><td>ViDoRe v1</td><td>nDCG@5</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.8598</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.7552</td><td>0.8744</td><td>0.7993</td></tr> <tr><td>BEIR-15</td><td>nDCG@10</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.4881</td><td style="background-color: rgba(37, 99, 235, 0.20);">0.3055</td><td>0.5126</td><td>0.3686</td></tr> </tbody> </table>

Usage

Use NeoMME-Retriever with transformers. MeanMaxSim scoring requires sentence-transformers>=6.0.0:

# accelerate is an optional dependency needed only when using device_map="auto".
pip install -U accelerate transformers "sentence-transformers>=6.0.0"

The example below scores a text query against document-page images with late interaction, then with dense cosine similarity. One forward pass returns both embeddings.

from typing import Any, Literal

import requests
import torch
from PIL import Image
from sentence_transformers.util import cos_sim, mean_maxsim

from transformers import BatchFeature, NeoMMEForRetrieval, NeoMMEProcessor


def encode(
    messages: list[list[dict[str, Any]]],
    task: Literal["query", "document"],
) -> BatchFeature:
    return processor.apply_chat_template(
        messages,
        task=task,
        tokenize=True,
        return_dict=True,
        return_tensors="pt",
        processor_kwargs={"padding": "longest"},
    )


model_name = "Hcompany/NeoMME-260M-Retriever"
processor = NeoMMEProcessor.from_pretrained(model_name)
model = NeoMMEForRetrieval.from_pretrained(model_name, device_map="auto")

# Document images (our corpus)
image_urls = [
    "https://github.com/tonywu71/colpali-cookbooks/blob/6ef1332da6bcb48c7ef1f19b25bfa555be7031a8/examples/data/shift_kazakhstan.jpg?raw=true",
    "https://github.com/tonywu71/colpali-cookbooks/blob/6ef1332da6bcb48c7ef1f19b25bfa555be7031a8/examples/data/energy_electricity_generation.jpg?raw=true",
]
documents = [Image.open(requests.get(url, stream=True).raw) for url in image_urls]

# Queries
queries = [
    "Quelle partie de la production pétrolière du Kazakhstan provient de champs en mer ?",
    "Which hour of the day had the highest overall electricity generation in 2019?",
]

document_messages = [
    [{"role": "user", "content": [{"type": "image", "image": document}]}] for document in documents
]
query_messages = [[{"role": "user", "content": query}] for query in queries]

inputs_documents = encode(document_messages, "document").to(model.device)
inputs_text = encode(query_messages, "query").to(model.device)

with torch.inference_mode():
    document_outputs = model(**inputs_documents)
    query_outputs = model(**inputs_text)

late_scores = mean_maxsim(
    query_outputs.embeddings,
    document_outputs.embeddings,
    a_mask=inputs_text["attention_mask"],
    b_mask=inputs_documents["attention_mask"],
)
dense_scores = cos_sim(query_outputs.dense_embeddings, document_outputs.dense_embeddings)

# Expected: late_scores[0, 0] > late_scores[0, 1] and late_scores[1, 1] > late_scores[1, 0].
print(late_scores, dense_scores)

The score tensors have shape (num_queries, num_documents) and late_scores[i, j] / dense_scores[i, j] is the score between query i and document j. A larger value indicates a closer match.

Training

NeoMME-260M-Retriever was fine-tuned from NeoMME-260M on text retrieval and document-page images. Training uses a joint late-interaction and Matryoshka dense contrastive objective.

The NeoMME technical report describes the full fine-tuning recipe.

License

Model weights are released under the Apache 2.0 license.

Citation

@misc{lac2026neommesingletowermultimodalnativemultilingual,
      title={NeoMME: A Single-Tower Multimodal-Native Multilingual Foundation Encoder for Efficient Fine-Tuning and Inference},
      author={Aurélien Lac and Tony Wu},
      year={2026},
      eprint={2609.01657},
      archivePrefix={arXiv},
      primaryClass={cs.IR},
      url={https://arxiv.org/abs/2609.01657},
}