Table of Contents What are Multi-Vector models? Why Finetune? Training Components Model Finetuning an existing multi-vector model Building one from a base transformer Which starting point should you pick? Dataset Data on the Hugging Face Hub Local Data Dataset Format Loss Function Training Arguments Evaluator Trainer Callbacks Multi-Dataset Training Evaluation Optimizing the index Acknowledgements Additional Resources Training Examples Documentation Sentence Transformers is a Python library for using and training embedding and reranker models for a wide range of applications, such as retrieval augmented generation, semantic search, semantic textual similarity, and more. Its v6.0 update introduces a fourth model type: MultiVectorEncoder, for ColBERT-style late interaction retrieval, alongside a complete training approach for it. In this blogpost, I'll show you how to use it to finetune a multi-vector model that outperforms general-purpose retrievers on your data. This method can also train strong new multi-vector models from scratch. Everything below runs on pip install -U "sentence-transformers[train]".
Training and Finetuning Multi-Vector Embedding Models with Sentence Transformers
We’re on a journey to advance and democratize artificial intelligence through open source and open science.

Key takeaways
The Hugging Face library has introduced a new model type, MultiVectorEncoder, for ColBERT-style late interaction retrieval, and provides a training approach for it.
- Multi-vector models keep one small vector per token and score a query against a document with the MaxSim operator.
- Finetuning multi-vector models improves their retrieval performance on a specific domain.
- The new model type is available in the v6.0 update of the Sentence Transformers library.
- Multi-vector models can be trained from scratch or finetuned from an existing model.
Summarised automatically by AI from the original article by Hugging Face Blog. AI can make mistakes, so check the original for details.
Story details
- Published
- By
- Tom Aarsen
- Format
- Article
- Original
- huggingface.co ↗



