RM新时代网站-首页

0
  • 聊天消息
  • 系統(tǒng)消息
  • 評論與回復
登錄后你可以
  • 下載海量資料
  • 學習在線課程
  • 觀看技術視頻
  • 寫文章/發(fā)帖/加入社區(qū)
會員中心
創(chuàng)作中心

完善資料讓更多小伙伴認識你,還能領取20積分哦,立即完善>

3天內不再提示

阿里開源大規(guī)模分布式圖學習框架:專為Graph嵌入,無縫對接TF/PyTorch

DPVg_AI_era ? 來源:lq ? 2019-01-20 08:51 ? 次閱讀

阿里媽媽開源大規(guī)模分布式圖表征學習框架Euler,面向工業(yè)級用戶和高級研究者,結合TF/XDL/PyTorch等深度學習基礎工具,支持超大規(guī)模復雜異構圖的模型訓練。

阿里媽媽宣布開源大規(guī)模分布式的圖表示學習框架Euler,Euler內置DeepWalk、Node2Vec等業(yè)界常見的Graph Embedding算法,以及3種阿里媽媽自研創(chuàng)新算法,可以支持數十億點和數百億邊的復雜異構圖上進行模型訓練。

日前,Euler開源框架開發(fā)團隊的幾位成員接受了新智元的采訪。

在采訪中,他們對Euler框架與現(xiàn)在流行的TensorFlow和PyTorch框架的一些區(qū)別做了闡述,比如“圖”這一概念在定義和指代對象上的區(qū)別。并指出了Euler框架主要面向的兩類目標用戶。

需要指出,在TensorFlow和PyTorch中,圖(graph)指的是把程序執(zhí)行流程分解成為一個有向無環(huán)圖,圖中每個節(jié)點代表一個需要執(zhí)行的操作,邊代表各個操作的依賴關系。

在Euler中,圖(graph)指的是用戶的業(yè)務數據可以表達成復雜的異構圖,用戶希望基于圖進行Graph Embedding,并進一步進行節(jié)點分類,邊預測,圖分類等任務。

Euler針對兩類用戶:

擁有海量復雜工業(yè)級圖數據的用戶,用Euler可以快速訓練,拿到業(yè)務效果。

屬于高校或者企業(yè)里邊的研究實驗室用戶,他們利用Euler靈活的圖操作算子可以快速進行算法創(chuàng)新與定制。

概述

過去幾年隨著數據規(guī)模和硬件計算力的迅速增長,深度學習技術在工業(yè)界被廣泛應用并產生了巨大的技術紅利。當前應用已經相對成熟,下一步的技術紅利在哪里還在積極探索之中。圖神經網絡將端到端學習與歸納推理相結合,有望解決深度學習無法處理的關系推理、可解釋性等一系列問題。

對結構知識的表達、計算和組合泛化是實現(xiàn)具備human-like AI的關鍵,圖神經網絡有希望在這些方面形成突破,使得機器能力進一步提升,因此對圖神經網絡的深入應用有希望形成下一波技術紅利。

圖(Graph)作為表達能力很強的通用的數據結構,可以用來刻畫現(xiàn)實世界中的很多問題,例如社交場景的用戶網絡、電商場景的用戶和商品網絡、電信場景的通信網絡、金融場景的交易網絡和醫(yī)療場景的藥物分子網絡等等。相比文本、語音和圖像領域的數據比較容易處理成歐式空間的Grid-like類型,適合現(xiàn)有的深度學習模型處理,圖是一種非歐空間下的數據,并不能直接應用現(xiàn)有方法,需要專門設計的圖神經網絡系統(tǒng)。

1.1Euler的核心能力

大規(guī)模圖的分布式學習

工業(yè)界的圖往往具有數十億節(jié)點和數百億邊,有些場景甚至可以到數百億節(jié)點和數千億邊,在這樣規(guī)模的圖上單機訓練是不可行的。Euler支持圖分割和高效穩(wěn)定的分布式訓練,可以輕松支撐數十億點、數百億邊的計算規(guī)模。

支持復雜異構圖的表征

工業(yè)界的圖關系大都錯綜復雜,體現(xiàn)在節(jié)點異構、邊關系異構,另外節(jié)點和邊上可能有非常豐富的屬性,這使得一些常見的圖神經網絡很難學到有效的表達。Euler在圖結構存儲和圖計算的抽象上均良好的支持異構點、異構邊類型的操作,并支持豐富的異構屬性,可以很容易的在圖學習算法中進行異構圖的表征學習。

