ApX 标志ApX 标志

趋近智

Falcon-3B

参数

3B

上下文长度

33K

模态

Text

架构

Dense

KV 缓存速率

4 KB/tok

许可证

TII Falcon-LLM License 2.0

发布日期

17 Dec 2024

训练数据截止日期

-

API 价格 (每 1M)

仅支持自建托管

硬件与系统要求

精度

Native

上下文

工作负载显存: 3.93 GB

INT4 · 8K context

基础权重: 1.40 GB

KV 缓存: 0.03 GB

运行时与计算图: 2.50 GB

最小所需硬件

1x NVIDIA L4 24GB

3.9 / 21.6 GB usable

1x RTX 3060 12GB

3.9 / 10.8 GB usable

Apple M4 24GB

3.9 / 18.0 GB usable

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

在显存计算器中打开

运行时要求

权重格式

FP16

CUDA

>= 11.8

PyTorch

>= 2.1

Transformers

>= 4.40.1

微架构精度注意力算子

Blackwell, Hopper

原生 FP16

FlashAttention-3

Ada Lovelace, Ampere

原生 FP16

FlashAttention-2

架构图

Input TokensToken EmbeddingPosition: RoPEHidden: 1.5k · Context: 33Kx 32 layersRMSNormPre-AttentionMulti-Query Attention48Q / 1KV headsHead dim: 32+RMSNormPre-FFNFeed-Forward NetworkSwiGLU+Final RMSNormOutput Logits

评估基准

没有可用的 Falcon-3B 评估基准。

排名

排名

-

编程排名

-

关于 Falcon-3B

Falcon-3B 是由技术创新研究院 (TII) 开发的 Falcon 3 系列仅解码器 (decoder-only) 大语言模型成员之一。该模型变体拥有 30 亿参数,专为在各种硬件上高效部署而设计,包括笔记本电脑和单 GPU 等资源受限的系统。其主要目标是在一系列自然语言处理任务中提供强大的性能,重点关注推理、语言理解、指令遵循、代码生成和数学能力。Falcon-3B 模型还支持多语言功能,特别是英语、法语、西班牙语和葡萄牙语。

Falcon-3B 的架构基础是基于 Transformer 的因果仅解码器设计。它结合了多项创新技术以增强效率和性能。值得注意的是,它采用了分组查询注意力 (GQA) 机制,通过在注意力头之间共享参数来优化推理速度并减少键值 (KV) 缓存的内存消耗。该模型采用 SwiGLU 作为激活函数,并使用 RMSNorm 进行归一化,有助于实现稳定且有效的学习。位置嵌入采用旋转位置嵌入 (RoPE) 处理,以支持扩展上下文的理解。此外,该模型利用 FlashAttention 2 加速注意力计算,并拥有 131,000 个标记 (token) 的超大词表量,从而提高了压缩效率和下游任务性能。

Falcon-3B 及其指令微调版本是通过对规模更大的 Falcon3-7B-Base 模型进行剪枝 (pruning) 和知识蒸馏 (knowledge distillation) 等技术开发而成的,从而获得了一个高效且高性能的紧凑型模型。基础版支持 8,000 个标记的上下文长度,而指令微调版将此能力扩展至 32,000 个标记,使其能够处理更长、更复杂的输入并生成响应。这种设计范式使 Falcon-3B 成为在需要考虑计算资源的运行环境中实现高级 AI 功能的理想选择。

Specifications

注意力

注意力结构

Multi-Query Attention

注意力头

48

键值头

1

位置嵌入

ROPE

归一化

RMS Normalization

激活函数

SwigLU

维度

基准精度

FP16

隐藏维度大小

1,536

层数

32

关于 Falcon

TII Falcon 模型系列由仅含解码器的因果语言模型(7B、40B)组成。其架构基于 GPT-3 改进,集成了旋转位置嵌入(RoPE)、旨在提升推理效率的多查询注意力(Multi-Query Attention)以及用于加速运算的 FlashAttention。该系列模型在 RefinedWeb 数据集上训练而成。


其他 Falcon 模型