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

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科学团队密切协作完成。

###



延伸阅读
相关推荐
热点推荐
悲哀!网传导游带着日本人插队,称外国人优先,网友爆粗口争执…

悲哀!网传导游带着日本人插队,称外国人优先,网友爆粗口争执…

火山诗话
2024-04-25 13:34:43
安理会上,俄美代表互相质问对方“为什么?”

安理会上,俄美代表互相质问对方“为什么?”

观察者网
2024-04-25 11:51:24
70岁赵雅芝去世,世间再无白素贞?赵雅芝工作室回应了

70岁赵雅芝去世,世间再无白素贞?赵雅芝工作室回应了

拾娱先生
2024-04-25 15:30:08
4元拿下5600亩地承包权,开鲁“张氏兄弟”身份曝光,果然不一般

4元拿下5600亩地承包权,开鲁“张氏兄弟”身份曝光,果然不一般

叶公子
2024-04-26 00:13:10
贾跃亭:已替乐视偿还100多亿美元债务!在美国法律体系下他已没有任何债务,尽快解决中国法律体系下债务

贾跃亭:已替乐视偿还100多亿美元债务!在美国法律体系下他已没有任何债务,尽快解决中国法律体系下债务

和讯网
2024-04-25 16:29:15
跌破60%大关!美国一季度GDP突破7万亿美元,对标中国优势变大了

跌破60%大关!美国一季度GDP突破7万亿美元,对标中国优势变大了

南生观察室
2024-04-26 00:07:08
马克龙警告:欧洲面临消亡威胁 应加强经济和国防领域的改革

马克龙警告:欧洲面临消亡威胁 应加强经济和国防领域的改革

财联社
2024-04-25 22:36:13
取消5450亿美债,中方拒做“接盘侠”,耶伦担忧:可怕的在后面

取消5450亿美债,中方拒做“接盘侠”,耶伦担忧:可怕的在后面

王五说说看
2024-04-25 18:50:13
曝光后才知道:财政如此吃紧,钱都花去哪了?原来是这样花掉的!

曝光后才知道:财政如此吃紧,钱都花去哪了?原来是这样花掉的!

清欢渡语
2024-04-25 21:35:47
曼晚:曼联会考虑对拉什福德6000万英镑的报价

曼晚:曼联会考虑对拉什福德6000万英镑的报价

懂球帝
2024-04-26 00:48:14
俄国防部副部长被抓,他是绍伊古的“老搭档”

俄国防部副部长被抓,他是绍伊古的“老搭档”

中国新闻周刊
2024-04-25 18:19:22
侃爷让她穿啥就穿啥,网友:她也愿意的,痛苦并快乐着

侃爷让她穿啥就穿啥,网友:她也愿意的,痛苦并快乐着

三月柳
2024-04-13 15:27:23
曼城4-0双杀布莱顿!76分升至英超第二,福登双响,德布劳内破门

曼城4-0双杀布莱顿!76分升至英超第二,福登双响,德布劳内破门

侃球熊弟
2024-04-26 04:56:59
NBA官宣年度大奖:库里当选最佳关键球员 10项关键数据联盟第一

NBA官宣年度大奖:库里当选最佳关键球员 10项关键数据联盟第一

罗说NBA
2024-04-26 07:04:11
禁止调试,阻止浏览器F12开发者工具

禁止调试,阻止浏览器F12开发者工具

搞笑的阿万
2024-04-23 00:37:23
“我不懂法”的镇党委副书记,纪云浩的背后,是一条多大的鱼?

“我不懂法”的镇党委副书记,纪云浩的背后,是一条多大的鱼?

燕梳楼2021
2024-04-25 14:56:11
统一着装不能在上海滨江跑步?管理方:并非两三人,系50人跑团;保安未阻拦只劝其收起旗帜;商业活动需报备,已后补

统一着装不能在上海滨江跑步?管理方:并非两三人,系50人跑团;保安未阻拦只劝其收起旗帜;商业活动需报备,已后补

纵相新闻
2024-04-25 17:16:31
她被称为中国黄金身材,巅峰期退圈结婚,丈夫身份特殊到不方便说

她被称为中国黄金身材,巅峰期退圈结婚,丈夫身份特殊到不方便说

简读视觉
2024-04-11 20:00:03
江西于都一矿企发生事故,致3死2伤

江西于都一矿企发生事故,致3死2伤

界面新闻
2024-04-26 08:17:06
U型锁们,请你们放下以色列的饮料!

U型锁们,请你们放下以色列的饮料!

缤纷色彩娱乐
2024-04-25 17:47:09
2024-04-26 08:40:51

科技要闻

雷军周鸿祎出圈:中年CEO,抢着当网红

头条要闻

贾跃亭称在美法律体系下已没有债务 再回应"何时回国"

头条要闻

贾跃亭称在美法律体系下已没有债务 再回应"何时回国"

体育要闻

库里当选最佳关键球员 10项数据联盟第一

娱乐要闻

心疼!伊能静曝儿子曾被狗仔追到洗手间

财经要闻

24年后再产纯净水 农夫山泉为何要打自己脸

汽车要闻

全新哈弗H9亮相 大号方盒子硬派SUV入列

态度原创

手机
时尚
房产
公开课
军事航空

手机要闻

匪夷所思,华为回归不仅自己销量上去了,也带火了荣耀手机

女人再美不如会穿!参考这些搭配解析更有魅力,美到老很简单

房产要闻

涉及黄埔、番禺、增城!广州新一轮大规模征地启动

公开课

睡前进食会让你发胖吗?

军事要闻

以军称已完成对拉法地面军事行动准备工作

无障碍浏览 进入关怀版
×