Hugging Face's logo
Xinping Zhao's picture
🔄 In a Training Loop

Xinping Zhao

Yuki131
27 87 179
zayidu's profile picture Chenliwei's profile picture Santu00's profile picture
·

AI & ML interests

LLMs, RAG, Embedding, Reranker——A Pokémon Trainer on a journey to become a Pokémon Master.

Recent Activity

upvoted a paper about 2 hours ago
zELO: ELO-inspired Training Method for Rerankers and Embedding Models
reacted to tomaarsen's post with 🔥 about 21 hours ago
🤗 I've released Sentence Transformers v6.3 and wrote a blogpost about its key new feature: token pooling-aware training, teaching multi-vector models to preserve retrieval quality with fewer document vectors. Details: Multi-vector models keep a vector per token, letting different query tokens match different parts of a document. Great for retrieval, but those vectors add up! Token pooling makes the representations smaller by merging similar document vectors into cluster means. Pooling-aware training teaches the encoder to preserve useful matches after those vectors are merged. With MultiVectorTokenPoolingLoss, you can wrap your standard retrieval loss to train on both full and pooled representations, reusing the same token embeddings without another encoder pass. In short: small overhead. I trained two ModernBERT-base models on the same 120k MS MARCO triplets, changing just the loss. At ~49 vectors per document instead of 193, pooling-aware training raises NanoBEIR@13 mean nDCG@10 from 49.03 to 51.23. That's 97.8% of its own unpooled score with ~4x fewer document vectors, compared with 94.6% retention for ordinary training. v6.3 also adds: - Adaptive token budgets, allocating vectors according to each document's content - Matryoshka training for shorter multi-vector token embeddings - Hugging Face kernels support for MaxSim, Ward clustering & supported transformers layers - Native AWS Neuron inference You can combine pooling-aware and Matryoshka training to learn both fewer vectors and fewer dimensions, then stack these with quantization to reduce storage per coordinate. For kernels, install sentence-transformers[kernels] and set use_kernels=True. The blog walks through the method, training code, checkpoints & evaluation, building on Stefan Josef's Learn to Pool and LightOn's pooling regularization work: ht tps://huggingface.co/blog/token-pooling I'd love to see what you train with this!
liked a model about 21 hours ago
facebook/meta-encoder-v1
View all activity

Organizations

Lychee Team's profile picture Zhejiang University's profile picture KaLM-Embedding's profile picture