ExTraSS: a Domain Decomposed 3D NLTE Radiative Transfer spectral synthesis code for nebular phase transients

ExTraSS 技术摘要：一种域分解的三维非局部热动平衡辐射转移光谱合成代码

问题陈述
模拟超新星（SNe）的星云相构成了重大的计算挑战。在此阶段，抛射物密度已降至足够低，使得结构在很大程度上变为光学薄，意味着整个体积都参与辐射。然而，精确建模需要非局部热动平衡（NLTE）物理以考虑复杂的微观物理过程，同时辐射转移（RT）效应不可忽略，特别是对于富氢的核心坍缩超新星。此外，超新星的不对称形态 necessitates 三维输入模型。

三维 NLTE 辐射转移的主要瓶颈在于光激发速率的生成与存储。对于包含约 $10^5$ 个网格单元的网格，存储数百万次跃迁的这些速率可能需要超过 1 TB 的数据，这超出了现代超级计算机上单个计算节点的内存容量。以往的三维 NLTE 工作要么局限于一维，要么使用光学薄近似，要么采用忽略全局辐射场效应的“原位”处理。因此，亟需一种能够执行完整三维 NLTE 辐射转移、并在辐射场与原子能级布居之间实现全局耦合的代码。

方法论
作者提出了 ExTraSS（EXplosive TRAnsient Spectral Simulator，爆炸瞬态光谱模拟器），这是一段经过升级以执行完整三维 NLTE 辐射转移的代码。该代码以“快照”模式运行，独立于时间追踪光子传输，适用于光深较低的晚期阶段。

• 网格与几何：ExTraSS 利用球坐标网格，允许在内部较慢运动的抛射物中实现高分辨率，并在外部半径处使用更大的网格单元。辐射场的角变化使用 $N_{\theta, VA} \times N_{\phi, VA}$ 个观测者（默认为 20 $\times$ 20）进行离散化。
• $\gamma$ 射线传输：该代码使用蒙特卡洛方法结合康普顿散射来追踪放射性衰变（主要是 $^{56}$Ni 和 $^{56}$Co）。正电子被假定为被局部捕获。$\gamma$ 射线的能量沉积通过 Spencer-Fano 形式驱动非热加热、电离和激发。
• NLTE 求解器：求解器迭代确定局部温度和能级布居。它耦合热物理与非热物理，平衡加热（来自 $\gamma$ 射线和光电离）与冷却（束缚 - 束缚、自由 - 束缚和双光子发射）。求解器采用具有自适应阻尼的牛顿 - 拉夫逊（Newton-Raphson）方案来收敛能级布居。
• 辐射转移（RT）：该代码实施完整的射线追踪技术以计算全局辐射场。射线从网格单元中心向所有观测角度发射。代码计算光电离的光深（对激发态使用类氢近似，对基态使用 Verner 等人的数据）和光激发（对谱线相互作用使用 Sobolev 近似）。
• 域分解：为了克服内存限制，ExTraSS 采用域分解算法。计算域被分割为基于方位角 $\phi$ 的切片（方位扇区），每个节点处理特定的扇区。
  • 通信架构：每个节点指定一个“管理”核心来处理节点间通信，而剩余的“工作”核心执行射线追踪和局部计算。
  • 优化：系统使用非阻塞 MPI 通信以最小化工作核心的空闲时间。数据以束（射线）的形式传输，而非逐个传输。管理核心协调域之间的射线传输，确保工作核心将最大时间用于计算而非等待数据。

主要贡献

1. 首个超新星三维 NLTE 辐射转移：据作者所知，这项工作代表了首次专门为超新星辐射转移代码实施的域分解，使得此前因内存限制而无法实现的完整三维 NLTE 计算成为可能。
2. 内存优化：论文详细阐述了共享内存、"锯齿数组"（Jagged Arrays，仅分配现有物种/能级的内存）以及域分解方案的实现，这些方案使内存使用量与所用节点数量成反比缩放。
3. 算法改进：作者优化了对计算昂贵的 Spencer-Fano 子程序调用频率（采用 4+10 方案：在第 1、4 次迭代调用，之后每 10 次调用一次），并改进了牛顿 - 拉夫逊能级布居求解器的阻尼方案，以平衡收敛速度与稳定性。

结果与验证
作者通过多个基准测试验证了 ExTraSS：

• 收敛性与效率：在 Dardel 超级计算机（使用 128 核节点）上的测试表明，当负载均衡优化时，该代码在辐射转移模块实现了约 0.8 的计算效率。全局收敛（总光度与 $\gamma$ 射线沉积之比）在约 17 次迭代内达到，通常稳定在 0.96–1.0 的比率附近。
• 与 SUMO（一维）的比较：ExTraSS 与一维代码 SUMO 使用"一维转三维”模型（J18）进行了比较。
  • 物理条件：温度和电子分数吻合良好，尽管由于中性分数较低，ExTraSS 在内部抛射物中显示出略高的温度。
  • 速率：光电离和光激发速率总体兼容，尽管 ExTraSS 在最内部区域显示出略高的速率。
  • 光谱：合成光谱显示出良好的一致性，Fe II 发射和 H$\alpha$ 强度的微小差异归因于所包含原子能级数量的不同（ExTraSS 每个离子限制为 100 个能级，而 SUMO 为数百个）。
• 与多代码基准测试（Blondin 等人 2022）的比较：ExTraSS 与 ARTIS-nebular、CMFGEN 和 SUMO 一起针对 'toy06' 模型进行了测试。
  • 物理条件：与某些代码相比，ExTraSS 计算出的内部抛射物温度略低，钴电离分数也略低，但结果总体上与集合一致。
  • 光谱：光谱输出与 SUMO 及其他代码吻合良好，特别是在 5100–6400 Å 和 >8800 Å 范围内。在 4600–5000 Å 附近的 Fe 发射峰处观察到差异，ExTraSS 预测的光度较低，这可能是由于能级布居截断或电离差异所致。

意义
该论文声称 ExTraSS 现已成为一个功能完备的、经过域分解的三维 NLTE 辐射转移光谱合成代码。其意义在于能够处理三维 NLTE 物理的巨大内存需求，为在星云相中高精度建模不对称超新星爆炸开辟了道路。虽然目前应用于核心坍缩超新星模型（特别是 II 型），但作者指出该代码旨在适用于各种输入模型和爆炸瞬变源，前提是未来的更新中纳入额外的物理成分（如时间依赖性、尘埃或分子）。这项工作为通过详细的光谱特征更好地约束爆炸性质、抛射物质量和成分奠定了基础。