2025-11-30T22:04:19.693191

Emulating Radiative Transfer in Astrophysical Environments

Rost, Branca, Buck

Radiative transfer is a fundamental process in astrophysics, essential for both interpreting observations and modeling thermal and dynamical feedback in simulations via ionizing radiation and photon pressure. However, numerically solving the underlying radiative transfer equation is computationally intensive due to the complex interaction of light with matter and the disparity between the speed of light and the typical gas velocities in astrophysical environments, making it particularly expensive to include the effects of on-the-fly radiation in hydrodynamic simulations. This motivates the development of surrogate models that can significantly accelerate radiative transfer calculations while preserving high accuracy. We present a surrogate model based on a Fourier Neural Operator architecture combined with U-Nets. Our model approximates three-dimensional, monochromatic radiative transfer in time-dependent regimes, in absorption-emission approximation, achieving speedups of more than 2 orders of magnitude while maintaining an average relative error below 3%, demonstrating our approach's potential to be integrated into state-of-the-art hydrodynamic simulations.

academic

Emulating Radiative Transfer in Astrophysical Environments

基本信息

论文ID: 2511.08219
标题: Emulating Radiative Transfer in Astrophysical Environments
作者: Rune Rost, Lorenzo Branca, Tobias Buck (海德堡大学跨学科科学计算中心)
分类: astro-ph.IM (天文仪器方法), astro-ph.GA (星系天体物理), cs.LG (机器学习)
发表会议: 1st Workshop on Differentiable Systems and Scientific Machine Learning EurIPS 2025
论文链接: https://arxiv.org/abs/2511.08219

其中 $I_\nu$ 是光谱辐射强度,具有高维依赖性(时间 $t$ 、空间位置 $x$ 、方向 $\omega$ 、频率 $\nu$ ),导致计算复杂度极高。

2. 问题的重要性

观测诊断: 连接物理模型与观测数据的关键桥梁
热动力学反馈: 通过光致加热影响气体热力学
辐射压力: 影响气体动力学演化
星系形成: 在巨分子云等恒星形成区域的辐射效应研究中至关重要

3. 现有方法的局限性

蒙特卡洛方法: 计算成本高
光线追踪: 内存需求随源数量和网格数线性增长 $O(N_{sources}N_{cells})$
矩方法: 光速远大于气体速度( $c >> v_{gas}$ )导致需要极小时间步长
传统深度学习方法: 在网格离散化和参数设置间泛化能力差,在高维PDE问题上面临稳定性和精度挑战
现有工作空白: 据作者所知,尚无针对时间依赖辐射传输的模拟器

4. 研究动机

开发基于神经算子的数据驱动代理模型,能够:

在无限维函数空间上操作,提升跨网格离散化的泛化能力
显著加速计算同时保持高精度
作为即插即用替代方案集成到流体动力学模拟中

核心贡献

首个时间依赖辐射传输模拟器: 据作者所知,这是首个针对时间演化辐射传输的神经算子模拟器
创新架构设计: 提出U-FNO架构,将傅里叶神经算子(FNO)与U-Net结合,兼顾全局依赖性捕获和精细尺度特征建模
显著性能提升:
- 时间演化预测: 实现~600×加速,平均相对误差2.9%
- 稳态预测: 实现~6750×加速,相对误差2.6%
- 内存成本与源数量无关,解决光线追踪的关键瓶颈
多频扩展方案: 提供了向多频辐射传输扩展的明确路径,符合现代流体动力学模拟需求
开源实现: 提供完整可复现代码 (https://github.com/RuneRost/Astro-RT.git)

方法详解

任务定义

输入:

$a(x) = k_{\nu,a}\rho(x)$ : 吸收场(与密度相关)
$j(x) = j_\nu\rho(x)$ : 发射场(辐射源分布)
$I_{\nu,t}(x)$ : 当前时刻辐射强度

输出:

$I_{\nu,t+1}(x)$ : 下一时刻辐射强度

简化假设:

无散射regime (符合实际on-the-fly计算)
暂时忽略角度依赖(未来工作)
单色辐射(可扩展到多频)
三维空间 $64^3$ 网格

模型架构

1. U-FNO整体架构

输入 [j(x), a(x), I_{ν,t}(x)] 
  ↓
提升层 P (映射到高维潜空间)
  ↓
U-Fourier层 × N (N=6)
  ↓
投影层 Q (映射到输出维度)
  ↓
输出 I_{ν,t+1}(x)

2. U-Fourier层结构

每个U-Fourier层包含三个并行/串行组件:

a) 卷积积分算子(通过傅里叶变换):

