近日,中国科学技术大学苏州高等研究院的一项研究成果在国际学术界引起了广泛关注。该研究院下属的医学影像智能与机器人研究中心DDL实验室,成功研发出一种名为Capsule的新型图神经网络(GNN)训练框架,这一创新成果已被ACM International Conference on Management of Data(SIGMOD)2025正式收录。
图神经网络作为一种强大的工具,在推荐系统、自然语言处理、计算化学及生物信息学等多个领域展现出了卓越的性能。现有的主流GNN训练框架,如DGL和PyG,主要依赖GPU的并行计算能力,从复杂的图数据中高效提取结构信息。然而,随着图数据规模的日益增大,GPU显存容量的限制成为制约GNN系统扩展性的关键因素。
DDL实验室针对这一难题,提出了Capsule这一核外(Out-of-Core)GNN训练框架,为大规模图数据的处理提供了全新的解决方案。Capsule通过创新的图划分和图裁剪策略,确保训练子图的结构及特征能够完全加载到GPU显存中,从而避免了反向传播过程中CPU与GPU之间的频繁I/O操作,显著提升了训练效率。

不仅如此,Capsule还采用了基于最短哈密顿回路的子图加载方式,并引入了流水线并行策略,进一步优化了系统的整体性能。这一框架的设计极具灵活性,能够无缝集成到现有的主流开源GNN训练框架中,为用户提供了极大的便利。
在实际应用中,Capsule展现出了惊人的性能提升。在针对大规模真实图数据集的测试中,Capsule在仅使用22.24%内存的情况下,相比当前最优的系统性能提升了最高可达12.02倍。Capsule还首次提供了关于训练所得嵌入方差的理论上界,为GNN的训练和应用提供了更为坚实的理论基础。
DDL实验室的这一研究成果,标志着我国在图计算系统领域取得了重大突破。Capsule框架的推出,将为社交网络分析、生物医学图谱构建等需要处理超大规模图数据的应用场景提供强有力的技术支持,推动我国相关领域的研究和应用迈向新的高度。
举报 0收藏 0打赏 0评论 0分享 0 更多>同类资讯
量子直接通信破纪录!104.8公里稳定传输速度达2.38千比特每秒02-24
印度Vanar一代人形机器人面世,定位亲民价格引期待02-24
西安咸阳国际机场新地标:西部机场博物馆惊艳亮相!02-24
华能重庆两江燃机二期3号机组并网,西南最大天然气电厂将诞生!02-24
微软Microsoft Rewards积分大“缩水”,兑换门槛骤升四成以上!02-24
科研新突破!超千时长效制氢技术诞生,氢能科技企业占比六成02-24
美的集团再添新商标:“空间魔法师”与“冰川风”来袭?02-24
AMD抛售AI服务器工厂,台湾四巨头竞相抢购,总价或达40亿美元!02-24
荣耀MagicBook Pro 14笔记本详情曝光:重量、电池及互联能力成亮点02-24
重庆两江燃机二期:西南地区最大天然气清洁能源电厂并网发电!02-24
荣耀PC YOYO助理2.0新功能曝光,3月内旧款笔记本将迎来升级02-24
算力热潮涌动!机构本周密集调研,相关上市公司股价大涨02-23矿渣硬盘翻新泛滥!希捷银河、酷狼Pro系列如何避坑?02-23
摩托古兹2025新款V100系列发布:三大版本,科技感十足!02-23
DeepSeek北京公司户外logo悄然更换,背后原因竟是打卡热潮?三言科技消息 有北京网友透露,DeepSeek北京公司在户外的logo已经被换掉,此前该公司logo与很多知名大公司一起在户外的竖的指引牌上。 网友称,DeepSeek的logo先是几天前被拆除,然后又换上了…02-23点击查看更多 +全站最新
微软Microsoft Rewards积分大“缩水”,兑换门槛骤升四成以上!
小鹏X9首批300辆启航南沙,中国智能电车驶向全球新篇章!
科研新突破!超千时长效制氢技术诞生,氢能科技企业占比六成
《激战2》如何挑服务器?这些因素将决定你的游戏体验!
不知火舞玩法大揭秘:掌握这些技巧,让你在游戏中大放异彩!
美的集团再添新商标:“空间魔法师”与“冰川风”来袭?热门内容
印度Vanar一代人形机器人面世,定位亲民价格引期待
西安咸阳国际机场新地标:西部机场博物馆惊艳亮相!
华能重庆两江燃机二期3号机组并网,西南最大天然气电厂将诞生!
微软Microsoft Rewards积分大“缩水”,兑换门槛骤升四成以上!
科研新突破!超千时长效制氢技术诞生,氢能科技企业占比六成
美的集团再添新商标:“空间魔法师”与“冰川风”来袭?
本文链接:http://www.28at.com/showinfo-16-133818-0.html中国科大研发Capsule框架,GNN训练性能大幅提升12倍!
声明:本网页内容旨在传播知识,不代表本站观点,若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。