面向 GPU 的高性能心脏电生理仿真:TorchCor 库
软件工程
2025-10-15 v1
摘要
心脏电生理(CEP)仿真日益增长,用以理解心脏阵发病及指导临床决策。然而,这些仿真通常需要大量 CPU 核心的高性能计算资源,而许多研究小组和临床医生往往无法获取。为此,我们提出 TorchCor,一个基于 PyTorch 的高性能 Python 库,利用通用 GPU 上的有限元方法进行 CEP 仿真。TorchCor 在特别适用于大规模三维网格的情形下显著加速了 CEP 仿真。求解器的准确性通过与制造解析解和 版本基准问题进行验证。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}
}