mlsanityData to 14 Oct 2026

KaLM-Embedding-V2: Superior Training Techniques and Data Inspire A Versatile Embedding Model

arXiv 2506.20923 ↗First posted 26 Jun 2025cs.CLcited by 36

Not on the map. Why?

Abstract (arXiv)

Recent advancements in Large Language Models (LLMs)-based text embedding models primarily focus on data scaling or synthesis, yet limited exploration of training techniques and data quality, thereby constraining performance. In this work, we propose KaLM-Embedding-V2 from the Lychee-KaLM team, a series of versatile and compact embedding models, systematically incentivizing advanced embedding capability in LLMs by superior training techniques and high-quality data. For model architecture, we implement the models in a 0.5B compact size with simple mean-pooling to produce fixed-length embeddings and remove the causal attention mask to enable fully bidirectional representation learning. For training techniques, we propose a progressive multi-stage training pipeline: pre-training on weakly supervised large-scale datasets, fine-tuning with supervised high-quality datasets, and contrastive distillation with fine-grained soft signals, integrated with focal-style reweighting and online hard-negative mixing to emphasize difficult samples and enrich hard negatives, respectively. For training data, we curate over 20 categories for pre-training and 100 categories for fine-tuning and contrastive distillation to improve both performance and generalization, leveraging task-specific instructions, hard-negative mining, and example-based multi-class labeling to ensure high quality. Combining these techniques, our KaLM-Embedding-V2 series achieves state-of-the-art performance on the Massive Text Embedding Benchmark, outperforming models of comparable size and rivaling models 3--26x larger, setting a new standard for versatile and compact embedding models under 1B parameters. The code, data, and models are available at https://kalm-embedding.github.io/.

Key facts. Cited by 36 papers in the map (data through 14 Oct 2026). Builds on 98 of its 138 references that are in the map. First posted on arXiv 26 Jun 2025.

Builds on

98 of 138 references.

References · 98

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  3. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  4. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  5. 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
  6. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
  7. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
  8. 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
  9. 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
  10. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  11. 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,212
  12. 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,173
38 more
  1. 2015A large annotated corpus for learning natural language inferencecited by 1,847
  2. 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,745
  3. 2016MS MARCO: A Human Generated MAchine Reading COmprehension Datasetcited by 1,605
  4. 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
  5. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  6. 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,375
  7. 2022No Language Left Behind: Scaling Human-Centered Machine Translationcited by 1,312
  8. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,288
  9. 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,261
  10. 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,253
  11. 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,252
  12. 2023LIMA: Less Is More for Alignmentcited by 1,100
  13. 2024M3-Embedding: Multi-Linguality, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillationcited by 1,073
  14. 2018XNLI: Evaluating Cross-lingual Sentence Representationscited by 915
  15. 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 723
  16. 2019CodeSearchNet Challenge: Evaluating the State of Semantic Code Searchcited by 654
  17. 2022MTEB: Massive Text Embedding Benchmarkcited by 640
  18. 2023C-Pack: Packed Resources For General Chinese Embeddingscited by 578
  19. 2019ELI5: Long Form Question Answeringcited by 575
  20. 2020Fact or Fiction: Verifying Scientific Claimscited by 554
  21. 2024NV-Embed: Improved Techniques for Training LLMs as Generalist Embedding Modelscited by 472
  22. 2021Large Dual Encoders Are Generalizable Retrieverscited by 456
  23. 2020Efficient Intent Detection with Dual Sentence Encoderscited by 433
  24. 2024Scaling Synthetic Data Creation with 1,000,000,000 Personascited by 417
  25. 2023Orca: Progressive Learning from Complex Explanation Traces of GPT-4cited by 379
  26. 2024Multilingual E5 Text Embeddings: A Technical Reportcited by 372
  27. 2021XL-Sum: Large-Scale Multilingual Abstractive Summarization for 44 Languagescited by 347
  28. 2022One Embedder, Any Task: Instruction-Finetuned Text Embeddingscited by 333
  29. 2023Improving Text Embeddings with Large Language Modelscited by 303
  30. 2022Matryoshka Representation Learningcited by 282
  31. 2020CLUE: A Chinese Language Understanding Evaluation Benchmarkcited by 278
  32. 2019PAWS-X: A Cross-lingual Adversarial Dataset for Paraphrase Identificationcited by 264
  33. 2024OpenCodeInterpreter: Integrating Code Generation with Execution and Refinementcited by 247
  34. 2024mGTE: Generalized Long-Context Text Representation and Reranking Models for Multilingual Text Retrievalcited by 236
  35. 2024Generative Representational Instruction Tuningcited by 233
  36. 2017SearchQA: A New Q&A Dataset Augmented with Context from a Search Enginecited by 225
  37. 2024Large Language Models for Data Annotation and Synthesis: A Surveycited by 205
  38. 2021PAQ: 65 Million Probably-Asked Questions and What You Can Do With Themcited by 200