FFT变换到频域: $\mathcal{F}$
可学习权重乘法: $R$
逆FFT变换回空间域: $\mathcal{F}^{-1}$
模式截断: 仅保留前K=4个模式(降低复杂度,鼓励平滑预测)

b) 仿射变换: $(W, b)$

c) U-Net模块:

编码器-解码器结构
对称的下采样和上采样路径
跳跃连接保留细节信息
小卷积核(2×2×2或3×3×3)精确定位精细尺度特征
宽度: 16(稳态)或32(时间演化)

3. 关键设计原理

FNO优势:

在无限维函数空间学习映射
捕获全局依赖性
理论上具有网格不变性

U-Net优势:

处理尖锐梯度和不连续性
小卷积核实现精确局部特征定位
跳跃连接保留多尺度信息

组合必要性: 纯FNO模型相对误差高达60%,加入U-Net后降至2.9%

技术创新点

1. 架构创新

U-Fourier层: 首次将U-Net集成到每个傅里叶层中,而非简单堆叠
多尺度特征融合: 全局傅里叶特征 + 局部U-Net特征

2. 递归预测策略

训练: 使用连续时间步对 $(I_{\nu,t}, I_{\nu,t+1})$
推理: 递归应用,将预测作为下一步输入
实现完整时间演化模拟

3. 数据预处理

对数变换: 处理多数量级跨度的数据
Min-max归一化: 缩放到0,1区间
分别归一化: 吸收、发射、强度场独立处理

4. 损失函数设计

$L = \frac{\sqrt{\sum_{i=1}^N(\hat{f}_i - f_i)^2}}{\sqrt{\sum_{i=1}^N(f_i)^2}} + \lambda\frac{\sqrt{\sum_{i=1}^N(\nabla\hat{f}_i - \nabla f_i)^2}}{\sqrt{\sum_{i=1}^N(\nabla f_i)^2}}$

相对误差损失(避免绝对误差偏向高值区域)
空间梯度项( $\lambda=0.5$ )鼓励尖锐特征重建
解决模糊预测问题

生成方法: 使用jf1uids流体动力学代码生成湍流周期盒
网格: $64^3$ 空间分辨率
多样性策略:
- 变化湍流随机种子
- 变化速度扰动幅度
- 变化湍流动能功率谱斜率
- 运行至湍流谱达到稳定平衡
物理场景: 模拟巨分子云(恒星形成区)条件
辐射源: 放置在密度场前1.5%的高密度区域
时间步: 每个样本10个时间步,从 $I_{\nu,0}(x)=0$ 开始
数值求解: 使用JAX实现的光线追踪算法
样本数: 400对 $(a(x), j(x))$ → 展开为3600个训练样本
划分: 训练70%(2520), 验证10%(360), 测试20%(720)

稳态数据集

密度建模: 对数正态随机场(模拟湍流恒星形成区)
发射场: 高斯分布,中心位于密度前1%的像素(模拟恒星发射)
吸收系数: $k_{\nu,a}=1$ (简化)
样本数: 1000个
划分: 训练70%(700), 验证10%(100), 测试20%(200)

评价指标

像素级平均相对误差: $\text{Relative Error} = \frac{1}{N}\sum_{i=1}^N\frac{|\hat{I}_i - I_i|}{|I_i|}$ (排除 $I_i=0$ 的像素)
加速比: 模型推理时间 vs 数值求解器时间
视觉评估: 通过不同截面的残差图