基础模型: google/gemma-3-1b-it
许可证: gemma
标签:
- gemma3
- gemma
- google
任务标签: 文本生成
库名称: transformers
额外授权标题: 在Hugging Face上访问Gemma
额外授权提示: >-
要访问Hugging Face上的Gemma,您需要审阅并同意Google的使用许可。请确保您已登录Hugging Face并点击下方按钮。请求将立即处理。
额外授权按钮内容: 确认许可
Gemma 3模型卡
模型页面: Gemma
[!注意]
本仓库对应Gemma 3模型的1B指令调优版本,采用量化感知训练(QAT)。
本仓库中的检查点未经量化,请确保使用您喜欢的工具进行int4量化
得益于QAT,该模型在显著降低加载内存需求的同时,能保持与bfloat16
相近的质量。
资源与技术文档:
- [Gemma 3技术报告][g3-tech-report]
- [负责任生成式AI工具包][rai-toolkit]
- [Kaggle上的Gemma][kaggle-gemma]
- [Vertex Model Garden上的Gemma 3][vertex-mg-gemma3]
使用条款: [条款][terms]
作者: Google DeepMind
模型信息
概述及输入输出的简要定义。
描述
Gemma是Google推出的一系列轻量级、先进的开放模型,基于与创建Gemini模型相同的研究和技术构建。Gemma 3模型是多模态的,可处理文本和图像输入并生成文本输出,提供预训练和指令调优变体的开放权重。Gemma 3拥有128K的大上下文窗口,支持超过140种语言,并提供了比以往版本更多的尺寸选择。Gemma 3模型适用于多种文本生成和图像理解任务,包括问答、摘要和推理。其相对较小的尺寸使其能够在资源有限的环境中部署,如笔记本电脑、台式机或您自己的云基础设施,从而普及先进AI模型的访问,促进所有人的创新。
输入与输出
-
输入:
- 文本字符串,如问题、提示或待摘要的文档
- 图像,归一化为896 x 896分辨率并编码为每个256个token
- 4B、12B和27B尺寸的总输入上下文为128K token,1B尺寸为32K token
-
输出:
- 生成的文本响应,如问题答案、图像内容分析或文档摘要
- 总输出上下文为8192 token
引用
@article{gemma_2025,
title={Gemma 3},
url={https://goo.gle/Gemma3Report},
publisher={Kaggle},
author={Gemma Team},
year={2025}
}
模型数据
用于模型训练的数据及数据处理方式。
训练数据集
这些模型在包含多种来源的文本数据集上训练。27B模型训练使用了14万亿token,12B模型使用了12万亿token,4B模型使用了4万亿token,1B模型使用了2万亿token。关键组成部分包括:
- 网页文档: 多样化的网页文本确保模型接触广泛的文体、主题和词汇。训练数据集包含超过140种语言的内容。
- 代码: 接触代码帮助模型学习编程语言的语法和模式,提高生成代码和理解代码相关问题的能力。
- 数学: 数学文本训练帮助模型学习逻辑推理、符号表示及解决数学查询。
- 图像: 广泛的图像使模型能够执行图像分析和视觉数据提取任务。
这些多样化数据源的组合对于训练一个能处理多种任务和数据格式的强大多模态模型至关重要。
数据预处理
关键的数据清理和过滤方法包括:
- CSAM过滤: 在数据准备过程中应用严格的CSAM(儿童性虐待材料)过滤,确保排除有害和非法内容。
- 敏感数据过滤: 作为使Gemma预训练模型安全可靠的一部分,使用自动化技术过滤训练集中的某些个人信息和其他敏感数据。
- 其他方法: 根据[我们的政策][safety-policies]进行基于内容质量和安全性的过滤。
实现信息
模型内部细节。
硬件
Gemma使用[Tensor Processing Unit (TPU)][tpu]硬件(TPUv4p、TPUv5p和TPUv5e)训练。训练视觉语言模型(VLM)需要大量计算能力。TPU专为机器学习中常见的矩阵运算设计,在此领域具有以下优势:
- 性能: TPU专为处理VLM训练中的大规模计算而设计,相比CPU可显著加速训练。
- 内存: TPU通常配备大量高带宽内存,可处理训练中的大模型和批量大小,从而提高模型质量。
- 可扩展性: TPU Pods(大型TPU集群)为处理大型基础模型日益增长的复杂性提供了可扩展的解决方案。
- 成本效益: 在许多场景下,TPU相比基于CPU的基础设施能提供更具成本效益的解决方案,尤其是考虑到因训练速度加快而节省的时间和资源。
- 这些优势与[Google的可持续发展承诺][sustainability]一致。
软件
训练使用[JAX][jax]和[ML Pathways][ml-pathways]完成。
JAX允许研究人员利用包括TPU在内的最新硬件,更快更高效地训练大型模型。ML Pathways是Google最新构建能跨多任务泛化的人工智能系统的努力。这特别适合包括这些大型语言模型在内的基础模型。
如[关于Gemini模型家族的论文][gemini-2-paper]所述,JAX和ML Pathways一起使用:“Jax和Pathways的‘单一控制器’编程模型允许单个Python进程协调整个训练运行,极大地简化了开发工作流程。”
评估
[!注意]
本节评估对应原始检查点,而非QAT检查点。
模型评估指标和结果。
基准测试结果
这些模型针对大量不同的数据集和指标进行了评估,涵盖文本生成的各个方面:
推理与事实性
基准测试 |
指标 |
Gemma 3 PT 1B |
Gemma 3 PT 4B |
Gemma 3 PT 12B |
Gemma 3 PT 27B |
[HellaSwag][hellaswag] |
10-shot |
62.3 |
77.2 |
84.2 |
85.6 |
[BoolQ][boolq] |
0-shot |
63.2 |
72.3 |
78.8 |
82.4 |
[PIQA][piqa] |
0-shot |
73.8 |
79.6 |
81.8 |
83.3 |
[SocialIQA][socialiqa] |
0-shot |
48.9 |
51.9 |
53.4 |
54.9 |
[TriviaQA][triviaqa] |
5-shot |
39.8 |
65.8 |
78.2 |
85.5 |
[Natural Questions][naturalq] |
5-shot |
9.48 |
20.0 |
31.4 |
36.1 |
[ARC-c][arc] |
25-shot |
38.4 |
56.2 |
68.9 |
70.6 |
[ARC-e][arc] |
0-shot |
73.0 |
82.4 |
88.3 |
89.0 |
[WinoGrande][winogrande] |
5-shot |
58.2 |
64.7 |
74.3 |
78.8 |
[BIG-Bench Hard][bbh] |
few-shot |
28.4 |
50.9 |
72.6 |
77.7 |
[DROP][drop] |
1-shot |
42.4 |
60.1 |
72.2 |
77.2 |
STEM与代码
基准测试 |
指标 |
Gemma 3 PT 4B |
Gemma 3 PT 12B |
Gemma 3 PT 27B |
[MMLU][mmlu] |
5-shot |
59.6 |
74.5 |
78.6 |
[MMLU][mmlu] (Pro COT) |
5-shot |
29.2 |
45.3 |
52.2 |
[AGIEval][agieval] |
3-5-shot |
42.1 |
57.4 |
66.2 |
[MATH][math] |
4-shot |
24.2 |
43.3 |
50.0 |
[GSM8K][gsm8k] |
8-shot |
38.4 |
71.0 |
82.6 |
[GPQA][gpqa] |
5-shot |
15.0 |
25.4 |
24.3 |
[MBPP][mbpp] |
3-shot |
46.0 |
60.4 |
65.6 |
[HumanEval][humaneval] |
0-shot |
36.0 |
45.7 |
48.8 |
多语言
基准测试 |
Gemma 3 PT 1B |
Gemma 3 PT 4B |
Gemma 3 PT 12B |
Gemma 3 PT 27B |
[MGSM][mgsm] |
2.04 |
34.7 |
64.3 |
74.3 |
[Global-MMLU-Lite][global-mmlu-lite] |
24.9 |
57.0 |
69.4 |
75.7 |
[WMT24++][wmt24pp] (ChrF) |
36.7 |
48.4 |
53.9 |
55.7 |
[FloRes][flores] |
29.5 |
39.2 |
46.0 |
48.8 |
[XQuAD][xquad] (all) |
43.9 |
68.0 |
74.5 |
76.8 |
[ECLeKTic][eclektic] |
4.69 |
11.0 |
17.2 |
24.4 |
[IndicGenBench][indicgenbench] |
41.4 |
57.2 |
61.7 |
63.4 |
多模态
基准测试 |
Gemma 3 PT 4B |
Gemma 3 PT 12B |
Gemma 3 PT 27B |
[COCOcap][coco-cap] |
102 |
111 |
116 |
[DocVQA][docvqa] (val) |
72.8 |
82.3 |
85.6 |
[InfoVQA][info-vqa] (val) |
44.1 |
54.8 |
59.4 |
[MMMU][mmmu] (pt) |
39.2 |
50.3 |
56.1 |
[TextVQA][textvqa] (val) |
58.9 |
66.5 |
68.6 |
[RealWorldQA][realworldqa] |
45.5 |
52.2 |
53.9 |
[ReMI][remi] |
27.3 |
38.5 |
44.8 |
[AI2D][ai2d] |
63.2 |
75.2 |
79.0 |
[ChartQA][chartqa] |
63.6 |
74.7 |
76.3 |
[VQAv2][vqav2] |
63.9 |
71.2 |
72.9 |
[BLINK][blinkvqa] |
38.0 |
35.9 |
39.6 |
[OKVQA][okvqa] |
51.0 |
58.7 |
60.2 |
[TallyQA][tallyqa] |
42.5 |
|
|