Papers
Topics
Authors
Recent
Search
2000 character limit reached

Giga-Embeddings: Large-Scale Embedding Representations

Updated 27 August 2026
  • Giga-Embeddings are scalable approaches for constructing, compressing, and training embedding representations at a very large scale, encompassing vocabularies with millions or billions of tokens, graph workloads with billions of edges, and long-document corpora.
  • Central engineering objectives for Giga-Embeddings include efficient memory usage, reducing communication, and maintaining quality under parameter sharing or distillation, considering constraints like vocabularies and embedding dimensions, and ensuring consistent indexing prior to deployment.
  • Key methods include parameter sharing replacing independent vectors with shared pools, sparse activation increasing model capacity per token, local computation avoiding global decoders, and distributed sharding placing embedding parameters across accelerator memory.

Giga-Embeddings denotes approaches for constructing, compressing, training, and serving embedding representations at very large scale. The term encompasses several distinct scalability problems: vocabularies containing millions or billions of tokens or entities, graph workloads with billions of edges, gigapixel image collections, long-document corpora, and high-throughput text-embedding models. Across these settings, the central engineering objectives are reducing per-item storage, limiting communication and pairwise computation, preserving representational quality under parameter sharing or distillation, and aligning training architecture with deployment constraints.

1. Scope and conceptual foundations

Embedding systems map discrete or structured objects to continuous vectors. For a categorical vocabulary of KK tokens and embedding dimension dd, a conventional embedding table contains KdKd trainable parameters. This linear dependence becomes prohibitive when tokens include character nn-grams, phrases, product identifiers, medical terms, URLs, or other long-tailed entities. A vocabulary of $3$ million words or phrases with d=300d=300 requires $900$ million parameters, approximately one billion (Svenstrup et al., 2017).

Giga-scale embedding problems differ in the object being embedded:

  • Tokens and categorical entities: words, identifiers, users, items, apps, or queries require highly memory-efficient lookup tables.
  • Graph nodes: entities in knowledge graphs or nodes in billion-edge networks require scalable sampling, partitioning, synchronization, and sparse optimization.
  • Documents and slides: long documents and whole-slide images require aggregation mechanisms that avoid representing each local fragment as an independently indexed vector.
  • Text inputs: general-purpose encoders must balance quality, dimensionality, inference throughput, and index size.

Several recurring strategies appear across the literature:

  1. Parameter sharing replaces independent vectors with shared component pools or factorized representations.
  2. Sparse activation increases resident model capacity while limiting computation per token.
  3. Local computation avoids global quadratic decoders or full-corpus intermediate representations.
  4. Distributed sharding places embedding parameters across machines or accelerator memory rather than replicating them.
  5. Self-supervision and distillation reduce dependence on labeled data and transfer structure from larger models.
  6. Dimensionality control uses pooling, truncation-aware objectives, or compact per-entity parameters to reduce vector-index cost.
  7. Adaptive sampling and synchronization concentrate computation and communication on informative or frequently accessed structures.

The term does not designate a single universally standardized architecture. “Giga-Embeddings” is also the name of a Russian-focused model (Kolodin et al., 25 Oct 2025) and of a family of high-throughput text encoders based on dense and Mixture-of-Experts architectures (Kolodin et al., 24 Aug 2026).

2. Memory-efficient categorical representations

Hash embeddings

Hash embeddings address the O(Kd)O(Kd) memory cost of conventional tables by combining a shared pool of BB vectors with token-specific importance parameters. For token ww, dd0 hash functions select component vectors from

dd1

and a trainable importance vector is assigned:

dd2

The resulting representation is

dd3

The trainable parameters comprise the shared component matrix and an importance matrix dd4, giving total parameter count

dd5

instead of dd6 for a standard embedding (Svenstrup et al., 2017).

The construction interpolates between two special cases. When dd7 and the importance coefficient is fixed to one, it becomes the hashing trick. When dd8 and hashing is the identity, it recovers a standard embedding table. Intermediate configurations allow tokens to share component vectors while learning token-specific combinations.

