ECHO高效生成百万点微分方程轨迹,突破精度与规模瓶颈
Efficient Generative Transformer Operators For Million-Point PDEs
- 采用分层卷积编码解码结构,实现100倍时空压缩仍保精度
- 支持从稀疏输入生成高分辨率解,可处理复杂几何与高频动态
- 生成式建模避免长期误差累积,适用于正问题、逆问题等多任务
我们提出ECHO,一种用于生成百万点偏微分方程(PDE)轨迹的Transformer-Operator框架。现有神经算子在密集网格上存在可扩展性差、动态推演时误差累积以及任务专用设计等问题。ECHO通过三项关键创新解决:(i) 采用分层卷积编码-解码架构,在保持网格点保真度的同时实现100倍时空压缩;(ii) 引入训练与适配策略,可从稀疏输入网格生成高分辨率解;(iii) 采用生成建模范式,学习完整轨迹段,缓解长时序误差漂移。该训练策略将表征学习与下游任务监督解耦,使模型能应对轨迹生成、正/逆问题及插值等多任务。生成模型还支持条件与无条件生成。我们在包含复杂几何、高频动力学和长时程的多种PDE系统上展示了其在百万点模拟中的领先性能。
原文摘要 · Abstract (English)
We introduce ECHO, a transformer-operator framework for generating million-point PDE trajectories. While existing neural operators (NOs) have shown promise for solving partial differential equations, they remain limited in practice due to poor scalability on dense grids, error accumulation during dynamic unrolling, and task-specific design. ECHO addresses these challenges through three key innovations. (i) It employs a hierarchical convolutional encode-decode architecture that achieves a 100 $\times$ spatio-temporal compression while preserving fidelity on mesh points. (ii) It incorporates a training and adaptation strategy that enables high-resolution PDE solution generation from sparse input grids. (iii) It adopts a generative modeling paradigm that learns complete trajectory segments, mitigating long-horizon error drift. The training strategy decouples representation learning from downstream task supervision, allowing the model to tackle multiple tasks such as trajectory generation, forward and inverse problems, and interpolation. The generative model further supports both conditional and unconditional generation. We demonstrate state-of-the-art performance on million-point simulations across diverse PDE systems featuring complex geometries, high-frequency dynamics, and long-term horizons.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。