实体 Rank-1 iKFAD

Rank-1 iKFAD

PulseAugur coverage of Rank-1 iKFAD — every cluster mentioning Rank-1 iKFAD across labs, papers, and developer communities, ranked by signal.

总计 · 30天

1

90 天内 1

发布 · 30天

0

90 天内 0

论文 · 30天

1

90 天内 1

层级分布 · 90 天

主题

情绪 · 30 天

1 天有情绪数据

最近 · 第 1/1 页 · 共 1 条

TOOL · CL_268230 · Sep 28 · 04:00

新的 Rank-1 iKFAD 优化器将 Transformer 预训练的内存减半

研究人员推出了一种新颖的优化技术 Rank-1 iKFAD (R-iKFAD)，旨在提高 Transformer 预训练的内存效率。该方法通过将 iKFAD 优化器的完整摩擦张量替换为秩为 1 的外积分解，将每层的内存占用从 O(mn) 显著降低到 O(m+n)。在 GPT2-Nano 和 DistilBERT 等各种模型上的实验表明，R-iKFAD 在性能上与原始 iKFAD 相当，同时将优化器的内存需求几乎减半。