AIAny
AI Model2022
Icon for item

all-MiniLM-L6-v2

Encodes sentences and short paragraphs into 384-dimensional semantic vectors for retrieval, clustering, and similarity tasks. Fine-tuned from MiniLM-L6-H384 on ~1.17B sentence pairs with a contrastive objective; defaults to 256 word-piece truncation.

Introduction

Most production retrieval stacks need embeddings that balance quality, speed, and footprint — this model was built exactly for that tradeoff. Trained by fine-tuning a compact MiniLM backbone on over a billion sentence pairs with a contrastive loss, it delivers dense 384‑dim vectors that are cheap to compute and effective for semantic search, clustering, and similarity scoring at scale.

What Sets It Apart
  • Size-for-performance: a lightweight encoder (≈22.7M parameters) producing 384‑dim embeddings, making it much faster and smaller than typical large sentence encoders while retaining strong semantic quality.
  • Large, diverse training mix: fine-tuned on a concatenation of datasets (total ≈1.17B training tuples) including Reddit comments, S2ORC citation pairs, QA pairs, MS MARCO triplets, COCO captions and many QA/duplicate-question corpora — this broad training improves generalization across IR and QA-style tasks.
  • Contrastive fine-tuning: trained with an in-batch contrastive objective to push paired sentences closer and unrelated sentences apart, producing embeddings that work well with cosine-similarity retrieval.
  • Deployment-friendly: multiple exported formats and backends (PyTorch, ONNX, safetensors, OpenVINO, Rust weights) and compatibility with sentence-transformers and Hugging Face transformers APIs.
Who it's for — Fits & tradeoffs

Great fit if you need fast, low-memory sentence embeddings for semantic search, dense retrieval, clustering, or similarity ranking in English-dominant settings, especially when serving many queries or running on constrained hardware. It's also useful as a default embedding model for RAG pipelines and semantic tooling.

Look elsewhere if you need cross-lingual embeddings, state-of-the-art ranking on niche academic benchmarks, or representations for very long documents: the model truncates inputs by default (≈256 word pieces) and was optimized for short sentences/paragraphs. For maximal semantic accuracy on specialized domains, larger or domain-adapted encoders may outperform this compact model.

Information

  • Websitehuggingface.co
  • Organizationssentence-transformers, Hugging Face
  • Published date2022/03/02

Categories

More Items

Hugging Face
AI Model2022

A compact pretrained bidirectional Transformer distilled from BERT base for masked language modeling and downstream NLP fine-tuning. Trained via knowledge distillation on BookCorpus and English Wikipedia, it reduces size/latency (≈67M params) while preserving BERT-like representations.

Hugging Face
AI Model2022

Pretrained uncased English BERT base model for masked language modeling and next-sentence prediction. ~110M parameters, pretrained on BookCorpus and English Wikipedia; commonly fine-tuned for classification, token labeling, and question answering.

Hugging Face
AI Model2022

Autoregressive transformer that generates English text from a prompt and can be fine-tuned for downstream NLP tasks. This is the 124M-parameter GPT-2 checkpoint (causal LM, byte-level BPE, 1024-token context) hosted on Hugging Face.