本文介绍了一种名为TopoGPU的GPU流式处理管线,用于计算二维和三维图像所诱导的立方复形的持续同调图。针对持续同调计算在深度学习网络训练中构成主要瓶颈的问题,作者提出将输入图像分块流式处理,利用大规模并行GPU内核在GPU块网格上逐块计算,并将边界关系累积到主机内存中由CPU完成边界矩阵约简。该方法引入了分层感知的离散莫尔斯匹配,可证明在流式处理下保持持续同调不变,同时提出了并行拓扑排序算法和并行V路径奇偶校验算法以在GPU上推导莫尔斯边界。在全部评测基准上,TopoGPU均优于当前最先进方法Cubical Ripser,端到端平均加速53.24倍,最高达198.01倍。作者进一步将其集成到保持拓扑的深度网络中,证明可显著降低训练期间持续同调计算的开销。该工作已被IEEE TPAMI接收,代码已开源。
| 持续同调 | 一种拓扑数据分析方法,通过追踪不同尺度下拓扑特征(如连通分量、环、空洞)的诞生与消亡来刻画数据形状。 |
| 立方复形 | 由图像像素或体素自然诱导的拓扑空间,由点、边、正方形、立方体等基本单元组成。 |
| 离散莫尔斯匹配 | 一种在离散莫尔斯理论中用于简化复形并保持同调不变性的配对方法,可减少计算量。 |
| 边界矩阵约简 | 计算持续同调的标准算法步骤,通过对边界矩阵进行列变换来提取持续对。 |
| Cubical Ripser | 当前针对立方复形持续同调计算的最先进高效算法,常作为性能基准。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