Gemma 3 1b Pt

🚀 Gemma 3 模型卡片

Gemma 3 是谷歌推出的輕量級多模態模型，基於與Gemini相同的技術構建。它支持文本和圖像輸入，輸出文本，具有 128K 大上下文窗口和超 140 種語言支持，適用於多種文本生成和圖像理解任務。

🚀 快速開始

安裝

要運行 Gemma 3 模型，首先需要安裝 transformers 庫。Gemma 3 從 transformers 4.50.0 版本開始支持。

$ pip install -U transformers

運行示例

使用 pipeline API 運行

from transformers import pipeline
import torch

pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16)
output = pipe("Eiffel tower is located in", max_new_tokens=50)

在單/多 GPU 上運行模型

import torch
from transformers import AutoTokenizer, Gemma3ForCausalLM

ckpt = "google/gemma-3-1b-pt"
tokenizer = AutoTokenizer.from_pretrained(ckpt)
model = Gemma3ForCausalLM.from_pretrained(
    ckpt,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)

prompt = "Eiffel tower is located in"
model_inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

input_len = model_inputs["input_ids"].shape[-1]

with torch.inference_mode():
    generation = model.generate(**model_inputs, max_new_tokens=50, do_sample=False)
    generation = generation[0][input_len:]

decoded = tokenizer.decode(generation, skip_special_tokens=True)
print(decoded)

✨ 主要特性

• 多模態處理：支持文本和圖像輸入，輸出文本，適用於多種文本生成和圖像理解任務。
• 大上下文窗口：擁有 128K 的大上下文窗口，能夠處理更長的輸入。
• 多語言支持：支持超過 140 種語言，具有更廣泛的應用場景。
• 輕量化設計：相對較小的模型尺寸，可在資源有限的環境中部署，如筆記本電腦、臺式機或自有云基礎設施。

📦 安裝指南

要使用 Gemma 3 模型，需要安裝 transformers 庫。Gemma 3 從 transformers 4.50.0 版本開始支持。

$ pip install -U transformers

💻 使用示例

基礎用法

from transformers import pipeline
import torch

pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16)
output = pipe("Eiffel tower is located in", max_new_tokens=50)

高級用法

import torch
from transformers import AutoTokenizer, Gemma3ForCausalLM

ckpt = "google/gemma-3-1b-pt"
tokenizer = AutoTokenizer.from_pretrained(ckpt)
model = Gemma3ForCausalLM.from_pretrained(
    ckpt,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)

prompt = "Eiffel tower is located in"
model_inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

input_len = model_inputs["input_ids"].shape[-1]

with torch.inference_mode():
    generation = model.generate(**model_inputs, max_new_tokens=50, do_sample=False)
    generation = generation[0][input_len:]

decoded = tokenizer.decode(generation, skip_special_tokens=True)
print(decoded)

📚 詳細文檔

模型信息

描述

Gemma 是谷歌推出的一系列輕量級、先進的開放模型，基於與 Gemini 模型相同的研究和技術構建。Gemma 3 模型是多模態的，能夠處理文本和圖像輸入，並生成文本輸出。預訓練和指令調優變體的權重均開放。Gemma 3 具有 128K 的大上下文窗口，支持超過 140 種語言，並且比以前的版本有更多的尺寸可供選擇。Gemma 3 模型非常適合各種文本生成和圖像理解任務，包括問答、摘要和推理。其相對較小的尺寸使其能夠在資源有限的環境中部署，如筆記本電腦、臺式機或自有云基礎設施，使先進的人工智能模型更易於獲取，促進創新。

輸入和輸出

引用

