arXiv · ML· Sidharth S. Menon, Ameya D. Jagtap·· 3 小时前AI 评分38
TB-Transolver：通过时间捆绑实现时依赖 PDE 的并行化长时程滚动
TB-Transolver: Parallelized and Coherent Long-Horizon Rollouts via Temporal Bundling for Time-Dependent PDEs
AI 导读
提出 TB-Transolver 框架，利用时间捆绑将 Transolver 扩展至时依赖偏微分方程（PDE），打破串行执行瓶颈。 引入时空一致性指标 C(K) 预测最佳捆绑宽度；在塑性、流体等四个领域，单 GPU 训练加速达 29 倍，多节点多 GPU 并行最高加速 45.6 倍。 解决了标准 Transolver 在瞬态问题中计算成本随时间线性增长的效率难题。
来源：arXiv · ML · arxiv.org