Multiple hashes reduce complete collisions. With dd9 independent assignments, the effective code space is approximately KdKd0. Importance parameters additionally allow tokens sharing components to use different mixtures or suppress unhelpful components. However, shared component vectors still receive gradients from multiple tokens, creating interference. In the main implementation, the same first hash KdKd1 maps tokens both to component assignments and to rows of KdKd2; therefore, first-layer collisions make tokens completely indistinguishable. A separate hash function for importance parameters reduces this risk, although that variant was not used in the main experiments.

Hash embeddings support dictionary-free operation: arbitrary strings can be mapped through a token hash without constructing a vocabulary in advance. This is relevant to online learning, streaming corpora, evolving vocabularies, rare entities, enormous KdKd3-gram spaces, and product, query, URL, code, or medical identifiers. The paper reports five times fewer parameters than a standard hashed embedding in its dictionary-free experiment, while matching or exceeding the baseline on six of seven text-classification datasets (Svenstrup et al., 2017).

MEmCom

Multi-Embedding Compression, or MEmCom, uses a shared hashed vector and a per-entity scalar. Let

KdKd4

be the shared table and

KdKd5

the entity-specific scalar table. For entity KdKd6,

KdKd7

An optional bias table KdKd8 gives

KdKd9

The unbiased parameter count is

nn0

and the biased parameter count is

nn1

compared with nn2 for an uncompressed table (Pansare et al., 2022).

Unlike naive hashing, MEmCom does not force entities sharing a bucket to have exactly the same vector. If nn3, then

nn4

The entities can therefore differ in magnitude, although they remain constrained to the same direction in the unbiased formulation. This scalar expressiveness is a fundamental limitation: unrelated entities sharing a bucket cannot obtain arbitrary independent directions.

The method sorts IDs by frequency. This is intended to accommodate nonuniform or power-law distributions, with frequent entities receiving stable per-entity parameters while the shared table remains small. In recommender-system experiments, MEmCom incurred approximately a nn5 relative loss in nDCG while compressing input embedding sizes by nn6, nn7, nn8, and nn9 on four datasets. On the Arcade RankNet experiment, it achieved $3$0 whole-model compression with less than $3$1 relative nDCG loss (Pansare et al., 2022).

MEmCom is distinct from hash embeddings in its parameterization. Hash embeddings use $3$2 token-specific mixing coefficients and $3$3 shared vectors, whereas MEmCom uses one shared vector and one entity-specific scalar. Both methods retain entity-specific information while replacing a full $3$4-dimensional parameter allocation with a compact shared representation.

3. Graph embeddings at large scale

Direct triple-to-skip-gram embedding

KG2Vec converts each RDF triple

$3$5

into a three-token sequence containing the subject URI, predicate URI, and object URI. A context window of two makes each element predict the other two. This avoids random-walk generation and applies skip-gram directly to triples (Soru et al., 2018).

The approach uses negative sampling with five negative words. Its principal scalability advantage is that the cost per positive pair depends on embedding dimension and the fixed number of negatives rather than the total vocabulary. KG2Vec processed more than $3$6 million triples in less than seven hours on conventional hardware. On the full DBpedia 2016-04 setting, it processed $3$7 triples in $3$8 seconds, while a thresholded version processed the same number of triples in $3$9 seconds.

The method captures local subject–predicate–object co-occurrence rather than arbitrary multi-hop paths. For link prediction, an LSTM scoring model trained on corrupted triples achieved Hits@10 of d=300d=3000 on the AKSW-bib graph, compared with d=300d=3001 for random negatives and d=300d=3002 for the analogy score. The results establish a CPU-oriented speed/storage trade-off, but do not demonstrate billion- or trillion-triple deployment (Soru et al., 2018).

Distributed information-oriented graph embedding