圖學習與深度學習的結合

工業(yè)界有很多經典場景,例如搜索/推薦/廣告場景,傳統(tǒng)的深度學習方法有不錯效果,如何把圖學習和傳統(tǒng)方法結合起來,進一步提升模型能力是很值得探索的。Euler支持基于深度學習樣本的mini-batch訓練,把圖表征直接輸入到深度學習網絡中聯(lián)合訓練。

分層抽象與靈活擴展

Euler系統(tǒng)抽象為圖引擎層、圖操作算子層、算法實現(xiàn)層三個層次,可以快速的在高層擴展一個圖學習算法。實際上,Euler也內置了大量的算法實現(xiàn)供大家直接使用。

1.2Euler內置的算法實現(xiàn)

考慮到框架的易用性,我們內置了多種知名算法以及幾種我們內部的創(chuàng)新算法。所有實現(xiàn),我們仔細進行了測試,保證了算法運行效率,且算法效果與原論文對齊。用戶無需進行開發(fā),注入數據到平臺后,可以直接使用。我們內置的算法列表見下表。鑒于公開算法知名度很高,我們不做出詳細描述,請參見原始論文。有關我們內部算法的詳細信息請見2.3節(jié)。


系統(tǒng)設計

Euler系統(tǒng)整體可以分為三層:最底層的分布式圖引擎,中間層圖語義的算子,高層的圖表示學習算法。

下邊我們分開描述各個層次的核心功能。

圖1 Euler架構總覽

2.1分布式圖引擎

為了支持我們的業(yè)務,我們不僅面臨超大規(guī)模圖存儲與計算的挑戰(zhàn),還需要處理由多種不同類型的點,邊及其屬性構成異構圖的復雜性。我們的分布式圖引擎針對海量圖存儲,分布式并行圖計算及異構圖進行了優(yōu)化設計,確保了工業(yè)場景下的有效應用。

首先為了存儲超大規(guī)模圖(數十億點,數百億邊),Euler必須突破單機的限制,從而采用了分布式的存儲架構。在圖加載時,整張圖在引擎內部被切分為多個子圖,每個計算節(jié)點被分配1個或幾個子圖進行加載。

為了充分利用各個計算節(jié)點的能力,在進行圖的操作時,頂層操作被分解為多個對子圖的操作由各個節(jié)點并行執(zhí)行。這樣隨著更多節(jié)點的加入,我們可以得到更好的服務能力。其次,我們引入了多replica的支持。從而用戶可以靈活平衡shard與replica的數量,取得更佳的服務能力。最后,我們針對圖表示學習優(yōu)化了底層的圖存儲數據結構與操作算法,單機的圖操作性能獲得了數倍的提升。

多種不同類型的邊,點與屬性所組成的異構圖,對很多復雜的業(yè)務場景必不可少。為了支持異構圖計算能力,底層存儲按照不同的節(jié)點與邊的類型分別組織。這樣我們可以高效支持異構的圖操作。

2.2中間圖操作算子

由于圖學習算法的多樣性以及業(yè)務的復雜性,固定的某幾種甚至幾十種算法實現(xiàn)無法滿足客戶的所有需求。所以在Euler設計中,我們圍繞底層系統(tǒng)的核心能力著重設計了靈活強大的圖操作算子,且所有算子均支持異構圖操作語義。用戶可以利用它來快速搭建自己的算法變體,滿足獨特的業(yè)務需求。

首先,Euler分布式圖引擎提供了C++的API來提供所有圖操作?;谶@個API,我們可以方便的基于某個深度學習框架添加圖操作的算子,從而利用Euler C++接口訪問底層圖引擎的能力。我們支持廣泛使用的深度學習框架,比如阿里巴巴的X-DeepLearning與流行的TensorFlow。后繼我們也會考慮支持其它的深度學習框架,比如PyTorch。

利用靈活的圖操作算子,機器學習框架可以在每個mini-batch與Euler交互,動態(tài)擴充與組織訓練樣本。這樣,Euler不僅支持傳統(tǒng)的以圖為中心的學習模式,且可以把圖學習的能力注入傳統(tǒng)的學習任務,實現(xiàn)端到端訓練。

按照功能分類,我們的核心系統(tǒng)提供的API可以分類如下:

全局帶權采樣點和邊的能力。主要用于mini-batch樣本的隨機生成以及Negative Sampling。

基于給定節(jié)點的鄰居操作。這個是圖計算的核心能力包括鄰居帶權采樣,取Top權重的鄰居等。

點/邊的屬性查找。這個能力使得算法可以使用更豐富的特征,而不僅限于點/邊的ID特征。

2.3高層算法實現(xiàn)

如1.2節(jié)所述,除了LINE算法以外,我們實現(xiàn)的算法可以分為隨機游走與鄰居匯聚兩大類算法。有關外部算法的詳細信息,請參見1.2節(jié)提供的論文鏈接。下面我們詳細介紹內部的三個創(chuàng)新算法。

Scalable-GCN

它是一種高效的GCN訓練算法。GCN以及更一般的Graph Neural Network (GNN)類的方法由于能有效的提取圖結構信息,在許多任務上均取得了超過以往方法的效果。但是GCN的模型會引入巨大的計算量,導致模型的訓練時間不可接受。

Scalable-GCN在保證優(yōu)秀效果的前提下,把mini-batch GCN的計算復雜度從層數的指數函數壓到線性。這使得在阿里媽媽的海量數據下應用三層GCN成為可能,廣告匹配的效果獲得了顯著提升。

LsHNE

LsHNE是我們結合阿里媽媽搜索廣告場景創(chuàng)新地提出一種無監(jiān)督的大規(guī)模異構網絡embedding學習方法。區(qū)別于DeepWalk類算法,LsHNE的特點包括:a) 采用深度神經網絡學習表達,可以有效融合Attribute信息;b)考慮embedding表示的距離敏感需求,提出兩個負采樣原則:分布一致性原則和弱相關性原則;c)支持異構網絡。

LasGNN

LasGNN是一種半監(jiān)督的大規(guī)模異構圖卷積神經網絡學習方法, 它有效融合了圖結構知識信息和海量用戶行為信息,大幅提升了模型精度,是工業(yè)界廣告場景下首次應用半監(jiān)督圖方法。該方法有多處創(chuàng)新,例如將metapath的思想應用于圖卷積網絡中,并提出了metapathGCN模型,有效解決了異構網絡的卷積問題;提出了metapathSAGE模型,在模型中我們設計高效的鄰居采樣的方法,使得大規(guī)模的多層鄰居卷積成為可能。

應用實例

Euler平臺已經在阿里媽媽搜索廣告的多個場景下廣泛實用,并取得了出色的業(yè)務效果,例如檢索匹配場景、CTR預估場景、營銷工具場景和反作弊場景等。我們以匹配場景的為例來看下Euler的應用。

廣告匹配的任務是給定用戶搜索請求,匹配模塊通過理解用戶意圖,快速準確地從海量廣告中找到高質量的小規(guī)模候選廣告集,輸送給下游的排序模塊進行排序。

我們首先使用一些傳統(tǒng)的挖掘算法,從用戶行為日志、內容屬性等維度挖掘出Query(查詢詞), Item(商品)和Ad(廣告)的多種關系,然后利用Euler平臺的LsHNE方法學習圖中節(jié)點的embedding,這里節(jié)點embedding后的空間距離刻畫了原來圖中的關系,對于在線過來的請求通過計算用戶查詢詞向量、前置行為中節(jié)點向量和廣告節(jié)點向量之間的距離進行高效的向量化最近鄰檢索,可以快速匹配到符合用戶意圖的廣告。圖2展示了LsHNE方法的離線和在線流程。圖3展示了樣本構造和網絡結構示意。

圖2 DeepMatch召回框架

圖3 離線訓練流程

開源信息

Euler已在Github開源,開源地址如下:

https://github.com/alibaba/euler

Euler開源框架受訪人信息

李金輝(花名見獨)

資深技術專家,阿里媽媽工程平臺技術負責人

北京理工大學模式識別專業(yè)碩士畢業(yè),現(xiàn)任阿里媽媽-工程平臺的技術負責人,定義了阿里媽媽最新一代廣告工程平臺AdOS,過往一直從事廣告和搜索系統(tǒng)架構領域的技術研發(fā)和管理工作,在大數據計算,廣告引擎,搜索引擎和機器學習系統(tǒng)的技術方向上具備10年以上經驗,還曾擔任淘寶搜索引擎的核心架構師,阿里云彈性云計算產品線的產品主管。

