G FlashPapers
Google DeepMind Technical Report (2024)

Gemini 1.5: Unlocking Multimodal Long-Context

Google DeepMind introduces the Gemini 1.5 family, demonstrating near-perfect retrieval recall across millions of tokens of text, audio, and video. Powered by a highly compute-efficient sparse Mixture-of-Experts (MoE) architecture.

Sparse MoE Architecture Up to 10M Tokens Context Native Multimodal Processing In-Context Translation

Why This Matters

Until recently, Large Language Models (LLMs) were severely constrained by their context window. Standard models capped out at 4,000 to 32,000 tokens. Even state-of-the-art models like GPT-4 Turbo (128k) and Claude 3 (200k) struggled to hold a single large book or a complex codebase in memory.

When models hit this "context wall," developers are forced to use complex RAG pipelines or chunking strategies. These workarounds are lossy—they miss global context, fail to resolve long-range dependencies, and cannot easily synthesize information scattered across hours of video or audio.

Gemini 1.5 shatters this limit. By supporting up to 10,000,000 tokens natively, it can ingest entire repositories, days of audio, or dozens of long documents simultaneously, treating all modalities with equal retrieval fidelity.

The Big Idea

The core breakthrough of Gemini 1.5 is achieving unprecedented context lengths without a linear explosion in compute cost. This is made possible by a sparse Mixture-of-Experts (MoE) architecture for Gemini 1.5 Pro, and optimized online distillation for the highly efficient Gemini 1.5 Flash.

Instead of activating all parameters for every single token, the MoE router dynamically directs input tokens to the most relevant "expert" sub-networks. This conditional computation allows the model to scale its capacity (total parameters) to match or exceed Gemini 1.0 Ultra, while maintaining the latency and serving efficiency of a much smaller model.

How It Works

Deep dive into the architecture, retrieval mechanics, and in-context learning capabilities.

1. Sparse Mixture-of-Experts (MoE)

Traditional transformers process every token through every layer. In contrast, Gemini 1.5 Pro utilizes an MoE design. A gating network routes each token to only a few "experts."

Mathematically, if $x$ is the input representation, the output of the MoE layer is: $$y = \sum_{i=1}^{N} G(x)_i E_i(x)$$ where $G(x)_i$ is the routing probability for expert $E_i$, and $N$ is the total number of experts. Because $G(x)$ is sparse (most entries are $0$), only a fraction of the compute is spent per token.

2. The "Needle-in-a-Haystack" Retrieval

To prove the model doesn't "forget" information in millions of tokens, the authors used the Needle-in-a-Haystack (NIAH) test. They hid a secret phrase at varying depths across different context lengths.

Unlike prior models whose retrieval performance degrades sharply past 100k tokens, Gemini 1.5 Pro maintains >99% recall across all modalities (Text, Audio, and Video) up to 1M tokens, and retains 99.2% recall up to 10M tokens in text.

Interactive NIAH Modality Simulator

Adjust context size and needle depth to see how Gemini 1.5 Pro compares to traditional baselines.

128K tokens 1M tokens 10M tokens
0% (Top) 50% 100% (Bottom)
Context Haystack Grid
Yellow block indicates hidden "needle"
Recall Probability
Gemini 1.5 Pro 99.7%
GPT-4 Turbo 0% (Failed)

At 1M tokens, traditional models exceed their context window limits entirely and fail to retrieve the needle.

3. In-Context Language Learning (MTOB)

One of the most striking capabilities enabled by Gemini 1.5's massive context window is in-context learning (ICL) at an extreme scale.

To test this, the researchers evaluated the model on the Machine Translation from One Book (MTOB) benchmark. They provided the model with a 500-page reference grammar book, a dictionary, and 400 parallel sentences for Kalamang—a Papuan language with fewer than 200 speakers and virtually zero web presence.

With all materials loaded directly into the prompt (~250k tokens), Gemini 1.5 Pro learned to translate English to Kalamang at a level comparable to a human language learner who studied the same materials.

Kalamang In-Context Translation Sandbox

Toggle context depth to see how the inclusion of the grammar book and wordlist directly improves translation accuracy.

Prompt Context Loaded
No linguistic materials loaded. The model must rely solely on its pre-trained weights.
Gemini 1.5 Pro Output

The wind is telling a story. (Hallucinated/Incorrect)

Human Reference Translation

Bayu esun et mulko mambaran.

Human Evaluation Score (0-6): 0.18 / 6.0

Headline Results

Gemini 1.5 Pro surpasses Gemini 1.0 Ultra across a wide array of text and multimodal benchmarks while using substantially less training compute.

Benchmark Performance Comparison

Gemini 1.5 Pro (May 2024)
Gemini 1.5 Flash
Gemini 1.0 Ultra
GPT-4 Turbo

Limitations & Open Questions

Despite its impressive technical leaps, the authors identify several critical limitations:

  • The "Needle" is Simple: Retrieving a single structured fact (a needle) is fundamentally easier than synthesizing complex inferences scattered across millions of tokens. The authors suggest moving to "multiple needles" benchmarks to push the boundaries further.
  • Long-Context Safety Risks: A massive context window increases the attack surface for prompt injection. Malicious instructions can be hidden deep within a 1,000,000-token document, making them highly difficult to detect during pre-filtering.
  • Tonal Regressions: The rigorous post-training safety alignment for Gemini 1.5 models has resulted in slight regressions in tone, sometimes causing the model to respond in an overly defensive or cold manner when refusing unsafe queries.

Glossary

Mixture-of-Experts (MoE)

A neural network architecture that splits layers into multiple independent sub-networks called "experts." A routing mechanism directs each input token to only the most appropriate experts, drastically reducing active parameter count and compute requirements.

Needle-in-a-Haystack (NIAH)

A diagnostic evaluation method that tests a model's information retrieval capacity by hiding a specific target fact (the "needle") inside a massive volume of irrelevant text (the "haystack").

In-Context Learning (ICL)

The emergent ability of large language models to adapt to new tasks during inference solely from instructions, templates, or reference materials provided in the prompt, without modifying the model's underlying weights.

Character Error Rate (CER) & Word Error Rate (WER)

Standard evaluation metrics for speech-to-text transcription. CER measures the percentage of characters transcribed incorrectly, while WER measures the percentage of words transcribed incorrectly. Lower values indicate higher accuracy.

Citation

@article{gemini1.5techreport,
  title={Gemini 1.5: Unlocking multimodal understanding across millions of tokens of context},
  author={Gemini Team, Google},
  journal={arXiv preprint arXiv:2403.05530},
  year={2024}
}
Made with Flash Papers — make your own