网易首页 > 网易科技 > 网易科技 > 正文

AWS上海人工智能研究院推动研发的DGL图神经网络框架已在Amazon SageMaker上推出

0
分享至

我们高兴地宣布,Deep Graph Library (DGL) ——一个便于部署图神经网络的开源框架——在Amazon SageMaker上推出啦。

近年来,深度学习席卷世界,各种原因是它具有从复杂数据(如自由格式的文本、图像或视频)中提取复杂模式的神奇能力。但是,许多数据集不符合这些类别,更适合用图或者说网络来表示。很自然的,我们能够意识到传统的神经网络架构,如卷积神经网络或循环神经网络,并不适合这样的数据集,需要一种新的方法。

图神经网络入门

图神经网络(GNN)是当今机器学习中最令人兴奋的进展之一,以下参考论文有助于您开始学习。

GNN用于训练如下数据集的预测模型:

1. 社交网络,用图显示相关人员之间的联系,

1. 推荐系统,用图显示客户和条目之间的交互,

1. 化学分析,化合物被表示为原子和键图,

1. 网络安全,用图描述源和目标IP地址之间的连接,

1. 还有更多!

大多数时候,这些数据集非常大,有标记的只有一小部分。例如在欺诈检测场景中,我们分析某些用户与已知欺诈者的关系,预测他们是否是欺诈参与者的可能性。这一问题可以定义为半监督的学习任务,也即只对其中一小部分图节点进行标记(’欺诈者’或’合法’)。这样的解决方案,胜过构建手工标记的大型数据集、对它进行"线性化"以便使用传统的机器学习算法。

解决这些问题需要领域知识(例如零售、金融、化学等)、计算机科学知识(Python、深度学习、开源工具)和基础架构知识(训练、部署和模型扩展)。然而很少有人掌握所有这些技能,所以就需要DGL图神经框架和Amazon SageMaker这样的工具。

DGL图神经框架介绍

DGL图神经框架于2018年12月在Github上发布,它是一个开源的Python框架,可帮助研究人员、数据科学家和科学家在其数据集上快速构建、训练和评估图神经网络。

DGL建立在流行的深度学习框架(如PyTorch和Apache MXNet)之上。如果您知道其中一个或两个,你会发现得心应手。我们没有忘记TensorFlow的粉丝:DGL的下一个小版本将增加对TensorFlow的初步支持,预计下一个大版本将完全支持。

无论使用哪个框架,您都可以借助这些适合初学者的示例轻松入门。我还发现 GTC 2019研讨会的幻灯片和代码非常有用。

完成简单示例之后,您可以开始探索在DGL中已经实现的一系列前沿模型。例如,您可以使用图形卷积网络(GCN)和CORA数据集,训练文档分类模型:

$ python3 train.py --dataset cora --gpu 0 --self-loop

所有模型的代码都可以检查和调整。AWS团队对这些实现方法进行了仔细验证,验证了其性能,确保可以重现结果。

DGL 还包括一系列图数据集,您可以轻松地下载和试验这些数据集。

当然,您可以在本地安装和运行DGL,但为了让用户有更简单、流畅的体验,我们将其添加到PyTorch 和 Apache MXNet深度学习容器中,这使得在Amazon SageMaker上使用DGL更方便,便于规模化训练和部署模型,无需管理单个服务器。

药明康德 (WuXi AppTec) 是一家全球性的制药和医疗器械服务公司。开发一款新药是一个复杂、昂贵、漫长的过程,通常要花费26亿美元,平均需要12年。为了加快这一过程,药明康德的计算机辅助药物设计(CADD)团队一直在探索神经网络模型,以预测候选药物分子的药物特性。使用传统的方法,科学家们要花几个星期甚至几个月的时间来构建和验证模型、设置应用模型所需的计算资源。DGL和Amazon SageMaker为科学家提供了快速部署药物特性预测模型的解决方案,将建模时间缩短了5倍,从而加快了药物开发过程。

Bio-Techne是一家全球性的生命科学和诊断公司,为世界各地的研究人员和临床医生提供服务。公司的产品有超过50万种,包括高质量的试剂、仪器、临床控制,以及组织和液体活检诊断测试等。以高效而有意义的方式向其广泛的客户组织和推荐产品,变得至关重要。Bio-Techne数据实验室正在与 AWS 协作,积极测试图神经网络(GNN)的使用,以改进其当前的推荐算法。DGL 通过简单易用的 API ,简化了实现,将开发时间从几个月缩短到数周。BioTechne的评估结果表明,使用 DGL 实现的、基于GNN的推荐模型得出的Top 10推荐,其精度比非GNN模型提高了70%。Bio-Techne将继续优化这些模型、正式使用 DGL和Amazon SageMaker部署推荐算法。

DGL的研发由2018年底成立的AWS上海人工智能研究院推动,与美国帕洛阿尔托的MXNet科学团队密切协作完成。

###



延伸阅读
相关推荐
热点推荐
别墅搜出23件国宝,全家集体失联!整整17年,徐湖平案终于落幕

别墅搜出23件国宝,全家集体失联!整整17年,徐湖平案终于落幕

鉴史录
2026-05-31 12:57:55
巴铁两架中国战机坠毁

巴铁两架中国战机坠毁

