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

CARE-LoRA: 用於記憶體高效LoRA的壓縮啟用重建

文章摘要

隨著大型預訓練模型規模增長,微調中的記憶體瓶頸日益突出。LoRA透過低秩適應減少引數,但啟用值儲存成為新的瓶頸。CARE-LoRA利用LoRA投影結構,將完整輸入啟用替換為低秩壓縮啟用,並計算輕量重建矩陣用於反向傳播,大幅降低記憶體佔用,同時保持或提升效能。

來源arXiv Machine Learning作者: Gengyu Zhang, Haiyin Ran, Zhengbao He, Yuhang Liu, Hanling Tian, Zhehao Huang, Xiaolin Huang
CARE-LoRA: 用於記憶體高效LoRA的壓縮啟用重建
回報錯誤

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

查看更正說明
直接讀正文

隨著大語言模型(如LLaMA、GPT等)引數規模持續增長,在有限GPU記憶體下進行微調變得越來越困難。低秩適應(LoRA)作為一種廣泛使用的引數高效微調方法,透過僅最佳化低秩適應矩陣大幅減少了可訓練引數數量,從而降低了記憶體壓力。然而,引數開銷降低後,用於反向傳播的啟用值儲存成為主要的記憶體瓶頸。針對這一問題,來自多所機構的研究人員提出了CARE-LoRA(壓縮啟用重建框架),相關論文於2026年7月11日提交至arXiv,程式碼已開源在GitHub上。

CARE-LoRA的核心創新在於資料感知的壓縮啟用重建。它利用LoRA分支固有的投影結構,將完整的輸入啟用替換為LoRA分支自然產生的低秩壓縮啟用。在前向傳播過程中,CARE-LoRA以極低的額外計算成本計算出一個輕量重建矩陣,該矩陣在反向傳播時用於重建梯度訊號,從而保持LoRA矩陣完全可訓練。這一方法無需修改LoRA的基本結構,即可顯著減少啟用記憶體佔用。論文篇幅為15頁,包含2幅圖和12張表格。

研究團隊在多種模型(包括BERT、RoBERTa、Llama)和多項下游任務(如文本分類、問答、自然語言推理)上進行了廣泛實驗。結果顯示,與標準LoRA相比,CARE-LoRA可減少約30%至50%的峰值記憶體使用,同時保持甚至提升模型效能。論文還提供了詳細的消融實驗和理論分析,證明了方法的有效性和魯棒性。

CARE-LoRA的提出具有重要意義。它直接針對LoRA微調中的啟用記憶體瓶頸,使得在相同硬體上可以訓練更大的模型或使用更大的批次。此外,該方法的資料感知特性使其能夠自動適應不同任務和資料分佈。程式碼已在https://github.com/fishandyu/CARE-LoRA公開,便於社群復現與進一步探索。

展開要點與分析

文章情報

工程師進階

要點

  • CARE-LoRA透過壓縮啟用重建解決LoRA微調中的啟用記憶體瓶頸。
  • 利用LoRA分支自然產生的低秩啟用替代完整啟用,減少儲存。
  • 在前向傳播中計算輕量重建矩陣,用於梯度恢復,不增加訓練負擔。
  • 實驗表明,CARE-LoRA在降低30-50%峰值記憶體的同時,效能與標準LoRA相當或更優。

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