DistGER treats giga-scale graph embedding as a joint optimization of sampling, graph placement, and Skip-Gram training. Its three main components are:

  • InCoM: incremental information-centric computation;
  • MPGP: multi-proximity-aware streaming parallel graph partitioning;
  • DSGL: distributed Skip-Gram learning optimized for locality and synchronization (Fang et al., 2023).

Information-centric walks adapt both walk length and number of walks. Walks terminate when the correlation between path entropy and length falls below d=300d=3003, while additional walks are stopped when the change in relative entropy falls below d=300d=3004. Relative to a fixed configuration of length d=300d=3005 and d=300d=3006 walks per node, the paper reports a d=300d=3007 reduction in average walk length and an d=300d=3008 reduction in the number of walks.

InCoM updates entropy and correlation statistics incrementally, avoiding full-path transmission. Walker messages contain fixed-size state, estimated at d=300d=3009 bytes, compared with approximately $900$0 bytes for the full-path HuGE-D implementation. MPGP assigns nodes using first- and second-order proximity together with a load-balancing factor. It reduces cross-machine communication by an average of $900$1 relative to a workload-balancing partition and improves random-walk efficiency by approximately $900$2.

DSGL uses frequency-ordered embedding matrices, local context and negative buffers, multi-window shared negatives, and hotness-block synchronization. It achieves an average $900$3 speedup over Pword2vec and throughput up to $900$4 million nodes per second. DistGER’s reported end-to-end speedup over evaluated competitors ranges from $900$5 to $900$6. On Twitter, it processes the graph in $900$7 seconds on eight machines, compared with $900$8 seconds on one machine (Fang et al., 2023).

The principal remaining memory cost is the embedding table itself, with $900$9 storage and separate input and output matrices during training. Approximate hotness-aware synchronization reduces communication but permits stale parameters for cold nodes. Thus, DistGER improves the scalability of graph sampling and training without eliminating the fundamental storage dependence on the number of nodes.

TPU-sharded DeepWalk

HUGE separates distributed random-walk sampling from embedding optimization. A FlumeC++ data-processing system generates short random walks, aggregates source–destination co-occurrences into records of the form

O(Kd)O(Kd)0

and streams the sharded records to TPU training infrastructure (Mayer et al., 2023).

The reported configuration uses O(Kd)O(Kd)1 walks per node, walk length O(Kd)O(Kd)2, embedding dimension O(Kd)O(Kd)3, and uniform neighbor sampling. HUGE-TPU uses TPUEmbedding to shard the embedding table across TPU devices and to perform sparse lookups and updates without replicating the complete table on every replica. The principal throughput configuration uses 64 TPU v4 chips, providing approximately O(Kd)O(Kd)4 TiB of aggregate HBM before implementation overhead and non-embedding state.

On OGBN-Papers100M, HUGE-TPU reaches classification quality O(Kd)O(Kd)5, compared with O(Kd)O(Kd)6 for HUGE-CPU, while providing a reported O(Kd)O(Kd)7 end-to-end speedup over the CPU baseline. The TPU system processes approximately O(Kd)O(Kd)8 as many examples per second as HUGE-CPU under the compared configurations. Synthetic experiments reach O(Kd)O(Kd)9 billion nodes and BB0 billion edges.

HUGE does not eliminate preprocessing or input-pipeline costs. Random-walk generation, joins, aggregation, file storage, negative generation, and tensor delivery remain substantial. Its practical scalability depends on sufficient aggregate HBM and high-bandwidth interconnects. The method therefore represents a systems-oriented scaling strategy for a conventional DeepWalk objective rather than a new embedding loss (Mayer et al., 2023).

Local-to-global graph autoencoders

L2G2G reduces the quadratic decoder cost of a conventional graph autoencoder. A standard GAE produces BB1 and reconstructs

BB2

which incurs BB3 decoder work. L2G2G instead partitions the graph into overlapping patches, applies a shared GCN locally, synchronizes the local latent coordinate systems, and computes reconstruction losses only within patches (OuYang et al., 2024).

The method uses overlap-based rotation and translation synchronization. Local embeddings are aligned before decoding, and the synchronization is periodically refreshed during training. The loss is

