14.5 hours on one RTX 3090: finetuned multi-vector retriever beats all general-purpose models

tomaarsen · x · 2026-08-26

Tom Aarsen, author of sentence-transformers, published an in-depth tutorial: v6.0 introduces a fourth model type, MultiVectorEncoder, for ColBERT-style late interaction retrieval with a complete training pipeline. Everything runs via pip install -U "sentence-transformers[train]".

The post walks through each component of finetuning multi-vector models — model, datasets, losses, training arguments, evaluators, and the trainer — with practical examples.

For evaluation, his multi-vector-encoder/mLateOn-medical, trained in 14.5 hours on a single RTX 3090, easily outperforms every general-purpose retrieval model he could find on his medical retrieval benchmark: dense, sparse, lexical, and multi-vector alike. The method also trains strong new multi-vector models from scratch.

Related event: Finetuning ColBERT on a single RTX 3090 beats general-purpose retrievers in medical search(13 posts)→

Original post →

More from coding & agent

coding & agent channel →