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

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering · Depth: Expert, quick

Summary

Hidden Decoding is a novel sequence-length scaling method designed to improve Large Language Models (LLMs) during continued pretraining without enlarging the Transformer backbone. This approach expands each token into "n" independent streams, utilizing separate embedding tables and maintaining the intermediate streams' key-value cache as context, thereby increasing internal computation per token. To manage computational costs at scale, Hidden Decoding incorporates Stream-Factorized Attention, which significantly reduces attention complexity from quadratic to roughly linear in "n" by limiting cross-stream attention to only a few layers. Experiments demonstrate its effectiveness at frontier scale, with models like WeLM-HD4-80B and WeLM-HD4-617B showing improvements over their non-HD baselines when trained at n=4. This marks Hidden Decoding as the first sequence-length scaling method proven at the 100B+ Mixture-of-Experts (MoE) scale, confirming that gains increase with higher "n" values.

Key takeaway

For Machine Learning Engineers optimizing large language models, if you are seeking to enhance an existing LLM's performance without incurring the immense cost of full backbone re-pretraining, consider implementing Hidden Decoding. This method allows you to scale computation per token effectively, improving model capabilities. You should explore its application during continued pretraining, utilizing Stream-Factorized Attention to maintain computational efficiency, especially for models at the 100B+ MoE scale.

Key insights

Hidden Decoding scales LLMs by increasing per-token computation via sequence-length expansion, improving performance without enlarging the Transformer backbone.

Principles

Method

Hidden Decoding expands each token into 'n' streams with independent embeddings and context. Stream-Factorized Attention limits cross-stream attention to few layers, reducing cost from quadratic to linear in 'n'.

In practice

Topics

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 Computation and Language.