跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

Jet-Long: 基於動態雙焦點RoPE的高效長上下文擴充套件

文章摘要

Jet-Long 提出了一種無需微調的零樣本方法,透過動態雙焦點RoPE擴充套件大語言模型的上下文視窗,根據序列長度自適應調整縮放因子,在多個基準測試中實現了高效率和強效能。

來源arXiv Machine Learning作者: Haozhan Tang, Zerui Wang, Yuxian Gu, Song Han, Han Cai
Jet-Long: 基於動態雙焦點RoPE的高效長上下文擴充套件
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

大語言模型(LLM)在檢索增強生成、倉庫級編碼和智慧體工作流等長上下文應用中部署日益廣泛。這些場景的累積推理和工具痕跡常使輸入長度遠超預訓練視窗,使得零樣本上下文擴充套件成為開放權重模型的主要部署路徑。然而,現有零樣本方法通常預先固定單個縮放因子:激進的因子會犧牲短上下文保真度,而保守的因子在長上下文時失效。這種權衡限制了模型在實際應用中的靈活性和效能。

為了解決這一問題,研究團隊提出了Jet-Long,一種無需微調的零樣本方法。該方法透過配對一個區域性RoPE忠實視窗和一個長程視窗,實現了動態雙焦點RoPE。其中,長程視窗的縮放因子根據當前序列長度自適應調整,從而在短輸入時精確恢復基模型效能,在長輸入時實現乾淨外推。這種設計避免了固定縮放因子的固有缺陷。

在推理效率方面,Jet-Long透過包含-排除注意力合併和即時RoPE校正旋轉,使得雙焦點構造幾乎不增加額外開銷。當融合為單個CuTe核心後,長上下文預填在NVIDIA H100 GPU上的吞吐量可達FlashAttention 2的1.39倍,接近僅支援Hopper架構的FlashAttention 4。單批次生成在所有上下文長度上的開銷不超過4%,這使得Jet-Long非常適用於實際部署場景。

實驗結果表明,在Qwen3-1.7B、4B和8B模型上,針對高達128K的上下文長度,Jet-Long在RULER基準測試中分別領先最強基線4.79、2.18和2.03個百分點。在HELMET-RAG基準上(該基準被HELMET認定為最有效的下游長上下文效能預測指標),Jet-Long取得了最佳整體準確率。此外,在PG-19語言建模基準上,Jet-Long達到了最低的困惑度。

Jet-Long還具備良好的泛化能力,能夠推廣到Jet-Nemotron等混合注意力架構,無需重新訓練即可進一步提升長上下文能力。該方法對超引數選擇具有魯棒性,便於實際部署。這些特性使得Jet-Long成為長上下文LLM部署中一個有前景的解決方案。

展開要點與分析

文章情報

工程師進階

要點

  • 現有零樣本上下文擴充套件方法使用固定縮放因子,在短上下文和長上下文之間存在權衡。
  • Jet-Long 使用動態雙焦點RoPE,包含區域性視窗和長程視窗,根據序列長度自動調整縮放因子。
  • 該方法推理效率高,開銷極低,在RULER、HELMET-RAG和PG-19上取得了最優結果。
  • Jet-Long 無需重訓練即可推廣到混合注意力架構,且對超引數選擇具有魯棒性。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。