arXiv:2505.23190cs.LG2025-05

用神经网络高效求解辐射传输方程,精度高且无需网格。

DeepRTE: Pre-trained Attention-based Neural Network for Radiative Transfer

  • 基于注意力机制和物理信息嵌入,构建可解析辐射传输的神经网络。
  • 参数量少但精度高,计算效率优于传统方法和现有神经网络。
  • 预训练实现零样本推理,适用于多种边界条件的快速泛化。

本文提出一种新型神经网络方法 DeepRTE,用于求解稳态辐射传输方程(RTE)。RTE 是描述辐射在参与介质中传播的微分-积分方程,在中子输运、大气辐射传输、热传递和光学成像等领域有广泛应用。DeepRTE 框架通过推导 RTE 并融合数学先验知识,构建物理信息导向的网络结构,显著提升求解效率,优于传统数值方法及现有神经网络。其高精度与极低参数量得益于多头注意力等机制。此外,DeepRTE 为无网格神经算子框架,具备内在零样本能力,通过引入格林函数理论并以点源边界条件进行预训练实现。大量数值实验验证了该方法的有效性。

原文摘要 · Abstract (English)

In this paper, we propose a novel neural network approach, termed DeepRTE, to address the steady-state Radiative Transfer Equation (RTE). The RTE is a differential-integral equation that governs the propagation of radiation through a participating medium, with applications spanning diverse domains such as neutron transport, atmospheric radiative transfer, heat transfer, and optical imaging. Our DeepRTE framework demonstrates superior computational efficiency for solving the steady-state RTE, surpassing traditional methods and existing neural network approaches. This efficiency is achieved by embedding physical information through derivation of the RTE and mathematically-informed network architecture. Concurrently, DeepRTE achieves high accuracy with significantly fewer parameters, largely due to its incorporation of mechanisms such as multi-head attention. Furthermore, DeepRTE is a mesh-free neural operator framework with inherent zero-shot capability. This is achieved by incorporating Green's function theory and pre-training with delta-function inflow boundary conditions into both its architecture design and training data construction. The efficacy of the proposed approach is substantiated through comprehensive numerical experiments.

辐射传输神经算子注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。