arXiv cs.LG· Alex Guerra, Joshua Aduol, Yaqian Tang, Daniel Williams, Ryan P. Adams, Peter Orbanz·· 22 天前AI 评分42
无坐标自动微分
Coordinate-Free Automatic Differentiation
AI 导读
研究提出一种无基底导数算子,可直接用现有自动微分(AD)原语表示旋转、体积形式和曲率等几何微分操作。该方法将几何信息转化为代数数据,能与 JAX 的批处理、编译及硬件加速无缝组合。实验表明,该算子可在多种流形算子间复用,并成功应用于时空反演问题管线及旋量(rotor)子群优化。
来源:arXiv cs.LG · arxiv.org