3月25日下午,中國人工智能領(lǐng)軍企業(yè)曠視科技舉辦線上發(fā)布會,正式宣布開源新一代AI生產(chǎn)力平臺Brain++的核心深度學(xué)習(xí)框架曠視天元(MegEngine),成為全球首個將底層框架開源的人工智能企業(yè)。

天元的前身是曠視內(nèi)部自用的算法訓(xùn)練推理引擎MegEngine,由曠視3名實習(xí)生在2013年啟動研發(fā),并與2014年正式完成上線。這款完全自研的深度學(xué)習(xí)框架支撐了曠視6年以來在國際AI競賽的出色表現(xiàn),以及公司產(chǎn)品和業(yè)務(wù)的落地,目前服務(wù)于曠視研究院1400余名AI開發(fā)者。

圖:曠視聯(lián)合創(chuàng)始人兼CTO唐文斌

發(fā)布會上,曠視聯(lián)合創(chuàng)始人兼CTO唐文斌正式宣布將MegEngine的代碼開源,并指出這是一套訓(xùn)練推理一體化、動靜態(tài)合一的工業(yè)級深度學(xué)習(xí)框架,中文名字叫做天元。

圖:曠視正式宣布開源自研深度學(xué)習(xí)框架天元

降低AI開發(fā)門檻提升算法研發(fā)效率

曠視是全球最早用深度學(xué)習(xí)方法開展人工智能商業(yè)化探索的科技企業(yè),但由于早期AI 開發(fā)工具的匱乏,曠視靠自主研發(fā),從 0 到 1 地解決了算法生產(chǎn)的問題。直播現(xiàn)場,天元開源項目負責(zé)人、曠視研究院高級技術(shù)總監(jiān)田忠博從開發(fā)經(jīng)驗中的幾個痛點入手,向開發(fā)者介紹了天元的核心優(yōu)勢和特性。

比如傳統(tǒng)深度學(xué)習(xí)研發(fā)中,產(chǎn)品從原型到生產(chǎn)部署往往需要分開設(shè)計和調(diào)用訓(xùn)練框架和推理框架兩種框架,這就導(dǎo)致模型在訓(xùn)練和推理的轉(zhuǎn)換的過程中出現(xiàn)不明原因的性能或精度損失,需要開發(fā)者手工進行優(yōu)化,而算法在計算平臺部署的時候出現(xiàn)各種問題也無法追溯。

天元框架在設(shè)計之初就避免了這樣的問題,通過訓(xùn)練、推理一體化的機制省去模型轉(zhuǎn)換的過程,可直接將訓(xùn)練后的模型進行推理,并保證跨設(shè)備的模型精度對齊,同時天元內(nèi)置自動模型優(yōu)化、簡化流程,減少因手工操作的機會,降低出錯概率。

除此之外,天元還配置了 Python 和 C++ 接口,支持動態(tài)圖、靜態(tài)圖一鍵轉(zhuǎn)換和混合編程,可使用高級編程語言進行圖優(yōu)化和圖編譯。在運行時管理環(huán)節(jié),天元具備執(zhí)行流與調(diào)度器,采用動態(tài)、靜態(tài)內(nèi)存分配并存的方式,并通過獨創(chuàng)的自動的亞線性內(nèi)存管理優(yōu)化器可以得到更好的內(nèi)存優(yōu)化效果;在底層設(shè)計上,天元的計算內(nèi)核不僅兼容主流計算設(shè)備,且支持多機多卡和分布式訓(xùn)練。而為了解決模型復(fù)現(xiàn)困難的問題,天元則支持 PyTorch Module 導(dǎo)入,可針對計算機視覺任務(wù)進行優(yōu)化。

相比主流的開源深度學(xué)習(xí)框架,天元具有訓(xùn)練推理一體化、動靜合一、兼容并包、靈活高效的顯性優(yōu)勢,無論對于深度學(xué)習(xí)”老司機“,還是剛剛轉(zhuǎn)型AI開發(fā)者的用戶來說都非常友好,極大程度上降低了深度學(xué)習(xí)的門檻,提升了算法訓(xùn)練和部署的效率。

國產(chǎn)深度學(xué)習(xí)框架開源勢在必行

