Mistral · 2024-04 · 스파스 MoE

Mixtral 8x22B

Mixtral 8x22B은(는) Mistral가 2024-04에 공개한 그룹화 쿼리 어텐션(GQA) Transformer로, 56개 층, 은닉 차원 6144, 컨텍스트 길이 65,536 토큰을 갖습니다.

레이어 스택

어텐션 ×56 · 레이어 수 56

핵심 사양

연구소Mistral
공개일2024-04
총 파라미터140.6 B
활성 파라미터39.1 B
컨텍스트65,536 tokens
어텐션GQA (48:8)
레이어 수56
은닉 차원6,144
어텐션 헤드48
어휘 크기32,000
위치 인코딩RoPE
정규화RMSNorm
활성화 함수silu
학습 정밀도
아키텍처 클래스MixtralForCausalLM

아키텍처 개요

t₁ t₂ t₃ t₄ 입력 토큰 Embedding · 어휘 크기 32,000 → 은닉 차원 6,144 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn 어텐션 헤드 48 KV 헤드 8 head dim 128 · RoPE 라우터 E1 E2 E3 E4 E5 E6 E7 E8 8 전문가 중 top-2 · 토큰당 약 39.1B 파라미터 활성 × 56 Transformer 블록 어텐션 FFN / MoE RMSNorm pre-norm 최종 정규화 · RMSNorm LM head → 어휘 크기 32,000 p p p → 다음 토큰 위치 인코딩 RoPE 학습 정밀도 컨텍스트 64K tok
동봉된 config.json을 바탕으로 작도 · 56층 / 은닉 6,144 / 컨텍스트 65,536. 도식은 본 사이트의 오리지널입니다.

어텐션

그룹화 쿼리 어텐션(GQA) — 48 q-heads / 8 kv-heads. RoPE θ=1,000,000.

FFN / MoE

스파스 Mixture-of-Experts 모델로, 8개 라우팅 전문가와 top-2 라우팅을 사용하며 토큰당 약 39.1B 파라미터가 활성화됩니다.

필드 단위 비교

같은 연구소의 이전 세대 모델과 비교합니다. 없으면 구조가 가장 유사한 모델과 비교. 배수 열 = 본 모델 ÷ 비교 대상.

이전 세대 Mistral 7B과의 전 필드 비교
모델Mixtral 8x22BMistral 7B배수
model_typemixtralmistral
architecturesMixtralForCausalLMMistralForCausalLM
hidden_size61444096×1.50
num_hidden_layers5632×1.75
num_attention_heads4832×1.50
num_key_value_heads88≈1
intermediate_size1638414336×1.14
hidden_actsilusilu
num_experts_per_tok2
rope_theta100000010000×100.00
max_position_embeddings6553632768×2.00
sliding_window4096
rms_norm_eps0.000010.00001≈1
vocab_size3200032000≈1
torch_dtypebfloat16bfloat16
attention_dropout0
bos_token_id11≈1
eos_token_id22≈1
initializer_range0.020.02≈1
num_local_experts8
output_router_logitsfalse
router_aux_loss_coef0.001
tie_word_embeddingsfalsefalse
transformers_version4.38.04.34.0.dev0
use_cachetruetrue

구조가 가장 유사한 모델

유사도는 0(공통 범주 특징 없음)에서 1(구조 프로파일 완전 일치) 사이의 값입니다.

config.json 전체 필드

24 개 필드
architecturesMixtralForCausalLM
attention_dropout0
bos_token_id1
eos_token_id2
hidden_actsilu
hidden_size6144
initializer_range0.02
intermediate_size16384
max_position_embeddings65536
model_typemixtral
num_attention_heads48
num_experts_per_tok2
num_hidden_layers56
num_key_value_heads8
num_local_experts8
output_router_logitsfalse
rms_norm_eps0.00001
rope_theta1000000
router_aux_loss_coef0.001
tie_word_embeddingsfalse
torch_dtypebfloat16
transformers_version4.38.0
use_cachetrue
vocab_size32000