Hidden Decoding at Scale: Latent Computation Scaling for Large Language Models

· Source: Takara TLDR - Daily AI Papers · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, medium

Summary

Hidden Decoding is a novel sequence-length scaling method designed to enhance Large Language Models (LLMs) during continued pretraining without requiring costly full pretraining or modifying the Transformer backbone's width. This approach addresses the scalability challenges of depth-recurrent Transformers by expanding each token into n independent streams, each with its own embedding table, and maintaining the intermediate streams' key-value cache as context. This allows for increased internal computation per token along the sequence-length dimension, compatible with standard large-model training. To ensure affordability at scale, Hidden Decoding incorporates Stream-Factorized Attention, which significantly reduces attention costs by primarily attending within streams and only selectively mixing across them. Experiments with WeLM-HD4-80B and WeLM-HD4-617B at n=4 demonstrated improvements over non-HD baselines, establishing Hidden Decoding as the first sequence-length scaling method successfully applied at the 100B+ Mixture-of-Experts (MoE) scale, with gains increasing as n grows.

Key takeaway

For AI Architects evaluating strategies to enhance existing large language models, Hidden Decoding offers a compelling alternative to costly full pretraining or widening models. You should investigate implementing this sequence-length scaling method during continued pretraining. Utilize its token stream expansion and Stream-Factorized Attention to boost performance on 100B+ MoE models. This approach allows you to achieve significant gains by allocating more computation per token without altering the core Transformer backbone.

Key insights

Scaling LLMs by expanding computation along the sequence-length dimension with token streams offers a fixed-backbone improvement path.

Principles

Method

Hidden Decoding expands each token into n streams using independent embedding tables and stream-factorized attention, keeping intermediate key-value caches as context.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Architect

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.