BB4

This makes computation local while allowing overlap structure to transmit global consistency information. Compared with the original Local2Global formulation, synchronization affects optimization during training rather than being applied only after independent local models have converged.

Experiments on graphs up to approximately BB5 nodes show that L2G2G substantially improves over FastGAE on nearly every dataset and generally improves over GAE+L2G. It does not dominate standard GAE universally: standard GAE remains better on several reported AUC and AP comparisons. The giga-scale implications depend on balanced patches, controlled overlap, sparse patch connectivity, efficient synchronization, and treatment of between-patch edges (OuYang et al., 2024).

4. Long-context and slide-level embeddings

Long-document text representations

Jina Embeddings 2 addresses the fragmentation caused by 512-token encoders. A document of BB6 tokens requires approximately

BB7

vectors with a non-overlapping context limit BB8. Increasing the context limit from BB9 to ww0 can reduce the number of vectors for sufficiently long documents by approximately ww1, although documents longer than ww2 tokens still require truncation or an external document-level strategy (Günther et al., 2023).

The model uses a bidirectional BERT-derived encoder with ALiBi positional biases, mean pooling, and no next-sentence-prediction objective. The backbone is pretrained on approximately ww3 million English web documents and ww4 billion tokens, although pretraining uses only the first ww5 tokens of each document. Bidirectional ALiBi allows evaluation at longer sequence lengths without a learned absolute position-embedding table tied to a fixed maximum length.

The long-context architecture reduces vector count, raw vector storage, index construction cost, candidate aggregation, and potentially search latency. It does not necessarily reduce encoder FLOPs: dense self-attention has ww6 complexity, so an ww7-token forward pass can be substantially more expensive than multiple short passes.

Jina base v2 achieves an average nDCG@10 of ww8 on LoCo, compared with ww9 for ada-002 in the reported table. The benefit of longer context is task-dependent. NarrativeQA and BigPatent clustering improve, whereas WikiCities clustering slightly degrades because additional text can dilute an early salient signal. Long-context vectors favor holistic retrieval; chunk vectors retain localized evidence. A hybrid architecture can therefore be appropriate when both document-level recall and passage-level precision are required (Günther et al., 2023).

Slide-level self-supervision

Giga-SSL learns a slide-level representation directly from unlabeled whole-slide images rather than pretraining only a tile encoder. A frozen ResNet-18 tile encoder produces dd00-dimensional tile vectors. A submanifold sparse CNN receives these vectors together with their spatial coordinates and produces a dd01-dimensional slide vector (Lazard et al., 2022).

Each view samples dd02 tiles from a slide. Two augmented views of the same slide form a positive pair, while views from different slides are negatives. The slide-level aggregator is trained with an NT-XENT contrastive objective. At inference, dd03 non-augmented views are sampled, their embeddings are averaged, and the result is L2-normalized.

The TCGA pretraining collection contains dd04 whole-slide images and more than dd05 TB of compressed image data. The released slide representations occupy approximately dd06 MB. This is a lossy semantic transformation, not lossless image compression: the embeddings cannot reconstruct the original slides or support arbitrary pixel-level analyses.

On six TCGA-derived classification tasks, Giga-SSL reports full-data AUC values ranging from dd07 for BRCA mHRD to dd08 for RCC subtyping. With only 50 labeled slides, it improves over DeepAttnMIL by an average of dd09 AUC points. The method’s limitations include sampling failure for focal lesions, weak biological interpretability, shortcut learning from staining or acquisition signatures, and the inability to recover tile-level evidence directly from the fixed slide vector (Lazard et al., 2022).

5. Text-embedding model design and compression

Representation-oriented initialization and distillation

EmbeddingGemma is a 308-million-parameter encoder based on the Gemma 3 family. Its initialization converts a decoder-only Gemma 3 model into an encoder-decoder model using the T5Gemma recipe, continues training with the UL2 objective, and uses the resulting encoder to initialize the embedding model (Vera et al., 24 Sep 2025).