烽火观天下
2026-05-31 11:27:42
独家:原中国航天科工集团董事长曾是著名导弹专家   其妻子曾在宾馆工作? 照顾了他半生

独家:原中国航天科工集团董事长曾是著名导弹专家  其妻子曾在宾馆工作? 照顾了他半生

新浪财经
2026-05-31 15:40:03
性压抑已经变态至此了吗!

性压抑已经变态至此了吗!

灯锦年
2026-05-31 11:39:51
忙活一桌子,狗都不吃!丈夫过生日,妻子自我感动 ,网友吐槽

忙活一桌子,狗都不吃!丈夫过生日,妻子自我感动 ,网友吐槽

蝴蝶花雨话教育
2026-05-31 00:05:20
成就三冠王!樊振东独得两分,萨尔布吕肯赢得乒乓球德甲冠军

成就三冠王!樊振东独得两分,萨尔布吕肯赢得乒乓球德甲冠军

澎湃新闻
2026-05-31 22:10:26
消息称尊界S800成功后,华为在帮助国产供应商发展高端子品牌

消息称尊界S800成功后,华为在帮助国产供应商发展高端子品牌

IT之家
2026-05-31 12:33:24
黄仁勋调侃居家办公:我把总部盖的这么漂亮,都没人要来上班吗?

黄仁勋调侃居家办公:我把总部盖的这么漂亮,都没人要来上班吗?

王爷说图表
2026-05-31 17:19:55
普京:已将残骸交给美国鉴定

普京:已将残骸交给美国鉴定

第一财经资讯
2026-05-30 18:20:21
吴官正:难忘那夜的秋雨

吴官正:难忘那夜的秋雨

坦腹斋
2018-09-01 16:32:58
日本正式加入北约援乌PURL计划,俄罗斯被联大纳入冲突暴力黑名单

日本正式加入北约援乌PURL计划,俄罗斯被联大纳入冲突暴力黑名单

史政先锋
2026-05-30 20:55:28
闹笑话了!黄仁勋摊牌:看不上华为韬定律,却被业内专家说不专业

闹笑话了!黄仁勋摊牌:看不上华为韬定律,却被业内专家说不专业

兵鉴史
2026-05-30 14:32:11
“大傻”去世15年后,儿子入狱孙子身亡,王晶曝他旧事:是狠角色

“大傻”去世15年后,儿子入狱孙子身亡,王晶曝他旧事:是狠角色

LULU生活家
2026-05-31 18:00:37
港媒:因操纵比赛,曾效力申花的中国香港后卫霍斌仁被判入狱17个月

港媒:因操纵比赛,曾效力申花的中国香港后卫霍斌仁被判入狱17个月

懂球帝
2026-05-31 17:25:28
柬埔寨一名中国地产商人被残忍杀害,生前遭控制殴打及施虐,妻子曾被绑匪索要1300多万元赎金,警方正全力追缉涉案人士

柬埔寨一名中国地产商人被残忍杀害,生前遭控制殴打及施虐,妻子曾被绑匪索要1300多万元赎金,警方正全力追缉涉案人士

极目新闻
2026-05-31 18:46:21
刚刚,国务院一纸令下,在中国延续了68年的户口战争终于落幕

刚刚,国务院一纸令下,在中国延续了68年的户口战争终于落幕

浪子的烟火人间
2026-05-31 01:25:03
香烟要被洗白了?中科院颠覆性研究,尼古丁或能降低40%死亡率?

香烟要被洗白了?中科院颠覆性研究,尼古丁或能降低40%死亡率?

心中的麦田
2026-05-29 21:08:59
41岁夫妻因“房事频繁”双双入院,医生提醒:每周不应超过一个数

41岁夫妻因“房事频繁”双双入院,医生提醒:每周不应超过一个数

医学原创故事会
2026-05-29 23:34:07
贵阳花果园闪婚骗局调查:娶妻流水线,女子上午离婚下午结婚

贵阳花果园闪婚骗局调查:娶妻流水线,女子上午离婚下午结婚

上游新闻
2026-05-31 14:46:25
周末炸雷!17家公司集中爆利空,3家立案10家退市

周末炸雷!17家公司集中爆利空,3家立案10家退市

慧眼看世界哈哈
2026-05-31 12:57:11
2026-06-01 01:23:00

科技要闻

戴尔诺基亚又回来了!AI重估老牌科技公司

头条要闻

媒体:印度多个领域面临严重问题 莫迪发出罕见的号召

头条要闻

媒体:印度多个领域面临严重问题 莫迪发出罕见的号召

体育要闻

阿森纳用最悲壮的方式,成就了巴黎王朝

娱乐要闻

朱军退休,正义虽迟但到,女方受惩

财经要闻

医学首席转岗搞科技,A股科技股遭遇巨震

汽车要闻

900V+3.2秒破百 领克10+&领克10上市16.99万元起

态度原创

家居
教育
房产
本地
公开课

家居要闻

云栖 舒展如流云

教育要闻

“根本夸不出口”,女同学展示初学的拉丁舞,服装成了最大减分项

房产要闻

红动五月!全国抢入核心资产,广州盯紧凯旋新世界!

本地新闻

用剪纸的方式,打开江苏扬州

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版
×