Batch Before You Lift:Cluster-TNN 实现大规模图上的可扩展拓扑深度学习
研究者提出 Cluster-TNN,一种领域无关的拓扑深度学习框架,通过预处理划分图、运行时动态采样节点簇并在 mini-batch 内局部执行 graph lifting,避免全图提升带来的计算瓶颈。在与全图执行的 21 组对比中,Cluster-TNN 在所有配置下都降低峰值 GPU 显存,平均降幅 83.2%,同时保持有竞争力的预测性能。这一方法据作者所知首次使多种高阶 Topological Neural Networks 能在 Reddit(233k 节点、57.3M 边)和 OGBN Products 等大数据集上训练。