對于開發(fā)者來說,開源最大的意義在于技術(shù)共享和共建的成就感。20世紀末期,Linux 系統(tǒng)為中國開發(fā)者打開了通往開源“集市”的大門,開源軟件為國產(chǎn)操作系統(tǒng)和國產(chǎn)軟件平臺的從 0 到 1 開辟了基礎(chǔ),同時中國的開發(fā)者們也為國際的開源社區(qū)做出了大量的貢獻。開源社區(qū)的發(fā)展壯大和成果共享,促使很多企業(yè)實現(xiàn)了快速的成長,同時也觸發(fā)了技術(shù)的加速迭代,和大批信息技術(shù)人才的涌現(xiàn)。

近年來,人工智能、大數(shù)據(jù)、云計算、移動互聯(lián)、物聯(lián)網(wǎng)等領(lǐng)域呈現(xiàn)多點聯(lián)動的集群式創(chuàng)新態(tài)勢,不僅使以感知和智能計算為核心的弱人工智能出現(xiàn)規(guī)模化應(yīng)用的態(tài)勢,同時也推動深度學(xué)習(xí)、計算機視覺等技術(shù)加速演進,顛覆性變革的人工智能新理論、新技術(shù)和新方法日新月異。因此,開源、開放國產(chǎn)自主可控的人工智能深度學(xué)習(xí),也將成為各主體提升創(chuàng)新效率的基礎(chǔ)。

圖:天元開源項目負責(zé)人、曠視研究院高級技術(shù)總監(jiān)田忠博

田忠博表示,目前,曠視已在我國新一代人工智能開源開放平臺OpenI啟智社區(qū)和全球最大的開源社區(qū) GitHub 上同步發(fā)布了天元 Alpha 版源代碼,開發(fā)者也可以通過天元官方網(wǎng)站的在線深度學(xué)習(xí)工具免費調(diào)用算力、獲取最新數(shù)據(jù)集和訓(xùn)練腳本,進行簡單訓(xùn)練和試用。曠視為天元的開發(fā)者還準備了全球頂尖算法預(yù)訓(xùn)練模型的模型中心ModelHub,支持開發(fā)者開箱即用。

圖:天元Alpha版的35萬行代碼已發(fā)布于 OpenI啟智社區(qū) 和 GitHub

對于天元框架開源后的發(fā)展規(guī)劃,曠視透露將在技術(shù)貢獻者的幫助下,于6月份推出 Beta 版本。“也許下一代天元并不是由曠視的研發(fā)團隊做出來的,而是與你一起共創(chuàng)出來。我們希望能夠吸引到一些優(yōu)秀的技術(shù)人員來給我們更好的建議,一起貢獻代碼、共建更好的深度學(xué)習(xí)框架。”唐文斌表示。

開源不僅是開發(fā)者們共享技術(shù)成果的方式,更是推進技術(shù)產(chǎn)業(yè)化的催化劑。此次曠視能夠把公司最核心的底層技術(shù)與全球開發(fā)者共享,體現(xiàn)了中國科技企業(yè)的責(zé)任與擔(dān)當。未來,讓我們一起期待曠視天元的表現(xiàn),也期待此次開源能真正地激發(fā)市場的創(chuàng)造力,讓傳統(tǒng)產(chǎn)業(yè)和中小企業(yè)都能用上深度學(xué)習(xí)和人工智能技術(shù),加速人工智能產(chǎn)業(yè)落地。

（免責(zé)聲明：本網(wǎng)站內(nèi)容主要來自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿，凡在本網(wǎng)站出現(xiàn)的信息，均僅供參考。本網(wǎng)站將盡力確保所提供信息的準確性及可靠性，但不保證有關(guān)資料的準確性及可靠性，讀者在使用前請進一步核實，并對任何自主決定的行為負責(zé)。本網(wǎng)站對有關(guān)資料所引致的錯誤、不確或遺漏，概不負任何法律責(zé)任。
任何單位或個人認為本網(wǎng)站中的網(wǎng)頁或鏈接內(nèi)容可能涉嫌侵犯其知識產(chǎn)權(quán)或存在不實內(nèi)容時，應(yīng)及時向本網(wǎng)站提出書面權(quán)利通知或不實情況說明，并提供身份證明、權(quán)屬證明及詳細侵權(quán)或不實情況證明。本網(wǎng)站在收到上述法律文件后，將會依法盡快聯(lián)系相關(guān)文章源頭核實，溝通刪除相關(guān)內(nèi)容或斷開相關(guān)鏈接。 ）