The model combines:

The released representation supports dd10, dd11, dd12, and dd13 dimensions. Matryoshka training explicitly optimizes overlapping prefixes, so truncation is part of the training objective rather than an arbitrary post-processing operation. On multilingual MTEB, the model obtains Mean(Task) dd14 at full dimension and remains at dd15 at dd16 dimensions. Quantization from bfloat16 to int4 produces only a small degradation in the reported aggregate results.

The model is trained through broad pre-finetuning, hard-negative fine-tuning, and multitask generalization. Its geometric distillation objective matches student and teacher embedding geometry rather than only teacher relevance scores. This approach is intended to transfer the arrangement of examples in embedding space, including distinctions around hard negatives. EmbeddingGemma demonstrates quality-efficiency improvements but does not establish billion-scale indexing, billion-parameter embedding models, or production-scale serving economics (Vera et al., 24 Sep 2025).

Russian-focused GigaEmbeddings

The 2025 Russian-focused GigaEmbeddings model adapts GigaChat-3B into a bidirectional encoder. It removes the causal mask, uses latent attention pooling, and removes the final nine of 36 transformer blocks, leaving an approximately 2.5-billion-parameter backbone (Kolodin et al., 25 Oct 2025).

Training has three stages:

  1. large-scale contrastive pretraining using web-scale title–passage data and synthetic queries;
  2. retrieval fine-tuning with seven curated hard negatives per query;
  3. multitask instruction tuning for retrieval, classification, and clustering.

The reported average score on 23 ruMTEB tasks is dd17. The model is strongest in classification, multiclass classification, clustering, and reranking, but larger models lead in some retrieval and semantic textual similarity categories. The pruning ablation decreases ruMTEB from dd18 to dd19, while instruction prompting improves the score from dd20 to dd21.

The paper identifies high-dimensional embeddings, approximately dd22 dimensions, and the approximately dd23-billion-parameter backbone as deployment constraints. Exact prompts, latent dimensions, corpus sizes, optimizer details, and complete task-level scores are not reported in the supplied text, and parts of the training mixture are not fully specified (Kolodin et al., 25 Oct 2025).

Mixture-of-Experts Giga-Embeddings

The 2026 Giga-Embeddings family contains dense 480M and 3B encoders and a sparse 10B-A1.8B encoder (Kolodin et al., 24 Aug 2026). All three are decoder-only LLMs adapted into bidirectional encoders and use mean pooling followed by normalization.

The largest model is a DeepSeekMoE-style encoder with:

  • dd24 billion resident parameters;
  • dd25 routed experts;
  • one shared expert;
  • top-4 routing;
  • approximately dd26 billion active parameters per token;
  • dd27-dimensional embeddings.

The 480M and 3B models are dense bidirectional Qwen3-based encoders with dd28- and dd29-dimensional embeddings, respectively. The sparse model therefore separates resident capacity from token-wise computation: its total memory footprint reflects approximately dd30 billion parameters, while only approximately dd31 billion parameters are active for each token.

The compact 480M model uses dimension-agnostic similarity-distribution distillation. Rather than matching teacher vectors directly, it matches teacher and student probability distributions over identical candidate sets:

dd32

The distillation temperature is dd33, and the combined objective weights distillation at dd34 and direct InfoNCE at dd35. The 480M model’s distillation ablation improves English, Russian, and code scores by dd36, dd37, and dd38, respectively.

Within the family, the 10B-A1.8B model leads on English, Russian, multilingual, and code MTEB suites, with scores of dd39, dd40, dd41, and dd42. The 480M distilled model scores dd43 on Russian MTEB, slightly exceeding FRIDA’s dd44 while using dd45 fewer parameters.

In the reported vLLM benchmark with 1024-token inputs, the 10B-A1.8B model processes dd46 thousand tokens per second, compared with dd47 thousand for the dense 3B model. This is a dd48 throughput improvement. Estimated FP16 weight storage is approximately dd49 GB for the 480M model, dd50 GB for the 3B model, and dd51 GB for the 10B-A1.8B model. The 480M model was not included in the throughput benchmark, and the serving measurements do not isolate the contribution of sparse activation from other architectural and implementation differences (Kolodin et al., 24 Aug 2026).

