Journal

Differentiable Neutron Transport

Xi Deng, Maosen Tang, Michael Czekanski, David Bindel, Steve Marschner

Cornell University

SIGGRAPH 2026Rendering
Differentiable RenderingMonte CarloNeutron TransportInverse ProblemGradient-Based Optimization
Paper

一句话总结

这篇论文把图形学里的可微渲染技术搬到核工程的中子输运问题上，构建了首个对几何与能量相关材料参数都完全可微的蒙特卡洛中子输运模拟器，从而用梯度下降来解屏蔽设计、传感器设计和敏感度分析等逆问题。

研究背景

  • 领域现状：中子输运在核反应堆（裂变与聚变）设计中至关重要，其控制方程与图形学里参与介质的体渲染方程共享同一套数学基础——都是描述粒子在空间中经发射、吸收、散射的概率传播。正向模拟已有 OpenMC、MCNP、Tripoli-4 等成熟工具。
  • 核心痛点：逆问题（如给定期望测量反推材料/几何配置）非常难。传统伴随方法在复杂几何、多响应场景下难以扩展；数据驱动的代理模型在高维参数空间里梯度不准、缺乏物理保证。此前不存在一个对复杂几何和材料参数都能给出高效、无偏梯度的自动微分中子输运模拟器。
  • 本文 idea：借鉴可微渲染，把中子输运路径积分改写成可微形式，通过对正向蒙特卡洛模拟做自动微分，一次采样即可同时累积对多个参数的梯度。作者指出中子输运既比光传输简单（中子主要穿透界面，没有反射/折射的复杂性），又更难（输出是高维空间通量场而非二维图像，材料参数还与能量相关，几何常用 CSG 表示而非网格）。

方法

整体框架：先把中子路径积分从传统”位置”参数化改写成”方向–距离”参数化，使几何依赖只出现在距离变量的积分上限里；再用一个巧妙的变量替换把上限”归一化”，从而把所有梯度项完全移进被积函数内部，彻底避免显式采样几何边界。对表面电流用这套重参数化，对体通量则额外改造 track-length 估计器使其可微。

flowchart LR
  A["中子源 + CSG 场景几何"] --> B["正向 MC 追踪路径"]
  B --> C["方向-距离参数化 + 距离重参数化"]
  C --> D["可微估计器: 表面电流 / 体通量"]
  D --> E["Dr.Jit 反向自动微分"]
  E --> F["对几何/材料参数的梯度"]
  F --> G["Adam 梯度优化 / 敏感度分析"]

