人大、東北大合開(kāi)發(fā)「圖機(jī)器學(xué)習(xí)庫(kù)」Jittor Geometric！性能超越PYG、DGL

原標(biāo)題：人大、東北大合開(kāi)發(fā)「圖機(jī)器學(xué)習(xí)庫(kù)」Jittor Geometric！性能超越PYG、DGL
文章來(lái)源：新智元
內(nèi)容字?jǐn)?shù)：6271字

Jittor Geometric 1.0：國(guó)產(chǎn)圖機(jī)器學(xué)習(xí)庫(kù)強(qiáng)勢(shì)來(lái)襲

中國(guó)人民大學(xué)和東北大合開(kāi)發(fā)的圖機(jī)器學(xué)習(xí)庫(kù)Jittor Geometric 1.0版本正式發(fā)布，該庫(kù)基于國(guó)產(chǎn)深度學(xué)習(xí)框架Jittor，在性能和易用性上都展現(xiàn)出顯著優(yōu)勢(shì)。

1. 性能優(yōu)越，超越主流框架

Jittor Geometric在多個(gè)圖學(xué)習(xí)任務(wù)上的運(yùn)行時(shí)間比PyTorch Geometric (PyG) 和Deep Graph Library (DGL) 等同類(lèi)型框架提升了10%~50%。其高性能源于AVX向量化、頂點(diǎn)級(jí)別并行計(jì)算、稀疏矩陣算子設(shè)計(jì)、基于warp的并行計(jì)算以及CPU-GPU自動(dòng)內(nèi)存交換等多項(xiàng)技術(shù)優(yōu)化。在小規(guī)模圖數(shù)據(jù)集上，訓(xùn)練速度提升25%；在大規(guī)模圖數(shù)據(jù)集上，顯著優(yōu)于PyTorch Geometric和Paddle Graph Learning，與Deep Graph Library相當(dāng)。

2. 功能全面，支持多種模型和數(shù)據(jù)集

Jittor Geometric支持多種類(lèi)型的圖數(shù)據(jù)，涵蓋圖神經(jīng)網(wǎng)絡(luò)各領(lǐng)域經(jīng)典任務(wù)的數(shù)據(jù)集，并提供靈活的API方便用戶(hù)導(dǎo)入自定義數(shù)據(jù)集。它集成了譜域、動(dòng)態(tài)、幾何圖神經(jīng)網(wǎng)絡(luò)模型等多種代表性圖學(xué)習(xí)模型，包括深層圖卷積模型GCNII和分子圖基座模型Uni-Mol等前沿模型，極大降低了用戶(hù)的學(xué)習(xí)成本。

3. 簡(jiǎn)潔易用，降低學(xué)習(xí)門(mén)檻

Jittor Geometric采用與PyTorch Geometric相似的模塊化接口，易于上手。研發(fā)團(tuán)隊(duì)提供了良好的課程支持，并已將其應(yīng)用于研究生課程教學(xué)，學(xué)生反饋良好，認(rèn)為該框架簡(jiǎn)明易懂，學(xué)習(xí)成本低，訓(xùn)練速度快。

4. 未來(lái)展望：持續(xù)迭代，拓展應(yīng)用

Jittor Geometric團(tuán)隊(duì)表示未來(lái)將持續(xù)改進(jìn)，主要方向包括：補(bǔ)充更多前沿模型、加強(qiáng)對(duì)動(dòng)態(tài)圖的高效計(jì)算支持、進(jìn)一步優(yōu)化分布式訓(xùn)練框架以及提供PyTorch Geometric到Jittor Geometric的自動(dòng)轉(zhuǎn)換腳本。

5. 研發(fā)團(tuán)隊(duì)及開(kāi)源信息

Jittor Geometric由中國(guó)人民大學(xué)魏哲巍教授和東北大學(xué)張巖峰教授及其團(tuán)隊(duì)共同打造。該項(xiàng)目已開(kāi)源，開(kāi)源地址：https://github.com/AlgRUC/JittorGeometric。

總而言之，Jittor Geometric 1.0 的發(fā)布為國(guó)內(nèi)圖機(jī)器學(xué)習(xí)領(lǐng)域帶來(lái)了一個(gè)強(qiáng)有力的工具，其高效的性能、全面的功能以及易用性將助力研究人員和開(kāi)發(fā)者們更便捷地進(jìn)行前沿探索和落地應(yīng)用。 團(tuán)隊(duì)致力于將其打造成國(guó)內(nèi)外學(xué)術(shù)界最受歡迎的圖機(jī)器學(xué)習(xí)庫(kù)之一，并積極推動(dòng)其在AI產(chǎn)業(yè)界的應(yīng)用。

聯(lián)系作者

文章來(lái)源：新智元
作者微信：
作者簡(jiǎn)介：智能+中國(guó)主平臺(tái)，致力于推動(dòng)中國(guó)從互聯(lián)網(wǎng)+邁向智能+新紀(jì)元。重點(diǎn)關(guān)注人工智能、機(jī)器人等前沿領(lǐng)域發(fā)展，關(guān)注人機(jī)融合、人工智能和機(jī)器人對(duì)人類(lèi)社會(huì)與文明進(jìn)化的影響，領(lǐng)航中國(guó)新智能時(shí)代。