KaLM-Embedding-V2: Superior Training Techniques and Data Inspire A Versatile Embedding Model
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
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,212
- 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,173
38 more
- 2015A large annotated corpus for learning natural language inferencecited by 1,847
- 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,745
- 2016MS MARCO: A Human Generated MAchine Reading COmprehension Datasetcited by 1,605
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,375
- 2022No Language Left Behind: Scaling Human-Centered Machine Translationcited by 1,312
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,288
- 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,261
- 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,253
- 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,252
- 2023LIMA: Less Is More for Alignmentcited by 1,100
- 2024M3-Embedding: Multi-Linguality, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillationcited by 1,073
- 2018XNLI: Evaluating Cross-lingual Sentence Representationscited by 915
- 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 723
- 2019CodeSearchNet Challenge: Evaluating the State of Semantic Code Searchcited by 654
- 2022MTEB: Massive Text Embedding Benchmarkcited by 640
- 2023C-Pack: Packed Resources For General Chinese Embeddingscited by 578
- 2019ELI5: Long Form Question Answeringcited by 575
- 2020Fact or Fiction: Verifying Scientific Claimscited by 554
- 2024NV-Embed: Improved Techniques for Training LLMs as Generalist Embedding Modelscited by 472
- 2021Large Dual Encoders Are Generalizable Retrieverscited by 456
- 2020Efficient Intent Detection with Dual Sentence Encoderscited by 433
- 2024Scaling Synthetic Data Creation with 1,000,000,000 Personascited by 417
- 2023Orca: Progressive Learning from Complex Explanation Traces of GPT-4cited by 379
- 2024Multilingual E5 Text Embeddings: A Technical Reportcited by 372
- 2021XL-Sum: Large-Scale Multilingual Abstractive Summarization for 44 Languagescited by 347
- 2022One Embedder, Any Task: Instruction-Finetuned Text Embeddingscited by 333
- 2023Improving Text Embeddings with Large Language Modelscited by 303
- 2022Matryoshka Representation Learningcited by 282
- 2020CLUE: A Chinese Language Understanding Evaluation Benchmarkcited by 278
- 2019PAWS-X: A Cross-lingual Adversarial Dataset for Paraphrase Identificationcited by 264
- 2024OpenCodeInterpreter: Integrating Code Generation with Execution and Refinementcited by 247
- 2024mGTE: Generalized Long-Context Text Representation and Reranking Models for Multilingual Text Retrievalcited by 236
- 2024Generative Representational Instruction Tuningcited by 233
- 2017SearchQA: A New Q&A Dataset Augmented with Context from a Search Enginecited by 225
- 2024Large Language Models for Data Annotation and Synthesis: A Surveycited by 205
- 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
Citing papers per year: 2025: 4; 2026: 32.
36 papers.
Sort:
By area:
- 2026SkillRet: A Large-Scale Benchmark for Skill Retrieval in LLM Agentscited by 39
- 2026jina-embeddings-v5-text: Task-Targeted Embedding Distillationcited by 22
- 2026MSVBench: Towards Human-Level Evaluation of Multi-Shot Video Generationcited by 12
- 2026F2LLM-v2: Inclusive, Performant, and Efficient Embeddings for a Multilingual Worldcited by 12
- 2025F2LLM Technical Report: Matching SOTA Embedding Performance with 6 Million Open-Source Datacited by 8
- 2026Unlocking Multimodal Document Intelligence: From Current Triumphs to Future Frontiers of Visual Document Retrievalcited by 8
- 2026LMEB: Long-horizon Memory Embedding Benchmarkcited by 7
- 2026ToolOmni: Enabling Open-World Tool Use via Agentic learning with Proactive Retrieval and Grounded Executioncited by 7
- 2026Query-focused and Memory-aware Reranker for Long Context Processingcited by 6
- 2026MSA: Memory Sparse Attention for Efficient End-to-End Memory Model Scaling to 100M Tokenscited by 6
- 2026BidirLM: From Text to Omnimodal Bidirectional Encoders by Adapting and Composing Causal LLMscited by 5
- 2026The Embedder's Dilemma: LLMs Are Better, but at What Cost?cited by 5
24 more
- 2025On The Role of Pretrained Language Models in General-Purpose Text Embeddings: A Surveycited by 3
- 2025C2LLM Technical Report: A New Frontier in Code Retrieval via Adaptive Cross-Attention Poolingcited by 3
- 2026Does My Embedding Reflect That A = B? Evaluating Mathematical Equivalence in Embedding Modelscited by 3
- 2026Structured Episodic Event Memorycited by 2
- 2026PosIR: Position-Aware Heterogeneous Information Retrieval Benchmarkcited by 2
- 2026MathAgent: Adversarial Evolution of Constraint Graphs for Mathematical Reasoning Data Synthesiscited by 2
- 2026EvoEmbedding: Evolvable Representations for Long-Context Retrieval and Agentic Memorycited by 2
- 2026SkillReason: Reasoning-Enhanced Agent Skill Retrieval for Implicit User Requestscited by 2
- 2025Learning to Extract Rational Evidence via Reinforcement Learning for Retrieval-Augmented Generationcited by 1
- 2026No More K-means: Single-Stage Sparse Coding for Efficient Multi-Vector Retrievalcited by 1
- 2026SABER-Math: Automated Benchmark for Information Retrieval Evaluation in Mathematicscited by 1
- 2026EmbeddingRWKV: State-Centric Retrieval with Reusable Statescited by 0
- 2026CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrievalcited by 0
- 2026CONE: Embeddings for Complex Numerical Data Preserving Unit and Variable Semanticscited by 0
- 2026Embedding-based In-Context Prompt Training for Enhancing LLMs as Text Encoderscited by 0
- 2026ML-Embed: Inclusive and Efficient Embeddings for a Multilingual Worldcited by 0
- 2026SEA-LION-Embedding: Open and Reproducible Text Embeddings for Southeast Asiacited by 0
- 2026ReverseEOL: Improving Training-free Text Embeddings via Text Reversal in Decoder-only LLMscited by 0
- 2026Lychee-Reranker: Fast but Not Late Interaction for Compressed Document Rerankingcited by 0
- 2026BitNet Text Embeddingscited by 0
- 2026MTEB-BR: A Text Embedding Benchmark for Brazilian Portuguesecited by 0
- 2026What Makes Linguistic Representations Good Models of High-Level Visual Perception in the Human Brain?cited by 0
- 2026An Instrument to Evaluate Governance Proposals: AI Policy Analysis at Scalecited by 0
- 2026Verifiable Retrieval for Legal Reasoning through Structure-driven Knowledge Graphcited by 0
- 2026Verifiable Retrieval for Legal Reasoning through Structure-driven Knowledge Graph9 Oct 2026
- 2026The Embedder's Dilemma: LLMs Are Better, but at What Cost?13 Aug 2026
- 2026SkillReason: Reasoning-Enhanced Agent Skill Retrieval for Implicit User Requests9 Aug 2026
- 2026An Instrument to Evaluate Governance Proposals: AI Policy Analysis at Scale30 Jul 2026
- 2026MTEB-BR: A Text Embedding Benchmark for Brazilian Portuguese6 Jul 2026
- 2026SABER-Math: Automated Benchmark for Information Retrieval Evaluation in Mathematics29 Jun 2026
- 2026BitNet Text Embeddings24 Jun 2026
- 2026Does My Embedding Reflect That A = B? Evaluating Mathematical Equivalence in Embedding Models22 Jun 2026
- 2026Lychee-Reranker: Fast but Not Late Interaction for Compressed Document Reranking22 Jun 2026
- 2026EvoEmbedding: Evolvable Representations for Long-Context Retrieval and Agentic Memory19 Jun 2026
- 2026ReverseEOL: Improving Training-free Text Embeddings via Text Reversal in Decoder-only LLMs4 Jun 2026
- 2026SEA-LION-Embedding: Open and Reproducible Text Embeddings for Southeast Asia2 Jun 2026
24 more
- 2026No More K-means: Single-Stage Sparse Coding for Efficient Multi-Vector Retrieval28 May 2026
- 2026What Makes Linguistic Representations Good Models of High-Level Visual Perception in the Human Brain?22 May 2026
- 2026ML-Embed: Inclusive and Efficient Embeddings for a Multilingual World14 May 2026
- 2026SkillRet: A Large-Scale Benchmark for Skill Retrieval in LLM Agents7 May 2026
- 2026Embedding-based In-Context Prompt Training for Enhancing LLMs as Text Encoders2 May 2026
- 2026ToolOmni: Enabling Open-World Tool Use via Agentic learning with Proactive Retrieval and Grounded Execution15 Apr 2026
- 2026MathAgent: Adversarial Evolution of Constraint Graphs for Mathematical Reasoning Data Synthesis13 Apr 2026
- 2026BidirLM: From Text to Omnimodal Bidirectional Encoders by Adapting and Composing Causal LLMs2 Apr 2026
- 2026F2LLM-v2: Inclusive, Performant, and Efficient Embeddings for a Multilingual World19 Mar 2026
- 2026LMEB: Long-horizon Memory Embedding Benchmark13 Mar 2026
- 2026MSA: Memory Sparse Attention for Efficient End-to-End Memory Model Scaling to 100M Tokens6 Mar 2026
- 2026CONE: Embeddings for Complex Numerical Data Preserving Unit and Variable Semantics5 Mar 2026
- 2026MSVBench: Towards Human-Level Evaluation of Multi-Shot Video Generation27 Feb 2026
- 2026CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrieval23 Feb 2026
- 2026Unlocking Multimodal Document Intelligence: From Current Triumphs to Future Frontiers of Visual Document Retrieval23 Feb 2026
- 2026jina-embeddings-v5-text: Task-Targeted Embedding Distillation17 Feb 2026
- 2026Query-focused and Memory-aware Reranker for Long Context Processing12 Feb 2026
- 2026PosIR: Position-Aware Heterogeneous Information Retrieval Benchmark13 Jan 2026
- 2026EmbeddingRWKV: State-Centric Retrieval with Reusable States10 Jan 2026
- 2026Structured Episodic Event Memory10 Jan 2026
- 2025C2LLM Technical Report: A New Frontier in Code Retrieval via Adaptive Cross-Attention Pooling24 Dec 2025
- 2025F2LLM Technical Report: Matching SOTA Embedding Performance with 6 Million Open-Source Data2 Oct 2025
- 2025On The Role of Pretrained Language Models in General-Purpose Text Embeddings: A Survey28 Jul 2025
- 2025Learning to Extract Rational Evidence via Reinforcement Learning for Retrieval-Augmented Generation21 Jul 2025
- 2026SkillRet: A Large-Scale Benchmark for Skill Retrieval in LLM Agentscited by 39
- 2026ToolOmni: Enabling Open-World Tool Use via Agentic learning with Proactive Retrieval and Grounded Executioncited by 7
- 2026EvoEmbedding: Evolvable Representations for Long-Context Retrieval and Agentic Memorycited by 2
- 2026SABER-Math: Automated Benchmark for Information Retrieval Evaluation in Mathematicscited by 1
- 2026SABER-Math: Automated Benchmark for Information Retrieval Evaluation in Mathematics29 Jun 2026
- 2026EvoEmbedding: Evolvable Representations for Long-Context Retrieval and Agentic Memory19 Jun 2026
- 2026SkillRet: A Large-Scale Benchmark for Skill Retrieval in LLM Agents7 May 2026
- 2026ToolOmni: Enabling Open-World Tool Use via Agentic learning with Proactive Retrieval and Grounded Execution15 Apr 2026
- 2026The Embedder's Dilemma: LLMs Are Better, but at What Cost?cited by 5
- 2026MathAgent: Adversarial Evolution of Constraint Graphs for Mathematical Reasoning Data Synthesiscited by 2
- 2026SkillReason: Reasoning-Enhanced Agent Skill Retrieval for Implicit User Requestscited by 2
- 2026Verifiable Retrieval for Legal Reasoning through Structure-driven Knowledge Graphcited by 0
- 2026Verifiable Retrieval for Legal Reasoning through Structure-driven Knowledge Graph9 Oct 2026
- 2026The Embedder's Dilemma: LLMs Are Better, but at What Cost?13 Aug 2026
- 2026SkillReason: Reasoning-Enhanced Agent Skill Retrieval for Implicit User Requests9 Aug 2026
- 2026MathAgent: Adversarial Evolution of Constraint Graphs for Mathematical Reasoning Data Synthesis13 Apr 2026
- 2026Structured Episodic Event Memorycited by 2
- 2025Learning to Extract Rational Evidence via Reinforcement Learning for Retrieval-Augmented Generationcited by 1
- 2026EmbeddingRWKV: State-Centric Retrieval with Reusable Statescited by 0
- 2026CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrievalcited by 0
- 2026CQ-CiM: Hardware-Aware Embedding Shaping for Robust CiM-Based Retrieval23 Feb 2026
- 2026EmbeddingRWKV: State-Centric Retrieval with Reusable States10 Jan 2026
- 2026Structured Episodic Event Memory10 Jan 2026
- 2025Learning to Extract Rational Evidence via Reinforcement Learning for Retrieval-Augmented Generation21 Jul 2025
- 2026jina-embeddings-v5-text: Task-Targeted Embedding Distillationcited by 22
- 2026MSVBench: Towards Human-Level Evaluation of Multi-Shot Video Generationcited by 12
- 2026F2LLM-v2: Inclusive, Performant, and Efficient Embeddings for a Multilingual Worldcited by 12
- 2025F2LLM Technical Report: Matching SOTA Embedding Performance with 6 Million Open-Source Datacited by 8
- 2026Unlocking Multimodal Document Intelligence: From Current Triumphs to Future Frontiers of Visual Document Retrievalcited by 8
- 2026LMEB: Long-horizon Memory Embedding Benchmarkcited by 7
- 2026Query-focused and Memory-aware Reranker for Long Context Processingcited by 6
- 2026MSA: Memory Sparse Attention for Efficient End-to-End Memory Model Scaling to 100M Tokenscited by 6
- 2026BidirLM: From Text to Omnimodal Bidirectional Encoders by Adapting and Composing Causal LLMscited by 5
- 2025On The Role of Pretrained Language Models in General-Purpose Text Embeddings: A Surveycited by 3
- 2025C2LLM Technical Report: A New Frontier in Code Retrieval via Adaptive Cross-Attention Poolingcited by 3
- 2026Does My Embedding Reflect That A = B? Evaluating Mathematical Equivalence in Embedding Modelscited by 3
12 most cited of 24 All 24 (JSON)
- 2026An Instrument to Evaluate Governance Proposals: AI Policy Analysis at Scale30 Jul 2026
- 2026MTEB-BR: A Text Embedding Benchmark for Brazilian Portuguese6 Jul 2026
- 2026BitNet Text Embeddings24 Jun 2026
- 2026Does My Embedding Reflect That A = B? Evaluating Mathematical Equivalence in Embedding Models22 Jun 2026
- 2026Lychee-Reranker: Fast but Not Late Interaction for Compressed Document Reranking22 Jun 2026
- 2026ReverseEOL: Improving Training-free Text Embeddings via Text Reversal in Decoder-only LLMs4 Jun 2026
- 2026SEA-LION-Embedding: Open and Reproducible Text Embeddings for Southeast Asia2 Jun 2026
- 2026No More K-means: Single-Stage Sparse Coding for Efficient Multi-Vector Retrieval28 May 2026
- 2026What Makes Linguistic Representations Good Models of High-Level Visual Perception in the Human Brain?22 May 2026
- 2026ML-Embed: Inclusive and Efficient Embeddings for a Multilingual World14 May 2026
- 2026Embedding-based In-Context Prompt Training for Enhancing LLMs as Text Encoders2 May 2026
- 2026BidirLM: From Text to Omnimodal Bidirectional Encoders by Adapting and Composing Causal LLMs2 Apr 2026
12 newest of 24 All 24 (JSON)