6. System-level trade-offs and unresolved problems

Giga-scale embedding systems must balance representation quality against memory, computation, communication, and index economics. No approach eliminates all scaling dependencies.

Parameter capacity remains fundamental. Hash embeddings reduce storage from dd52 to dd53, MEmCom to approximately dd54, and MoE models reduce active computation without reducing resident parameter storage proportionally. DistGER and HUGE reduce corpus, sampling, or communication overhead while retaining dd55 embedding storage. A compressed representation can therefore remain impractical when the number of entities is extremely large.

Collision and sharing interference are unavoidable in hash-based methods. Multiple hashes reduce complete collisions, but partial collisions still couple gradients. MEmCom preserves entity-specific scalar identity but constrains colliding entities to a shared direction. Hash assignments are fixed and cannot be directly optimized in the described implementations.

Locality and synchronization determine distributed graph performance. DistGER improves partition quality through proximity-aware streaming and reduces communication through fixed-size walker state and hotness-block synchronization. L2G2G aligns local latent spaces through overlap information, while HUGE shards embeddings across TPU HBM. These mechanisms exchange exact global coordination for locality, periodic synchronization, or specialized hardware.

Long context does not guarantee better representations. Jina Embeddings 2 reduces vector count for long documents, but dense attention increases encoder cost quadratically with sequence length, and mean pooling can dilute localized evidence. Giga-SSL similarly obtains compact slide vectors but can miss focal lesions when only five tiles are sampled.

Distillation and truncation involve geometry-quality trade-offs. EmbeddingGemma uses geometric teacher matching, spread-out regularization, and Matryoshka training. The newer Giga-Embeddings family instead matches teacher similarity distributions, enabling teacher and student dimensions to differ. These approaches improve compact models, but reported gains are generally modest and do not establish that compression preserves every downstream capability.

Evaluation remains workload-specific. KG2Vec demonstrates hundreds of millions of triples on a single server; DistGER demonstrates billion-edge graphs and billion-node synthetic graphs on an eight-machine cluster; HUGE demonstrates billion-node and ten-billion-edge synthetic graphs on TPU infrastructure; L2G2G evaluates graphs up to approximately dd56 nodes; Giga-SSL evaluates TCGA slides; and text encoders report benchmark quality and selected serving measurements. These results should not be treated as interchangeable evidence for arbitrary billion-document, trillion-edge, or giga-query-per-second deployments.

Important unresolved problems include:

  • distributed training and serving for billion-entity embedding tables;
  • optimizer-state and checkpoint memory;
  • dynamic vocabulary and entity insertion;
  • cold-start quality for rare or newly introduced items;
  • adversarial or structured hash collisions;
  • reproducibility with non-public training mixtures;
  • ANN recall and latency under aggressive dimensionality reduction;
  • routing balance and expert utilization in sparse encoders;
  • external clinical and cross-domain validation for compact slide representations;
  • focal-evidence preservation in long-context document embeddings;
  • interpretable mappings from compressed vectors back to source evidence;
  • jointly optimizing encoding, indexing, retrieval, and reranking cost.

Taken together, the research establishes several complementary principles. Hash-based parameter sharing and scalar identity parameters reduce the cost of categorical tables. Adaptive random walks, locality-aware partitioning, and accelerator sharding make graph embeddings feasible at billion-edge or billion-node scale. Long-context and slide-level aggregation reduce the number of vectors required to represent complex objects. Distillation, Matryoshka objectives, quantization, pruning, and Mixture-of-Experts architectures improve the quality-throughput-memory frontier of text encoders. The resulting conception of Giga-Embeddings is therefore not a single model class, but a systems framework in which representation capacity, data processing, distributed memory, sparse computation, and downstream index structure are designed together.

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Giga-Embeddings.