研究人员推出了一种新颖的优化技术 Rank-1 iKFAD (R-iKFAD),旨在提高 Transformer 预训练的内存效率。该方法通过将 iKFAD 优化器的完整摩擦张量替换为秩为 1 的外积分解,将每层的内存占用从 O(mn) 显著降低到 O(m+n)。在 GPT2-Nano 和 DistilBERT 等各种模型上的实验表明,R-iKFAD 在性能上与原始 iKFAD 相当,同时将优化器的内存需求几乎减半。 AI
影响 这种内存高效的优化技术可以使在现有硬件上训练更大的 Transformer 模型成为可能,从而可能加速研究和开发。
排序理由 该集群包含一篇详细介绍 Transformer 模型新优化技术的论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →