用物理定律强化神经网络,让地震波模拟更准更快
Enforcing Reciprocity in Operator Learning for Seismic Wave Propagation
- 用注意力机制和交换运算强制波场满足互易原理
- 多源并行推理速度提升十倍,内存相当
- 适合需要高物理一致性地震模拟的研究者
精确高效的波场建模是地震结构与震源研究的基础。传统方法虽符合物理规律但计算成本高;数据驱动方法虽有潜力,却缺乏严格的物理一致性。互易性是波动传播中最基本的物理定律之一。本文提出基于Transformer的递归神经算子RENO,通过交叉注意力与交换对称操作,硬编码互易性约束,确保源与接收点互换后结果不变。该架构不仅提升物理一致性,还支持多源同时模拟,在真实多源场景下推理速度提升一个数量级,内存开销相近。实验验证了单力作用下粒子速度场的互易关系。该方法还可推广至体膨胀源的压力场及双曲型射线时间场,为编码更复杂互易关系铺路。
原文摘要 · Abstract (English)
Accurate and efficient wavefield modeling underpins seismic structure and source studies. Traditional methods comply with physical laws but are computationally intensive. Data-driven methods, while opening new avenues for advancement, have yet to incorporate strict physical consistency. The principle of reciprocity is one of the most fundamental physical laws in wave propagation. We introduce the Reciprocity-Enforced Neural Operator (RENO), a transformer-based architecture for modeling seismic wave propagation that hard-codes the reciprocity principle. The model leverages the cross-attention mechanism and commutative operations to guarantee invariance under swapping source and receiver positions. Beyond improved physical consistency, the proposed architecture supports simultaneous realizations for multiple sources. This yields an order-of-magnitude inference speedup at a similar memory footprint over a conventional neural operator on a realistic multi-source configuration. We demonstrate the functionality using the reciprocity relation for particle velocity fields under single forces. This architecture is also applicable to pressure fields under dilatational sources and travel-time fields governed by the eikonal equation, paving the way for encoding more complex reciprocity relations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。