研究人员开发了一种名为分层全局注意力(HGA)的新方法,可在有限显存下高效微调大型语言模型。该技术结合了分段反向传播和分层KV存储,将旧数据卸载到RAM或NVMe,同时将活动段保留在显存中进行区分。将其应用于Qwen3_8B模型,HGA在16 GB Quadro RTX 5000 GPU上成功训练了16,384个token,这是在同一硬件上使用标准密集训练方法不可能实现的壮举。HGA训练的适配器在较短的上下文长度下也表现出与密集训练适配器相当的性能,并有望随着上下文的增加而超越密集方法。 AI
影响 这项技术可以显著降低训练和微调大型语言模型的硬件门槛,使先进的人工智能更加普及。
排序理由 该集群描述了arXiv论文中提出的一种用于提高LLM微调效率的新方法。
- arXiv
- Hierarchical Global Attention
- Hugging Face
- NVM Express
- PG19
- QLoRA
- Quadro RTX 5000
- Qwen3_8B
- random-access memory
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →