研究人员开发了Cluster-TNN,一个旨在提高大规模图拓扑深度学习可扩展性的新框架。传统的全域训练方法由于计算需求巨大,难以处理Reddit等海量数据集。Cluster-TNN通过划分图并动态采样节点簇来创建小批量,从而在这些批次中构建高阶拓扑结构,解决了这一问题。这种方法显著降低了GPU内存使用量(平均降低83.2%),同时保持了有竞争力的性能,使得在以前无法处理的数据集上训练复杂的拓扑神经网络成为可能。 AI
影响 使得在以前无法处理的大规模数据集上训练复杂的拓扑神经网络成为可能。
排序理由 这是一篇详细介绍机器学习特定领域新计算框架的研究论文。[lever_c_demoted from research: ic=1 ai=1.0]
- arXiv
- Cluster-TNN
- graphics processing unit
- large graphs
- OGBN Products
- topological deep learning
- Topological Neural Networks
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →