西安电子科技大学盛凯教授团队论文被高性能计算领域国际顶级会议ICS 2025录用
14 小时前

西安电子科技大学盛凯教授团队的论文被2025年国际超级计算会议录用。该论文题为"Cherry: Breaking the GPU Memory Wall for Large-Scale GNN Training via Micro-Batching",由团队教师何鑫指导的2021级本科生汪焱担任第一作者,何鑫担任通讯作者。论文针对大规模图神经网络(GNN)训练面临的GPU内存墙挑战,提出了一种新颖的微批量训练方法Cherry。该方法结合了消息传递流图辅助的分区技术和基于微批的数据加载机制,有效减少了微批量分区中的冗余和负载不平衡,降低了训练过程中的数据准备开销。实验表明,Cherry能支持超出设备内存容量的大规模GNN训练,且在减少内存消耗和提升训练效率方面均优于现有方法,有助于降低大规模GNN训练的硬件部署成本。此次录用标志着该团队在高性能计算领域的研究获得了国际认可,该会议将于6月在美国犹他州盐湖城召开。