Qwen2-7B

Qwen2-7B是基於Transformer架構的大語言模型,採用分組查詢注意力和雙塊注意力技術最佳化長文本處理能力。該模型在覆蓋30種語言的7萬億token數據集上完成預訓練,特彆強化代碼和數學領域的數據質量。通過監督式微調和直接偏好最佳化兩階段後訓練,模型可適配文本生成、代碼解釋等任務場景。支持通過LoRA等微調技術實現跨模態套用開發,已成為開發生態最活躍的中等規模開源語言模型之一。

基本介紹

  • 發布時間:2024年6月
  • 參數規模:7億參數
  • 模型架構:Transformer+GQA架構
  • 開源協定:Apache 2.0協定
  • 訓練數據量:超7萬億token
  • 上下文長度:131,072 tokens
模型架構,預訓練過程,指令微調技術,套用場景,微調實踐,開源生態,

模型架構

採用28層Transformer架構,Q/K/V隱層維度為3584,注意力頭數縮減至28個。引入滑動視窗注意力機制,視窗尺寸擴展至128K tokens實現長文本處理最佳化。使用SwiGLU激活函式和因果掩碼機制,模型維度提升至18,944增強表達能力。
分組查詢注意力(GQA)技術實現注意力頭的動態分組,在保持推理效率的同時降低顯存占用。通過YARN機制擴展模型的長文本處理上限,最長支持131,072 tokens的序列輸入。

預訓練過程

在涵蓋30種語言的數據集上完成訓練,其中中文占比40%、英文35%、其他語言25%。代碼數據占比提升至8%,數學領域數據通過合成最佳化策略提升質量。採用改進後的數據過濾算法,清洗掉低質量的網頁內容和重複樣本。
訓練過程採用BF16混合精度,批量大小設定為6144,使用AdamW最佳化器進行參數更新。通過動態掩碼策略提升模型的語言建模能力,預訓練周期持續90天。

指令微調技術

監督式微調(SFT)階段使用50萬條人工標註的指令數據,涵蓋問答、推理、創作等場景。直接偏好最佳化(DPO)階段採用自動化合成的偏好數據,通過拒絕採樣和執行反饋機制生成對比樣本。
微調支持LoRA技術實現參數高效適配,典型配置包括秩參數r=8、縮放因子alpha=32。在百度智慧型雲千帆平台提供可視化微調界面,支持學習率調度和混合精度訓練。

套用場景

  1. 代碼生成:支持Python、Java等主流程式語言的代碼補全與糾錯
  2. 跨模態任務:通過微調適配圖像描述、視頻理解等多媒體場景
  3. 工具調用:集成RAG檢索增強生成和AI Agent功能模組
  4. 數學推理:在GSM8K等數學基準測試中達到75.3%準確率

微調實踐

使用LLaMA-Factory框架可在單台A100伺服器完成微調,推薦配置包括:
  • 學習率範圍5e-05至1e-04
  • 批量大小設定為4並進行4步梯度累積
  • 採用Warmup策略和Cosine衰減調度
數據處理需構建instruction-input-output三元組,典型格式包含:
json
{
"instruction": "以甄嬛體風格回答問題",
"input": "今天的天氣怎么樣?",
"output": "今兒個這天兒倒是極好的..."
}
支持通過modelscope庫下載預訓練權重,檔案大小約15GB。

開源生態

提供三種部署方案:
  • Hugging Face Transformers原生支持
  • vLLM框架實現高吞吐推理
  • MLX框架適配蘋果晶片硬體
開源社區已形成完整工具鏈,包含:
  • 模型量化工具(支持q8_0.gguf格式)
  • 對話模板管理系統
  • 訓練監控可視化組件
截止2024年6月,在ModelScope平台累計下載量突破150萬次,衍生出32個官方認證的微調版本。

相關詞條

熱門詞條

聯絡我們