ApX 标志ApX 标志

趋近智

ERNIE-4.5-21B-A3B-Base

活跃参数

21B

上下文长度

131K

模态

Text

架构

Mixture of Experts

KV 缓存速率

56 KB/tok

许可证

Apache 2.0

发布日期

30 Jun 2025

训练数据截止日期

Dec 2024

API 价格 (每 1M)

仅支持自建托管

硬件与系统要求

精度

Native

上下文

工作负载显存: 14.11 GB

INT4 · 8K context

基础权重: 11.18 GB

KV 缓存: 0.44 GB

运行时与计算图: 2.50 GB

最小所需硬件

1x NVIDIA L4 24GB

14.1 / 21.6 GB usable

1x RTX 4060 Ti 16GB

14.1 / 14.4 GB usable

Apple M4 24GB

14.1 / 18.0 GB usable

确定特定工作负载的硬件需求:

在显存计算器中打开

运行时要求

权重格式

FP16

CUDA

>= 11.8

PyTorch

>= 2.1

Transformers

>= 4.54.0.dev0

微架构精度注意力算子

Blackwell, Hopper

原生 FP16

FlashAttention-3

Ada Lovelace, Ampere

原生 FP16

FlashAttention-2

架构图

Input TokensToken EmbeddingPosition: AbsoluteHidden: 2.6k · Context: 131K · Vocab: 103.4kx 28 layersRMSNormPre-AttentionGrouped-Query Attention20Q / 4KV headsHead dim: 128+RMSNormPre-FFNSparse MoE FFN (6/64 experts)SwiGLUIntermediate: 1.5k+Final RMSNormOutput Logits

评估基准

没有可用的 ERNIE-4.5-21B-A3B-Base 评估基准。

排名

排名

-

编程排名

-

关于 ERNIE-4.5-21B-A3B-Base

ERNIE-4.5-21B-A3B-Base 模型是一款以文本为核心的混合专家(MoE)Transformer 模型,是百度文心(ERNIE)4.5 模型系列的核心组件。该特定变体通过模态特定提取过程衍生而来,从涉及数万亿 token 的大规模多模态预训练阶段中分离出文本相关的参数。其架构特征在于异构 MoE 结构,支持在训练期间跨模态共享参数,同时为特定数据类型保留专用专家。这种设计确保了文本表征不会因多模态联合训练而受损,从而在中文和英文的自然语言理解与生成方面实现高性能。

在技术层面,该模型采用稀疏架构,每层包含 64 个专家,通过路由机制为每个 token 激活 6 个专家,使得每次前向传播约有 30 亿个激活参数。这种稀疏性显著降低了计算开销,同时保持了 210 亿参数大模型的表征能力。注意力机制采用分组查询注意力(GQA),包含 20 个查询头和 4 个键值头,优化了显存带宽和推理速度。二维旋转位置嵌入(2D RoPE)的集成以及对 131,072 token 上下文窗口的支持,使其在处理长文档和复杂推理任务时非常高效。

为便于高效部署,文心 4.5 系列基于飞桨(PaddlePaddle)框架构建,并结合了多项硬件级优化,包括 FP8 混合精度训练和多专家并行协作。该模型支持 4 位和 2 位无损压缩等先进量化技术,使其能够在降低显存需求的情况下在多种硬件平台上运行。通过利用模态隔离路由和专用路由损失,该模型实现了极高的参数效率,适用于从复杂摘要到生产环境下的跨模态推理等工业级应用。

Specifications

注意力

注意力结构

Grouped-Query Attention

注意力头

20

键值头

4

注意力头维度

128

位置嵌入

Absolute Position Embedding

RoPE Theta

500,000

归一化

RMS Normalization

激活函数

SwigLU

维度

基准精度

FP16

隐藏维度大小

2,560

层数

28

FFN 中间层大小(稠密层)

1,536

分词器

词汇量大小

103,424

混合专家

专家参数总数

3.0B

专家数量

64

活跃专家

6

共享专家数

2

FFN 中间层大小(每专家)

1,536

MoE 前的稠密层数

1

关于 ERNIE 4.5

百度 ERNIE 4.5 系列包含十个大规模多模态模型。该系列采用异构专家混合(MoE)架构,在支持跨模态参数共享的同时,也为特定模态配置了专用参数,从而实现了高效的语言和多模态处理。


其他 ERNIE 4.5 模型