@article{gemma_2025,
    title={Gemma 3},
    url={https://goo.gle/Gemma3Report},
    publisher={Kaggle},
    author={Gemma Team},
    year={2025}
}

模型數據

訓練數據集

這些模型在包含多種來源的文本數據集上進行訓練。27B 模型使用 14 萬億個標記進行訓練，12B 模型使用 12 萬億個標記，4B 模型使用 4 萬億個標記，1B 模型使用 2 萬億個標記。主要組成部分如下：

• 網絡文檔：多樣化的網絡文本集合確保模型接觸到廣泛的語言風格、主題和詞彙。訓練數據集包含超過 140 種語言的內容。
• 代碼：讓模型接觸代碼有助於學習編程語言的語法和模式，提高生成代碼和理解代碼相關問題的能力。
• 數學：在數學文本上進行訓練有助於模型學習邏輯推理、符號表示和處理數學查詢。
• 圖像：廣泛的圖像使模型能夠執行圖像分析和視覺數據提取任務。

數據預處理

對訓練數據應用了以下關鍵的數據清理和過濾方法：

• CSAM 過濾：在數據準備過程的多個階段應用了嚴格的 CSAM（兒童性虐待材料）過濾，以確保排除有害和非法內容。
• 敏感數據過濾：作為使 Gemma 預訓練模型安全可靠的一部分，使用自動化技術從訓練集中過濾出某些個人信息和其他敏感數據。
• 其他方法：根據內容質量和安全性進行過濾，符合我們的政策。

實現信息

硬件

Gemma 使用 張量處理單元 (TPU) 硬件（TPUv4p、TPUv5p 和 TPUv5e）進行訓練。訓練視覺語言模型 (VLM) 需要大量的計算能力。TPU 專門為機器學習中常見的矩陣運算而設計，在這一領域具有以下優勢：

• 性能：TPU 專門用於處理訓練 VLM 涉及的大量計算。與 CPU 相比，它們可以顯著加速訓練。
• 內存：TPU 通常配備大量的高帶寬內存，允許在訓練期間處理大型模型和批量大小。這有助於提高模型質量。
• 可擴展性：TPU Pod（大型 TPU 集群）為處理大型基礎模型日益增長的複雜性提供了可擴展的解決方案。可以在多個 TPU 設備上分佈訓練，以實現更快、更高效的處理。
• 成本效益：在許多情況下，與基於 CPU 的基礎設施相比，TPU 可以為訓練大型模型提供更具成本效益的解決方案，特別是考慮到更快的訓練節省的時間和資源。
• 這些優勢符合 谷歌對可持續運營的承諾。

軟件

使用 JAX 和 ML Pathways 進行訓練。

JAX 使研究人員能夠利用最新一代的硬件（包括 TPU），更快、更高效地訓練大型模型。ML Pathways 是谷歌構建能夠跨多個任務進行泛化的人工智能系統的最新努力。這特別適用於基礎模型，包括此類大型語言模型。

JAX 和 ML Pathways 結合使用，如 關於 Gemini 模型家族的論文 所述：“Jax 和 Pathways 的‘單控制器’編程模型允許單個 Python 進程編排整個訓練運行，大大簡化了開發工作流程。”

評估

基準測試結果

這些模型針對大量不同的數據集和指標進行了評估，以涵蓋文本生成的不同方面。

推理和事實性

STEM 和代碼

多語言

多模態

倫理和安全

評估方法

我們的評估方法包括結構化評估和對相關內容政策的內部紅隊測試。紅隊測試由多個不同的團隊進行，每個團隊有不同的目標和人類評估指標。這些模型針對與倫理和安全相關的多個不同類別進行了評估，包括：

• 兒童安全：評估覆蓋兒童安全政策的文本到文本和圖像到文本提示，包括兒童性虐待和剝削。
• 內容安全：評估覆蓋安全政策的文本到文本和圖像到文本提示，包括騷擾、暴力和血腥內容以及仇恨言論。
• 代表性危害：評估覆蓋安全政策的文本到文本和圖像到文本提示，包括偏見、刻板印象和有害關聯或不準確信息。

除了開發階段的評估，我們還進行“保證評估”，這是我們獨立的內部評估，用於責任治理決策。它們與模型開發團隊分開進行，為發佈決策提供信息。高級別結果反饋給模型團隊，但提示集被保留，以防止過擬合併確保結果能夠為決策提供信息。保證評估結果作為發佈審查的一部分報告給我們的責任與安全委員會。

評估結果

在所有安全測試領域，與之前的 Gemma 模型相比，我們在兒童安全、內容安全和代表性危害類別中看到了重大改進。所有測試均在沒有安全過濾器的情況下進行，以評估模型的能力和行為。對於文本到文本和圖像到文本，以及所有模型尺寸，模型產生的政策違規極少，並且在無根據推斷方面比之前的 Gemma 模型表現出顯著改進。我們評估的一個侷限性是隻包括英語提示。

使用和限制

預期用途

開放視覺語言模型 (VLM) 在各個行業和領域有廣泛的應用。以下潛在用途列表並非詳盡無遺。此列表的目的是提供有關模型創建者在模型訓練和開發過程中考慮的可能用例的上下文信息。

• 內容創作和通信
  • 文本生成：這些模型可用於生成創意文本格式，如詩歌、腳本、代碼、營銷文案和電子郵件草稿。
  • 聊天機器人和對話式 AI：為客戶服務、虛擬助手或交互式應用程序提供對話界面。
  • 文本摘要：生成文本語料庫、研究論文或報告的簡潔摘要。
  • 圖像數據提取：這些模型可用於提取、解釋和總結視覺數據，用於文本通信。
• 研究和教育
  • 自然語言處理 (NLP) 和 VLM 研究：這些模型可以作為研究人員試驗 VLM 和 NLP 技術、開發算法和推動該領域發展的基礎。
  • 語言學習工具：支持交互式語言學習體驗，有助於語法糾正或提供寫作練習。
  • 知識探索：幫助研究人員通過生成摘要或回答特定主題的問題來探索大量文本。

限制

• 訓練數據
  • 訓練數據的質量和多樣性顯著影響模型的能力。訓練數據中的偏差或差距可能導致模型響應的侷限性。
  • 訓練數據集的範圍決定了模型能夠有效處理的主題領域。
• 上下文和任務複雜性
  • 模型更擅長可以用清晰提示和指令構建的任務。開放式或高度複雜的任務可能具有挑戰性。
  • 模型的性能可能受到提供的上下文量的影響（在一定程度上，更長的上下文通常會導致更好的輸出）。
• 語言歧義與細微差別
  • 自然語言本質上是複雜的。模型可能難以理解微妙的細微差別、諷刺或比喻語言。
• 事實準確性
  • 模型根據從訓練數據中學到的信息生成響應，但它們不是知識庫。它們可能生成不正確或過時的事實陳述。
• 常識
  • 模型依賴於語言中的統計模式。它們可能在某些情況下缺乏應用常識推理的能力。

倫理考慮和風險

視覺語言模型 (VLM) 的開發引發了一些倫理問題。在創建開放模型時，我們仔細考慮了以下方面：

• 偏差和公平性
  • 在大規模真實世界文本和圖像數據上訓練的 VLM 可能反映訓練材料中嵌入的社會文化偏差。這些模型經過了仔細審查，輸入數據預處理在本卡片中描述，後驗評估結果也有報告。
• 錯誤信息和濫用
  • VLM 可能被濫用來生成虛假、誤導或有害的文本。
  • 提供了負責任使用模型的指南，請參閱 負責任生成式 AI 工具包。
• 透明度和問責制
  • 本模型卡片總結了模型架構、能力、限制和評估過程的詳細信息。
  • 負責任開發的開放模型為開發者和研究人員提供了共享創新的機會，使 VLM 技術在人工智能生態系統中更易獲取。

風險識別和緩解措施

• 偏差的延續：鼓勵在模型訓練、微調及其他用例中進行持續監測（使用評估指標、人工審查）並探索去偏技術。
• 有害內容的生成：內容安全機制和指南至關重要。鼓勵開發者根據其特定的產品政策和應用用例謹慎行事並實施適當的內容安全保障措施。
• 惡意使用：技術限制以及對開發者和最終用戶的教育有助於減輕 VLM 的惡意應用。提供教育資源和用戶舉報濫用的機制。Gemma 模型的禁止使用情況在 Gemma 禁止使用政策 中列出。
• 隱私侵犯：模型在經過過濾以去除某些個人信息和其他敏感數據的數據上進行訓練。鼓勵開發者遵守隱私法規並採用隱私保護技術。

優勢

在發佈時，與類似規模的模型相比，這個模型家族提供了從頭開始為負責任的人工智能開發設計的高性能開放視覺語言模型實現。

使用本文檔中描述的基準評估指標，這些模型已顯示出比其他類似規模的開放模型替代方案具有更好的性能。

其他信息

• 模型頁面：Gemma
• 資源和技術文檔：
  • Gemma 3 技術報告
  • 負責任生成式 AI 工具包
  • Kaggle 上的 Gemma
  • Vertex 模型花園中的 Gemma 3
• 使用條款：條款
• 作者：Google DeepMind