超越Newton-Schulz:新型极分解算法实现更快的矩阵逆平方根计算

这项研究提出了一种利用极分解来计算矩阵逆平方根的新型算法,在性能上超越了目前广泛使用的Newton-Schulz迭代法。矩阵逆平方根是深度学习中二阶优化器(如K-FAC、Shampoo)及数据白化的核心算子。新方法通过改进底层数学逻辑,显著提升了计算速度和数值稳定性。这一突破对于降低大模型训练成本、提升AI芯片在处理复杂线性运算时的能效比具有重要价值。

原文链接:Hacker News

C code80.ai · AI 编码 API 聚合 Claude / GPT 多模型统一接入,稳定不限速,按量计费,几行配置接入 Claude Code。 了解一下 ›

抢沙发

评论前必须登录!

立即登录   注册