朱曉宇 (花名赤空)

阿里媽媽研究員/資深總監(jiān)

負責阿里搜索直通車算法與工程技術,帶領團隊打造中國規(guī)模最大的商業(yè)廣告平臺,幫助數百萬電商商家建立數字化智能化的電商營銷解決方案

張迪(花名樂迪)

資深技術專家,阿里媽媽大數據計算與機器學習平臺的工程架構負責人

2010年上海交大碩士畢業(yè)后加入阿里巴巴,先后參與研發(fā)了淘寶新一代搜索引擎內核、神馬搜索大規(guī)模網頁處理系統(tǒng)、阿里媽媽大數據與機器學習平臺?,F(xiàn)在其團隊負責阿里媽媽全域數據倉庫、離線實時數據處理、深度學習建模等平臺的研發(fā)建設,支撐著阿里巴巴搜索廣告、精準定向廣告、品牌廣告等眾多核心業(yè)務場景。

林偉(花名楊焜)

阿里巴巴高級算法專家、阿里媽媽搜索廣告匹配和創(chuàng)意算法負責人

2009年北京航空航天大學碩士畢業(yè)后加入騰訊搜搜,從事搜索研發(fā)。2011年加入阿里巴巴,先后參與阿里推薦算法ETREC、定向廣告大規(guī)模興趣點算法UIT++、搜索廣告近年來一系列創(chuàng)新的匹配算法、Euler圖深度學習算法和平臺等。在WWW、WSDM等頂級會議發(fā)表過多篇論文?,F(xiàn)在其團隊負責阿里媽媽圖深度學習算法、搜索廣告Query分析算法、相關性算法、匹配算法、創(chuàng)意算法和粗排算法等眾多核心業(yè)務場景。

聲明:本文內容及配圖由入駐作者撰寫或者入駐合作網站授權轉載。文章觀點僅代表作者本人,不代表電子發(fā)燒友網立場。文章及其配圖僅供工程師學習之用,如有內容侵權或者其他違規(guī)問題,請聯(lián)系本站處理。 舉報投訴
  • 神經網絡
    +關注

    關注

    42

    文章

    4771

    瀏覽量

    100712
  • 機器學習
    +關注

    關注

    66

    文章

    8406

    瀏覽量

    132558
  • 深度學習
    +關注

    關注

    73

    文章

    5500

    瀏覽量

    121111
  • pytorch
    +關注

    關注

    2

    文章

    807

    瀏覽量

    13198

原文標題:阿里開源大規(guī)模分布式圖學習框架:專為Graph嵌入,無縫對接TF/PyTorch

文章出處:【微信號:AI_era,微信公眾號:新智元】歡迎添加關注!文章轉載請注明出處。

