gliner-bi-small-v2.0

GLiNER-bi-Encoder: Scalable Zero-Shot Named Entity Recognition

About

GLiNER-bi-Encoder is a novel architecture for Named Entity Recognition (NER) that combines zero-shot flexibility with industrial-scale efficiency. Unlike the original GLiNER, which uses joint encoding, the bi-encoder design decouples text and entity-type encoding, enabling the recognition of thousands of entity types simultaneously with minimal computational overhead.

Key Advantages

Massive Scalability: Handle 1000+ entity types with near-constant inference speed when using pre-computed label embeddings

130× Faster: Up to 130× throughput improvement compared to uni-encoder approaches at 1024 entity types

State-of-the-Art Performance: Achieves 61.5% Micro-F1 on CrossNER benchmark in zero-shot setting

Efficient Caching: Pre-compute and cache entity type embeddings for instant reuse across millions of documents

Architecture

The bi-encoder architecture employs two specialized, independent transformers:

• Text Encoder: Processes input sequences using ModernBERT-based encoders (Ettin family)
• Label Encoder: Embeds entity type descriptions using specialized sentence transformers (BGE, MiniLM)

This separation removes the context-window bottleneck and enables:

• Pre-computation of entity type embeddings
• Constant memory usage for text encoding regardless of entity count
• Efficient nearest-neighbor search for entity matching

Model Variants

GLiNER-bi-V2 Models:

Recommendation: The base variant (194M) achieves 98% of large model performance while operating 2.6× faster, making it optimal for most production scenarios.

Installation & Usage

Installation

pip install gliner -U
pip install transformers==4.57.0

For flash attention support:

pip install flash-attn triton

Basic Usage

from gliner import GLiNER

# Load model
model = GLiNER.from_pretrained("knowledgator/gliner-bi-base-v2.0")

text = """
Cristiano Ronaldo dos Santos Aveiro (Portuguese pronunciation: [kɾiʃˈtjɐnu ʁɔˈnaldu]; born 5 February 1985) is a Portuguese professional footballer who plays as a forward for and captains both Saudi Pro League club Al Nassr and the Portugal national team. Widely regarded as one of the greatest players of all time, Ronaldo has won five Ballon d'Or awards, a record three UEFA Men's Player of the Year Awards, and four European Golden Shoes, the most by a European player.
"""

labels = ["person", "award", "date", "competitions", "teams"]

entities = model.predict_entities(text, labels, threshold=0.3)

for entity in entities:
    print(entity["text"], "=>", entity["label"])

Output:

Cristiano Ronaldo dos Santos Aveiro => person
5 February 1985 => date
Al Nassr => teams
Portugal national team => teams
Ballon d'Or => award
UEFA Men's Player of the Year Awards => award
European Golden Shoes => award

Advanced Usage: Pre-computing Entity Embeddings

For scenarios with large, static entity taxonomies (hundreds to millions of types):

from gliner import GLiNER

model = GLiNER.from_pretrained("knowledgator/gliner-bi-base-v2.0")

# Pre-compute embeddings for thousands of entity types
entity_types = ["person", "organization", "location", ...] # Can be thousands
texts = ["Your documents here", ...]

# Encode entity types once
entity_embeddings = model.encode_labels(entity_types, batch_size=8)

# Use pre-computed embeddings for fast inference
outputs = model.batch_predict_with_embeds(texts, entity_embeddings, entity_types)

This approach provides:

• 130× speedup at 1024 entity types
• Constant inference time regardless of entity count
• Efficient caching for repeated use

Flash Attention & Extended Context

model = GLiNER.from_pretrained(
    "knowledgator/gliner-bi-base-v2.0",
    _attn_implementation='flash_attention_2',
    max_len=2048
).to('cuda:0')

Zero-Shot NER Performance

Comprehensive evaluation across 19 diverse NER datasets:

CrossNER Zero-Shot Benchmark

Inference Speed Comparison

Throughput (examples/second) by number of entity types on H100 GPU (batch_size=1):

Key Insight: Bi-encoder with pre-computed embeddings maintains near-constant speed (5.2% degradation from 1→1024 labels) while uni-encoder shows 98.7% degradation.

Use Cases

Biomedical Entity Linking

Process millions of documents against UMLS (4M+ concepts), SNOMED CT, or other large medical ontologies with pre-computed embeddings.

Enterprise Knowledge Extraction

Deploy dynamic taxonomies that evolve without model retraining. Add new entity types instantly by computing their embeddings.

Scientific Literature Mining

Extract entities across multiple specialized domains (chemistry, biology, physics) with domain-specific label encoders.

Entity Linking with GLiNKER

GLiNER-bi-Encoder extends naturally to entity linking through the GLiNKER framework—a modular DAG-based pipeline for:

• Mention extraction with GLiNER
• Candidate retrieval from knowledge bases via pre-computed embeddings
• Entity disambiguation using bi-encoder scoring

Learn more: GLiNKER Repository

Model Details

Training Data

• Pre-training: 8M samples (Large/Base/Small), 10M samples (Edge) from FineFineWeb, annotated with GPT-4o
• Post-training: 40K high-quality samples with sequences up to 2048 tokens for long-context refinement

Training Configuration

• Focal Loss: α=0.7 (pre-training), α=0.8 (post-training), γ=2.0
• Optimizer: AdamW with differential learning rates (encoder: 1e-5, other: 3e-5)
• Context Length: 1024 tokens (pre-training), 2048 tokens (post-training)
• Maximum Span Width: 12 tokens
• Dropout: 0.35

Citation

If you use GLiNER-bi-Encoder in your research, please cite:

@misc{stepanov2026millionlabelnerbreakingscale,
      title={The Million-Label NER: Breaking Scale Barriers with GLiNER bi-encoder}, 
      author={Ihor Stepanov and Mykhailo Shtopko and Dmytro Vodianytskyi and Oleksandr Lukashov},
      year={2026},
      eprint={2602.18487},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2602.18487}, 
}

Acknowledgments

We sincerely thank Urchade Zaratiana (creator of GLiNER) and Tom Aarsen (maintainer of Sentence Transformers) for their foundational work.

Join Our Community

Connect with our community on Discord for news, support, and discussions: Join Discord

Resources

• Paper: arXiv preprint (coming soon)
• GLiNKER Framework: GLiNKER
• Model Collection: HuggingFace Collection

---

Knowledgator Engineering © 2026