Nov, 2023

可微分和加速的球面谐波和Wigner变换

TL;DR在该研究中,我们开发了新颖的算法结构,用于在球面数据和旋转群中加速和可微分计算广义傅立叶变换,包括球谐变换和Wigner变换。通过紧密耦合可分离的球面变换,结合混合自动和手动微分方法计算梯度,我们在JAX可微分编程框架中实现了这些算法。我们的方法在不同的球面采样方法上进行了支持,并具有出色的计算准确性和可伸缩性,达到了前所未有的线性时间复杂度。