50 most cited of 98 All 98 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 4; 2026: 32.

36 papers.

Sort:

By area:

  1. 2026SkillRet: A Large-Scale Benchmark for Skill Retrieval in LLM Agentscited by 39
  2. 2026jina-embeddings-v5-text: Task-Targeted Embedding Distillationcited by 22
  3. 2026MSVBench: Towards Human-Level Evaluation of Multi-Shot Video Generationcited by 12
  4. 2026F2LLM-v2: Inclusive, Performant, and Efficient Embeddings for a Multilingual Worldcited by 12
  5. 2025F2LLM Technical Report: Matching SOTA Embedding Performance with 6 Million Open-Source Datacited by 8
  6. 2026Unlocking Multimodal Document Intelligence: From Current Triumphs to Future Frontiers of Visual Document Retrievalcited by 8
  7. 2026LMEB: Long-horizon Memory Embedding Benchmarkcited by 7
  8. 2026ToolOmni: Enabling Open-World Tool Use via Agentic learning with Proactive Retrieval and Grounded Executioncited by 7
  9. 2026Query-focused and Memory-aware Reranker for Long Context Processingcited by 6
  10. 2026MSA: Memory Sparse Attention for Efficient End-to-End Memory Model Scaling to 100M Tokenscited by 6
  11. 2026BidirLM: From Text to Omnimodal Bidirectional Encoders by Adapting and Composing Causal LLMscited by 5
  12. 2026The Embedder's Dilemma: LLMs Are Better, but at What Cost?cited by 5
24 more
  1. 2025On The Role of Pretrained Language Models in General-Purpose Text Embeddings: A Surveycited by 3
  2. 2025C2LLM Technical Report: A New Frontier in Code Retrieval via Adaptive Cross-Attention Poolingcited by 3
  3. 2026Does My Embedding Reflect That A = B? Evaluating Mathematical Equivalence in Embedding Modelscited by 3
  4. 2026Structured Episodic Event Memorycited by 2
  5. 2026PosIR: Position-Aware Heterogeneous Information Retrieval Benchmarkcited by 2
  6. 2026MathAgent: Adversarial Evolution of Constraint Graphs for Mathematical Reasoning Data Synthesiscited by 2
  7. 2026EvoEmbedding: Evolvable Representations for Long-Context Retrieval and Agentic Memorycited by 2
  8. 2026SkillReason: Reasoning-Enhanced Agent Skill Retrieval for Implicit User Requestscited by 2
  9. 2025Learning to Extract Rational Evidence via Reinforcement Learning for Retrieval-Augmented Generationcited by 1
  10. 2026No More K-means: Single-Stage Sparse Coding for Efficient Multi-Vector Retrievalcited by 1
  11. 2026SABER-Math: Automated Benchmark for Information Retrieval Evaluation in Mathematicscited by 1
  12. 2026EmbeddingRWKV: State-Centric Retrieval with Reusable Statescited by 0
  13. 2026CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrievalcited by 0
  14. 2026CONE: Embeddings for Complex Numerical Data Preserving Unit and Variable Semanticscited by 0
  15. 2026Embedding-based In-Context Prompt Training for Enhancing LLMs as Text Encoderscited by 0
  16. 2026ML-Embed: Inclusive and Efficient Embeddings for a Multilingual Worldcited by 0
  17. 2026SEA-LION-Embedding: Open and Reproducible Text Embeddings for Southeast Asiacited by 0
  18. 2026ReverseEOL: Improving Training-free Text Embeddings via Text Reversal in Decoder-only LLMscited by 0
  19. 2026Lychee-Reranker: Fast but Not Late Interaction for Compressed Document Rerankingcited by 0
  20. 2026BitNet Text Embeddingscited by 0
  21. 2026MTEB-BR: A Text Embedding Benchmark for Brazilian Portuguesecited by 0
  22. 2026What Makes Linguistic Representations Good Models of High-Level Visual Perception in the Human Brain?cited by 0
  23. 2026An Instrument to Evaluate Governance Proposals: AI Policy Analysis at Scalecited by 0
  24. 2026Verifiable Retrieval for Legal Reasoning through Structure-driven Knowledge Graphcited by 0