跳到主要內容
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相當或更優。

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