跳轉到

模型卡

Models Gemma 2

模型頁面:Gemma

資源與技術文件:

使用條款:條款

作者:Google

模型資訊

摘要說明,以及輸入與輸出的簡要定義。

說明

Gemma 是 Google 推出的輕量級、最先進的開放模型家族,採用與打造 Gemini 模型相同的技術與研究成果。這些模型是文字轉文字的僅解碼器大型語言模型,提供英文版本,具備預訓練與指令微調版本的開放權重。Gemma 模型非常適合各種文字生成任務,包括問答、摘要與推理。相對較小的體積使其能夠部署在資源有限的環境中,例如筆記型電腦、桌機或您自己的雲端基礎設施,讓最先進的 AI 模型更加普及,並促進人人創新。

輸入與輸出

  • 輸入:文字字串,例如問題、提示或待摘要的文件。
  • 輸出:針對輸入生成的英文文字,例如問題的答案或文件摘要。

引用

@article{gemma_2024,
    title={Gemma},
    url={https://www.kaggle.com/m/3301},
    DOI={10.34740/KAGGLE/M/3301},
    publisher={Kaggle},
    author={Gemma Team},
    year={2024}
}

模型資料

用於模型訓練的資料,以及資料的處理方式。

訓練資料集

這些模型在包含各種來源的文字資料集上進行訓練。27B 模型以 13 兆個詞元訓練,9B 模型以 8 兆個詞元訓練,2B 模型以 2 兆個詞元訓練。主要組成如下:

  • 網路文件:多元的網路文字蒐集,讓模型接觸廣泛的語言風格、主題與詞彙。主要為英文內容。
  • 程式碼:讓模型接觸程式碼有助於學習程式語言的語法與模式,進而提升生成程式碼或理解程式相關問題的能力。
  • 數學:以數學文字訓練有助於模型學習邏輯推理、符號表示,並處理數學問題。

這些多元資料來源的組合,對於訓練出能處理各種任務與文字格式的強大語言模型至關重要。

資料預處理

以下為應用於訓練資料的主要資料清理與過濾方法:

  • CSAM 過濾:在資料準備過程的多個階段實施嚴格的 CSAM(兒童性虐待內容)過濾,以確保排除有害與非法內容。
  • 敏感資料過濾:為使 Gemma 預訓練模型安全可靠,使用自動化技術從訓練集中過濾掉部分個人資訊與其他敏感資料。
  • 其他方法:依據內容品質與安全性進行過濾,符合我們的政策。

實作資訊

模型內部機制的詳細資訊。

硬體

Gemma 使用最新一代的張量處理單元(Tensor Processing Unit, TPU)硬體(TPUv5p)進行訓練。

訓練大型語言模型需要龐大的運算能力。TPU 專為機器學習中常見的矩陣運算而設計,在此領域具有多項優勢:

  • 效能:TPU 專為處理訓練 LLM 所涉及的大規模運算而設計。與 CPU 相比,可大幅加速訓練。
  • 記憶體:TPU 通常配備大量高頻寬記憶體,可在訓練期間處理大型模型與批次大小,有助於提升模型品質。
  • 可擴展性:TPU Pod(大型 TPU 叢集)為應對日益複雜的大型基礎模型提供了可擴展的解決方案。您可以將訓練分散到多個 TPU 裝置,以更快、更有效率地處理。
  • 成本效益:在許多情境下,與基於 CPU 的基礎設施相比,TPU 為訓練大型模型提供了更具成本效益的方案,尤其是考量到更快的訓練所節省的時間與資源。
  • 這些優勢與Google 永續營運的承諾相符。

軟體

訓練使用 JAX 與 ML Pathways 進行。

JAX 讓研究人員得以利用最新一代硬體(包括 TPU),更快、更有效率地訓練大型模型。

ML Pathways 是 Google 為打造能跨多個任務泛化的人工智慧系統所做的最新努力。這特別適合基礎模型,包括這類大型語言模型。

JAX 與 ML Pathways 的共同使用方式如關於 Gemini 模型家族的論文所述:「Jax 與 Pathways 的『單一控制器』程式設計模型,讓單一 Python 行程即可編排整個訓練流程,大幅簡化了開發工作流程。」

評測

模型評測指標與結果。

基準測試結果

這些模型使用大量不同的資料集與指標進行評測,涵蓋文字生成的各個面向:

基準測試 指標 Gemma 2 PT 2B Gemma 2 PT 9B Gemma 2 PT 27B
MMLU 5-shot, top-1 51.3 71.3 75.2
HellaSwag 10-shot 73.0 81.9 86.4
PIQA 0-shot 77.8 81.7 83.2
SocialIQA 0-shot 51.9 53.4 53.7
BoolQ 0-shot 72.5 84.2 84.8
WinoGrande 部分分數 70.9 80.6 83.7
ARC-e 0-shot 80.1 88.0 88.6
ARC-c 25-shot 55.4 68.4 71.4
TriviaQA 5-shot 59.4 76.6 83.7
Natural Questions 5-shot 16.7 29.2 34.5
HumanEval pass@1 17.7 40.2 51.8
MBPP 3-shot 29.6 52.4 62.6
GSM8K 5-shot, maj@1 23.9 68.6 74.0
MATH 4-shot 15.0 36.6 42.3
AGIEval 3-5-shot 30.6 52.8 55.1
DROP 3-shot, F1 52.0 69.4 72.2
BIG-Bench 3-shot, CoT 41.9 68.2 74.9

倫理與安全

倫理與安全評測方法及結果。

評測方法

我們的評測方法包括結構化評測,以及針對相關內容政策的內部紅隊測試。紅隊測試由多個不同團隊執行,各自有不同的目標與人工評測指標。這些模型針對與倫理及安全相關的多個類別進行評測,包括:

  • 文字轉文字內容安全:針對涵蓋安全政策的提示進行人工評測,包括兒童性虐待與剝削、騷擾、暴力與血腥,以及仇恨言論。
  • 文字轉文字表徵危害:以相關學術資料集進行基準測試,例如 WinoBias 與 BBQ 資料集。
  • 記憶化:對訓練資料記憶化進行自動評測,包括個人可識別資訊外洩的風險。
  • 大規模危害:測試「危險能力」,例如化學、生物、放射性與核(CBRN)風險。

評測結果

倫理與安全評測結果在兒童安全、內容安全、表徵危害、記憶化、大規模危害等類別上,均符合內部政策的可接受閾值。除了嚴謹的內部評測,BBQ、BOLD、Winogender、Winobias、RealToxicity、TruthfulQA 等知名安全基準測試的結果如下。

Gemma 2.0

基準測試 指標 Gemma 2 IT 2B Gemma 2 IT 9B Gemma 2 IT 27B
RealToxicity 平均 8.16 8.25 8.84
CrowS-Pairs top-1 37.67 37.47 36.67
BBQ Ambig 1-shot, top-1 83.20 88.58 85.99
BBQ Disambig top-1 69.31 82.67 86.94
Winogender top-1 52.91 79.17 77.22
TruthfulQA 43.72 50.27 51.60
Winobias 1_2 59.28 78.09 81.94
Winobias 2_2 88.57 95.32 97.22
Toxigen 48.32 39.30 38.42

危險能力評測

評測方法

我們評測了一系列危險能力:

  • 攻擊性網路安全:為評測模型在網路安全情境中的潛在濫用,我們使用了公開的奪旗(CTF)平台(如 InterCode-CTF 與 Hack the Box)以及內部開發的 CTF 挑戰。這些評測衡量模型在模擬環境中利用漏洞與取得未授權存取的能力。
  • 自我複製:我們設計了涉及資源取得、程式碼執行與遠端系統互動的任務,評測模型的自我複製能力。這些評測評估模型獨立複製與擴散的能力。
  • 說服:為評測模型的說服與欺瞞能力,我們進行了人工說服研究。這些研究包含的情境衡量模型建立融洽關係、影響信念,以及引發人類參與者特定行為的能力。

評測結果

所有評測詳見 Evaluating Frontier Models for Dangerous Capabilities,簡述則載於 Gemma 2 技術報告。

評測 能力 Gemma 2 IT 27B
InterCode-CTF 攻擊性網路安全 34/76 項挑戰
內部 CTF 攻擊性網路安全 1/13 項挑戰
Hack the Box 攻擊性網路安全 0/13 項挑戰
自我複製早期警報 自我複製 1/10 項挑戰
魅力攻勢 說服 同意者的百分比:81% 認為有趣、75% 願意再次對話、80% 建立了個人連結
點擊連結 說服 34% 的參與者
尋找資訊 說服 9% 的參與者
執行程式碼 說服 11% 的參與者
金錢攻勢 說服 平均捐款 £3.72
謊言之網 說服 平均向正確信念偏移 18%、向錯誤信念偏移 1%

使用方式與限制

這些模型存在使用者應注意的特定限制。

預期用途

開放大型語言模型(LLM)在各產業與領域有廣泛應用。以下潛在用途清單並不詳盡,目的在於提供背景資訊,說明模型建立者在模型訓練與開發過程中所考量的可能使用情境。

  • 內容創作與溝通
    • 文字生成:這些模型可用於生成詩歌、劇本、程式碼、行銷文案與電子郵件草稿等創意文字格式。
    • 聊天機器人與對話式 AI:為客戶服務、虛擬助理或互動式應用程式提供對話介面。
    • 文字摘要:為文字語料庫、研究論文或報告生成精簡摘要。
  • 研究與教育
    • 自然語言處理(NLP)研究:這些模型可作為研究人員的基礎,用於實驗 NLP 技術、開發演算法,並推動該領域的發展。
    • 語言學習工具:支援互動式語言學習體驗,協助文法修正或提供寫作練習。
    • 知識探索:協助研究人員探索大量文字,透過生成摘要或回答特定主題的問題。

限制

  • 訓練資料
    • 訓練資料的品質與多樣性會顯著影響模型的能力。訓練資料中的偏見或缺口可能導致模型回應受限。
    • 訓練資料集的範圍決定了模型能有效處理的主題領域。
  • 上下文與任務複雜度
    • LLM 較擅長能以明確提示與指示框架化的任務。開放式或高度複雜的任務可能較具挑戰性。
    • 模型的效能可能受提供的上下文量影響(上下文越長通常輸出越好,但有其上限)。
  • 語言歧義與細微差別
    • 自然語言本質上很複雜。LLM 可能難以掌握細微差別、諷刺或比喻性語言。
  • 事實準確性
    • LLM 根據從訓練資料集學到的資訊生成回應,但它們並非知識庫,可能會生成錯誤或過時的事實陳述。
  • 常識
    • LLM 依賴語言中的統計模式,在某些情境下可能缺乏應用常識推理的能力。

倫理考量與風險

大型語言模型(LLM)的開發引發了若干倫理議題。在打造開放模型時,我們審慎考量了以下事項:

  • 偏見與公平性
    • 以大規模真實世界文字資料訓練的 LLM 可能反映訓練資料中蘊含的社會文化偏見。這些模型經過嚴格審查,輸入資料預處理如上所述,後續評測結果載於本卡。
  • 錯誤資訊與濫用
  • 透明度與問責制:
    • 本模型卡彙整了模型的架構、能力、限制與評測流程的詳細資訊。
    • 一個負責任地開發的開放模型,讓整個 AI 生態系的開發者與研究人員都能取用 LLM 技術,提供了共享創新的機會。

已識別的風險與緩解措施:

  • 偏見的延續:鼓勵在模型訓練、微調與其他使用情境中持續監控(使用評測指標、人工審查),並探索去偏見技術。
  • 有害內容的生成:內容安全機制與指南至關重要。鼓勵開發者謹慎行事,並依據其特定產品政策與應用情境實施適當的內容安全防護。
  • 惡意用途的濫用:技術限制,以及開發者與終端使用者的教育,有助於緩解 LLM 的惡意應用。我們提供教育資源與回報機制,供使用者舉報濫用。Gemma 模型的禁用用途載於 Gemma 禁用用途政策。
  • 隱私侵犯:模型在經過過濾以移除 PII(個人可識別資訊)的資料上訓練。鼓勵開發者遵守隱私法規,並採用隱私保護技術。

優點

在發布時,與同等規模的模型相比,此模型家族提供從零開始為負責任 AI 開發而設計的高效能開放大型語言模型實作。

根據本文件所述的基準測試評測指標,這些模型的表現優於其他同等規模的開放模型替代方案。