Pooling-aware training cuts retrieval vectors 4x while boosting nDCG to 97.8% of unpooled

lateinteraction · x · 2026-11-11

tonywu71 highlights that token pooling already achieves extreme compression for late-interaction multimodal retrieval (ColPali, NeoMME), and pooling-aware training can push it further. Quoting tomaarsen's experiment: two ModernBERT-base models trained on the same 120k MS MARCO triplets, changing only the loss — pooling-aware training drops vectors per document from 193 to 49 while raising NanoBEIR@13 mean nDCG@10 from 49.03 to 51.23, retaining 97.8% of the unpooled score vs 94.6% with ordinary training.

Related event: Sentence Transformers v6.3 Adds Token Pooling-Aware Training for Multi-Vector Retrieval(6 posts)→

Original post →

More from Multimodal

Multimodal channel →