中文

面向 GPU 的高性能心脏电生理仿真:TorchCor 库

软件工程 2025-10-15 v1

摘要

心脏电生理(CEP)仿真日益增长,用以理解心脏阵发病及指导临床决策。然而,这些仿真通常需要大量 CPU 核心的高性能计算资源,而许多研究小组和临床医生往往无法获取。为此,我们提出 TorchCor,一个基于 PyTorch 的高性能 Python 库,利用通用 GPU 上的有限元方法进行 CEP 仿真。TorchCor 在特别适用于大规模三维网格的情形下显著加速了 CEP 仿真。求解器的准确性通过与制造解析解和 NN 版本基准问题进行验证。TorchCor 可供学术和商业用途无限制使用。

关键词

引用

@article{arxiv.2510.12011,
  title  = {TorchCor: High-Performance Cardiac Electrophysiology Simulations with the Finite Element Method on GPUs},
  author = {Bei Zhou and Maximilian Balmus and Cesare Corrado and Ludovica Cicci and Shuang Qian and Steven A. Niederer},
  journal= {arXiv preprint arXiv:2510.12011},
  year   = {2025}
}