GPT-4 Has 1.8 Trillion Parameters. It Uses 2% of Them Per Token. Here Is Why That Is a Feature.

· Source: Towards AI - Medium · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Advanced, quick

Summary

GPT-4 employs a Mixture-of-Experts (MoE) architecture, featuring a parameter dictionary totaling 1.8 trillion parameters. When processing each token, the model activates approximately 36 billion parameters, representing only 2% of its total capacity, with the remaining 98% dormant. This selective activation is an intentional design choice, not a limitation, and is supported by a precise mathematical justification. In an MoE layer, the standard transformer feedforward block is replaced by a set of expert networks, and a routing function dynamically selects k of these experts for each token. The layer's output is formally computed as y = Σᵢ∈Top-k(x) G(x)ᵢ · Eᵢ(x), where x is the input token, Eᵢ(x) is expert i's output, G(x)ᵢ is its gating weight, and Top-k(x) identifies the highest-scoring experts.

Key takeaway

For AI Architects evaluating large language model deployments, understanding GPT-4's Mixture-of-Experts architecture is crucial. This design, activating only 2% of its 1.8 trillion parameters per token, implies significant efficiency gains for inference at scale. You should consider how sparse activation models can optimize resource utilization and reduce computational costs compared to dense models, influencing your infrastructure planning and operational strategies.

Key insights

GPT-4's Mixture-of-Experts architecture selectively activates 2% of its 1.8 trillion parameters per token, optimizing scale.

Principles

Method

The MoE layer computes y = Σᵢ∈Top-k(x) G(x)ᵢ · Eᵢ(x), where a gating function G(x)ᵢ selects k experts Eᵢ(x) based on input x for each token.

Topics

Best for: Research Scientist, CTO, VP of Engineering/Data, AI Scientist, Machine Learning Engineer, AI Architect

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Towards AI - Medium.