Spectrally accurate, reverse-mode differentiable bounce-averaging algorithm and its applications

技术摘要：谱精确、反向模式可微弹跳平均算法及其应用

问题陈述
仿星器设计涉及一个复杂的优化问题，包含数百个自由度，旨在最小化新古典输运并改善约束。传统的优化方法依赖于有限差分技术来估计梯度。这些方法存在阶数较低的精度问题，并且当可优化参数数量庞大时，计算上变得难以实现，因为它们需要针对每个参数进行多次目标函数评估。此外，现有的用于简化动力学模型（如漂移动力学方程）的弹跳平均算法通常与自动微分（AD）不兼容，且利用较低阶的离散化，限制了其精度和效用。

方法论
作者提出了一种快速、具有谱（指数级）精度且自动可微的弹跳平均算法，并将其实现在 DESC（直接仿星器优化代码）仿星器优化套件中。该方法解决了以下计算挑战：

1. 弹跳积分的谱求积法： 核心算法通过沿粒子轨迹在弹跳点之间进行积分来计算弹跳平均量。为了处理由于弹跳点处平行速度 $|v_\parallel|$ 消失而产生的奇异性，作者采用了坐标变换结合高斯求积法。具体而言，他们使用了由正弦变换组成的 Gauss-Chebyshev (GC) 和 Gauss-Legendre (GL) 求积法。这种方法实现了谱收敛，显著优于标准的中间值法、辛普森法或双指数方案，特别是在积分函数接近奇异点时。
2. 逆 MHD 与谱插值： 该算法利用逆方法求解理想磁流体动力学（MHD）力平衡方程。等离子体边界和内部场使用通量坐标下的 Fourier-Zernike 级数进行参数化。为了避免优化过程中“移动网格”插值的计算瓶颈，作者预计算了磁力线映射的谱投影。他们利用快速傅里叶变换（FFT）和非均匀 FFT 在沿磁力线高效评估映射。
3. 反向模式微分： 一个关键组成部分是直接对场线坐标（求解 $\theta + \Lambda - \iota(\zeta + \omega) = \alpha$）关于优化参数进行微分。通过使用隐函数定理，作者直接推导出了切向法和伴随法。这使得微分的计算成本能够与可控参数的数量保持独立，这是相对于有限差分法的一个关键优势。
4. 目标函数： 主要应用展示了有效纹波 $\epsilon_{eff}$ 的最小化，这是一个在低碰撞率（$1/\nu$）机制下与新古典输运系数成比例的无量纲度量。目标函数包括长宽比、曲率、拉伸度、全向性（omnigeneity）和纹波项，并受理想 MHD 力平衡约束。

主要贡献

• 可微弹跳平均： 本文引入了第一个完全兼容反向模式自动微分的弹跳平均算法实现。这使得高效优化基于动力学的目标成为可能，其中梯度的计算成本与单个函数评估的成本相当，且与参数数量无关。
• 谱精度： 通过采用专门的求积规则和坐标变换，该算法实现了指数级精度，确保优化梯度反映的是真实的物理改进而非数值噪声。
• 有限 $\beta$ 优化： 该工作展示了针对降低新古典纹波输运的有限 $\beta$ 仿星器构型的优化。
• 集成至 DESC： 该算法被集成到 DESC 套件中，允许同时优化多个基于物理的目标（例如力平衡、几何约束和输运度量），而无需在每一步都重新求解 MHD 平衡。

结果
作者应用该算法优化了一个螺旋全向（OH）平衡态。从一个初始构型开始，优化显著降低了有效纹波 $\epsilon_{eff}$。该过程涉及最小化一个包含纹波度量的加权总和目标，在单块 GPU 上用不到两小时便完成了。结果显示：

• 减少了弹跳平均径向漂移的幅度。
• 与初始状态相比，优化后的平衡态具有更平滑的磁场结构。
• 在整个优化过程中成功维持了力平衡和几何约束（拉伸度、曲率）。
• 优化后的构型表现出降低的新古典输运，特别是在陷俘粒子损失占主导地位的低碰撞率机制下。

意义与主张
论文声称这项工作实现了“以指数精度进行快速优化以提升仿星器性能”。通过消除与动力学模型微分相关的计算障碍，作者断言，现在通过反向模式微分直接优化旨在降低新古典输运的仿星器构型已变得可行。这被强调为一项创新的能力，特别指出这是首次通过优化有限 $\beta$ 仿星器来直接降低新古典纹波输运。

作者强调，谱精度确保了优化梯度的可靠性，防止优化器被数值误差误导。他们进一步指出，虽然目前的工作侧重于新古典输运，但该框架可以扩展到其他目标，如能量粒子约束、湍流输运代理（如可用能量）以及第二绝热不变性的最大化。这项工作为可以将高保真动力学模型直接耦合到优化循环中的更复杂的仿星器逆向设计问题奠定了基础。