关键设计：

  1. 距离子空间重参数化（几何梯度的核心）：中子路径写成 \(\mathbf{d}_k = \{x_0, \boldsymbol{\omega}, t, x_k\}\)，几何参数 \(\theta\) 只通过积分上限 \(r(\omega,\theta)\)（到最近界面的距离）进入积分。对单次散射做变量替换 \(t_0 = t_0' \, r_0(\omega_0,\theta)\) 后，积分区间变成固定的 \([0,1]\)、被积函数连续，于是梯度算子可以直接搬进积分内部，用自动微分就能得到无偏的几何梯度。多次散射时对每段距离都做同样替换，其雅可比行列式因采样的因果性恰好是下三角阵，行列式等于对角元乘积 \(\lvert J_k(\theta) \rvert = \prod_{i=0}^{k-2} r_i(\theta,\omega_i)\)，形式简洁。这属于”重参数化”流派，但和已有方法不同，作者的问题设定让所有梯度项都能移进被积函数，完全不需要处理边界或边缘。

  2. 分段连续被积函数的处理：一条射线会穿过多种材料（各自 cross section 不同），形成以界面位置分段的积分。作者对每段材料分别做分段重参数化 \(t_0 = r_j(\omega_0,\theta)\, t_0^j\)，保证每段的上限固定、被积函数连续，梯度照样可算。

  3. 可微 track-length 估计器（体通量梯度的核心）：体通量需要在 \(N\times N\times N\) 的体素网格上测量。传统 track-length 估计器让一条路径按其在每个体素内走过的长度贡献通量，比 collision 估计器方差低得多；但直接对它自动微分会退化成 collision 型梯度、噪声大。作者推导出一个”值上等价于 track-length、求导后却是低方差梯度估计器”的形式：给估计器乘上一个特制常数 \(\mathcal{C}_M = e^{-\sum_i \sigma_t^{D_i} r_i(\theta)}\)（其中 \(\sigma_t^{D}\) 是界面两侧材料总截面之差），并把该常数的一部分从计算图中 detach，使得 \(\frac{\partial}{\partial\theta}\mathbb{E}[I_a] = \mathbb{E}[\frac{\partial}{\partial\theta} I_a]\) 成立。

  4. 能量相关性与可微 CSG 工程方案：用离散能量组建模——每个中子任一时刻属于某个能量组，材料属性按组查询，散射时中子可留在原组或跳到别的组。几何用 CSG 表示，但不真正做布尔运算构造网格（那会引入不可微的拓扑变化），而是在光线求交时对叶子几何隐式地评估布尔关系，从而在保留 CSG 便利性的同时对网格参数保持可微。实现基于 Dr.Jit 后端与 Mitsuba 3 的求交例程。

实验结果

作者先做梯度验证，再展示三个逆问题应用。梯度验证部分把体通量梯度估计器与 collision 估计器在相同样本数下对比，以有限差分/解析解为参照计算相对均方误差（RelMSE），这是最能体现方法价值的”主实验”：

场景 / 估计器 Collision 通量 RelMSE↓ Track-length 通量 RelMSE↓ Collision 梯度 RelMSE↓ Track-length 梯度 RelMSE↓
2D（对解析解） 0.1951 0.0087 0.7669 0.006
3D dragon（对有限差分） — — 1.0115 0.1908
3D gear（对有限差分） — — 0.5177 0.0348

可微 track-length 估计器在相同样本数下，无论通量本身还是其梯度，方差都显著低于 collision 估计器。此外，作者用 OpenMC 做正向一致性校验：在球体与双环面两个多能量组场景里，本文自动微分得到的表面电流梯度与 OpenMC 有限差分（带误差棒）吻合。

三个应用：屏蔽优化——在体积约束下最小化传感器处的中子电流，优化结果把材料推向源附近、增大中子在屏蔽材料内的路径长度以增强衰减，符合物理直觉；传感器设计——复现并推广了已有的 Boron-10 探测器聚乙烯套筒实验，把套筒从圆柱松弛为可变网格后，找到比最优圆柱构型响应更高的新形状；敏感度分析——用局部梯度构造概率盒（p-box）来量化认知不确定性，效率远高于暴力蒙特卡洛。

亮点与局限

  • 亮点：
    • 首个对几何与能量相关材料参数都完全可微的蒙特卡洛中子输运框架，把可微渲染理论真正拓展到一个新物理领域，而非简单套用。
    • 重参数化让几何梯度完全避免边界采样，数学形式干净；可微 track-length 估计器巧妙解决了”值等价但导数低方差”的矛盾。
    • 隐式 CSG 方案兼顾了核工程惯用的几何表示与可微性，且只需光线求交查询、支持任意几何表示。
    • 与 OpenMC 有限差分和解析解都做了定量吻合验证，可信度高。
  • 局限：
    • track-length 估计器目前只用在路径的最后一段，\(k>1\) 的多次散射路径仍会退化为 collision 型梯度，且当束流方向接近平行于体素边界时可能出现奇异。
    • p-box 敏感度分析基于一阶梯度，在参数–响应强非线性/非单调（如几何变量）时会给出过于保守、长尾的界。
    • 复杂非对称几何（如仿星器）中，内表面位移收敛到同一点时会产生奇异，需要更鲁棒的几何表示。

延伸思考

这项工作是”图形学方法外溢到科学计算”的典型案例：可微渲染社区积累的路径空间微分、重参数化、track-length 等工具，几乎可以原样迁移到中子输运，反过来也逼出了新问题（高维通量场输出、能量相关材料、CSG 可微化），推动可微传输理论本身演化。作者已明确指出向伽马射线、alpha/beta 粒子乃至中性束注入等其他粒子输运推广的方向——这些都用蒙特卡洛粒子输运建模，天然适合同一套可微框架。值得追问的是：把 track-length 可微估计器推广到路径的非末段、或用连续基函数替代盒式滤波，能否系统性地降低多次散射路径的梯度方差，这可能是让该框架在真正复杂反应堆几何上稳定收敛的关键。