用偏微分方程基础模型加速激光聚变参数反演,数据少也能准。
PDE foundation model-accelerated inverse estimation of system parameters in inertial confinement fusion
- 基于预训练的PDE基础模型,微调后联合重建光谱图像与回归系统参数。
- 测试集上图像重建均方误差1.2e-3,参数估计决定系数达0.995。
- 低数据下仍表现优异,预训练权重显著提升样本效率,适合数据稀缺场景。
偏微分方程(PDE)基础模型通常在大规模、多样化的PDE数据集上预训练,并可在少量任务特定数据下适配。然而,大多数下游评估聚焦于前向问题,如自回归滚动预测。本文研究惯性约束聚变(ICF)中的一个逆问题:从多模态、快照式观测中估计系统参数(输入)。利用开源的JAG基准数据集(每模拟提供高光谱X射线图像和标量可观测值),我们对PDE基础模型进行微调,并训练一个轻量级任务专用头,联合重建高光谱图像并回归系统参数。微调后的模型在测试集上实现高精度的高光谱图像重建(测试均方误差1.2e-3)和强参数估计性能(决定系数最高达R²=0.995)。数据缩放实验(使用训练集的5%–100%)显示,随着训练数据增加,重建与回归损失持续下降,尤其在低数据条件下边际增益最大。此外,从预训练的MORPH权重微调,优于从零开始训练相同架构,证明基础模型初始化能显著提升数据受限的ICF逆问题中的样本效率。
原文摘要 · Abstract (English)
PDE foundation models are typically pretrained on large, diverse corpora of PDE datasets and can be adapted to new settings with limited task-specific data. However, most downstream evaluations focus on forward problems, such as autoregressive rollout prediction. In this work, we study an inverse problem in inertial confinement fusion (ICF): estimating system parameters (inputs) from multi-modal, snapshot-style observations (outputs). Using the open JAG benchmark, which provides hyperspectral X-ray images and scalar observables per simulation, we finetune the PDE foundation model and train a lightweight task-specific head to jointly reconstruct hyperspectral images and regress system parameters. The fine-tuned model achieves accurate hyperspectral reconstruction (test MSE 1.2e-3) and strong parameter-estimation performance (up to R^2=0.995). Data-scaling experiments (5%-100% of the training set) show consistent improvements in both reconstruction and regression losses as the amount of training data increases, with the largest marginal gains in the low-data regime. Finally, finetuning from pretrained MORPH weights outperforms training the same architecture from scratch, demonstrating that foundation-model initialization improves sample efficiency for data-limited inverse problems in ICF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。