收藏 人收藏

    評論

    相關推薦

    HarmonyOS分布式應用框架深入解讀

    KB級到GB級設備)。針對上述挑戰(zhàn),HarmonyOS作為一款面向萬物互聯(lián)時代的、全新的分布式操作系統(tǒng),將迎刃而解,這得益于HarmonyOS的分布式應用框架,這些多設備組成一個超級終端,充分發(fā)揮
    發(fā)表于 11-22 15:15

    如何高效完成HarmonyOS分布式應用測試?

    問題和UX顯示問題占比率高達85%。1 HarmonyOS分布式應用上架問題分析分布式應用測試效率低:分布式應用涉及多臺設備協(xié)同時,由于沒有統(tǒng)一的測試
    發(fā)表于 12-13 18:07

    OpenHarmony3.1分布式技術資料合集

    。屏幕服務(ScreenService):分布式屏幕主控端SA服務和分布式屏幕被控端SA服務,負責處理分布式硬件管理框架的IPC調用軟總線適配器(SoftbusAdapter):
    發(fā)表于 04-11 11:50

    大規(guī)模分布式互聯(lián)網應用的測量

    日益提高的效率與安全性需求催生了大量的大規(guī)模分布式新型互聯(lián)網應用。這些網絡應用所呈現(xiàn)的動態(tài)、分散、可擴展、高容錯、隱私性強等特點,給網絡應用的設計與管理帶來了挑戰(zhàn)
    發(fā)表于 05-30 10:04 ?15次下載
    <b class='flag-5'>大規(guī)模</b><b class='flag-5'>分布式</b>互聯(lián)網應用的測量

    大規(guī)模分布式機器學習系統(tǒng)分析

    ,通過對比三款開源分布式LDA系統(tǒng)-Spark LDA、PLDA+和LightLDA,在系統(tǒng)資源消耗、算法收斂性能和可擴展性等方面的表現(xiàn),分析各系統(tǒng)在設計、實現(xiàn)和性能上的差異。實驗結果表明:面對小規(guī)模的樣本集和模型,LightL
    發(fā)表于 12-05 19:02 ?0次下載
    <b class='flag-5'>大規(guī)模</b><b class='flag-5'>分布式</b>機器<b class='flag-5'>學習</b>系統(tǒng)分析

    Apache Spark的分布式深度學習框架BigDL的概述

    該視頻概述了Apache Spark *的BigDL分布式深度學習框架。
    的頭像 發(fā)表于 10-30 06:41 ?3247次閱讀

    阿里發(fā)布分布式計算平臺,將極大推動計算進入“普惠時代”

    11月9日,在第二屆世界科技與發(fā)展論壇上,阿里發(fā)布全球首個一站大規(guī)模分布式計算平臺GraphScope,并入選中國科學技術協(xié)會“科創(chuàng)中
    的頭像 發(fā)表于 11-10 09:39 ?2022次閱讀

    Google Brain和DeepMind聯(lián)手發(fā)布可以分布式訓練模型的框架

    。 正如吳恩達所言,當代機器學習算法的成功很大程度上是由于模型和數據集大小的增加,在大規(guī)模數據下進行分布式訓練也逐漸變得普遍,而如何在大規(guī)模數據、大模型的情況下進行計算,還是一個挑戰(zhàn)。
    的頭像 發(fā)表于 06-26 15:42 ?2243次閱讀
    Google Brain和DeepMind聯(lián)手發(fā)布可以<b class='flag-5'>分布式</b>訓練模型的<b class='flag-5'>框架</b>

    如何使用TensorFlow進行大規(guī)模分布式的QML模擬

    中,我們將逐步帶您了解如何使用 TensorFlow 和 TensorFlow Quantum 進行大規(guī)模分布式
    的頭像 發(fā)表于 08-10 17:31 ?2624次閱讀

    探究超大Transformer語言模型的分布式訓練框架

    。 優(yōu)化的分布式集群架構:NVIDIA DGX SuperPOD 有了高效的分布式訓練框架,自然也需要優(yōu)化的分布式訓練集群。 NVIDIA DGX SuperPOD 便是 NVIDIA
    的頭像 發(fā)表于 10-20 09:25 ?2435次閱讀

    基于OpenHarmony的分布式應用開發(fā)框架使用教程

    電子發(fā)燒友網站提供《基于OpenHarmony的分布式應用開發(fā)框架使用教程.zip》資料免費下載
    發(fā)表于 04-12 11:19 ?9次下載

    PyTorch開源深度學習框架簡介

    PyTorch 是一種開源深度學習框架,以出色的靈活性和易用性著稱。這在一定程度上是因為與機器學習開發(fā)者和數據科學家所青睞的熱門 Pytho
    的頭像 發(fā)表于 07-29 10:26 ?4430次閱讀

    深度學習框架pytorch入門與實踐

    的。PyTorch是一個開源的深度學習框架,在深度學習領域得到了廣泛應用。本文將介紹PyTorch
    的頭像 發(fā)表于 08-17 16:03 ?1591次閱讀

    深度學習框架pytorch介紹

    。PyTorch具有易于使用的API和文檔,并強制執(zhí)行Python編碼標準。這使得它成為機器學習從業(yè)者的首選框架之一。PyTorch支持CPU和GPU計算以及
    的頭像 發(fā)表于 08-17 16:10 ?1772次閱讀

    基于PyTorch的模型并行分布式訓練Megatron解析

    NVIDIA Megatron 是一個基于 PyTorch分布式訓練框架,用來訓練超大Transformer語言模型,其通過綜合應用了數據并行,Tensor并行和Pipeline并行來復現(xiàn) GPT3,值得我們深入分析其背后機
    的頭像 發(fā)表于 10-23 11:01 ?2966次閱讀
    基于<b class='flag-5'>PyTorch</b>的模型并行<b class='flag-5'>分布式</b>訓練Megatron解析
    RM新时代网站-首页