AutoGL: A Library for Automated Graph Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Ziwei, Qin, Yijian, Zhang, Zeyang, Guan, Chaoyu, Cai, Jie, Chang, Heng, Jiang, Jiyan, Li, Haoyang, Sun, Zixin, Xie, Beini, Yao, Yang, Zhang, Yipeng, Wang, Xin, Zhu, Wenwu
Format: Preprint
Veröffentlicht: 2021
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916145951408128
author Zhang, Ziwei
Qin, Yijian
Zhang, Zeyang
Guan, Chaoyu
Cai, Jie
Chang, Heng
Jiang, Jiyan
Li, Haoyang
Sun, Zixin
Xie, Beini
Yao, Yang
Zhang, Yipeng
Wang, Xin
Zhu, Wenwu
author_facet Zhang, Ziwei
Qin, Yijian
Zhang, Zeyang
Guan, Chaoyu
Cai, Jie
Chang, Heng
Jiang, Jiyan
Li, Haoyang
Sun, Zixin
Xie, Beini
Yao, Yang
Zhang, Yipeng
Wang, Xin
Zhu, Wenwu
contents Recent years have witnessed an upsurge in research interests and applications of machine learning on graphs. However, manually designing the optimal machine learning algorithms for different graph datasets and tasks is inflexible, labor-intensive, and requires expert knowledge, limiting its adaptivity and applicability. Automated machine learning (AutoML) on graphs, aiming to automatically design the optimal machine learning algorithm for a given graph dataset and task, has received considerable attention. However, none of the existing libraries can fully support AutoML on graphs. To fill this gap, we present Automated Graph Learning (AutoGL), the first dedicated library for automated machine learning on graphs. AutoGL is open-source, easy to use, and flexible to be extended. Specifically, we propose a three-layer architecture, consisting of backends to interface with devices, a complete automated graph learning pipeline, and supported graph applications. The automated machine learning pipeline further contains five functional modules: auto feature engineering, neural architecture search, hyper-parameter optimization, model training, and auto ensemble, covering the majority of existing AutoML methods on graphs. For each module, we provide numerous state-of-the-art methods and flexible base classes and APIs, which allow easy usage and customization. We further provide experimental results to showcase the usage of our AutoGL library. We also present AutoGL-light, a lightweight version of AutoGL to facilitate customizing pipelines and enriching applications, as well as benchmarks for graph neural architecture search. The codes of AutoGL are publicly available at https://github.com/THUMNLab/AutoGL.
format Preprint
id arxiv_https___arxiv_org_abs_2104_04987
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle AutoGL: A Library for Automated Graph Learning
Zhang, Ziwei
Qin, Yijian
Zhang, Zeyang
Guan, Chaoyu
Cai, Jie
Chang, Heng
Jiang, Jiyan
Li, Haoyang
Sun, Zixin
Xie, Beini
Yao, Yang
Zhang, Yipeng
Wang, Xin
Zhu, Wenwu
Machine Learning
Artificial Intelligence
Recent years have witnessed an upsurge in research interests and applications of machine learning on graphs. However, manually designing the optimal machine learning algorithms for different graph datasets and tasks is inflexible, labor-intensive, and requires expert knowledge, limiting its adaptivity and applicability. Automated machine learning (AutoML) on graphs, aiming to automatically design the optimal machine learning algorithm for a given graph dataset and task, has received considerable attention. However, none of the existing libraries can fully support AutoML on graphs. To fill this gap, we present Automated Graph Learning (AutoGL), the first dedicated library for automated machine learning on graphs. AutoGL is open-source, easy to use, and flexible to be extended. Specifically, we propose a three-layer architecture, consisting of backends to interface with devices, a complete automated graph learning pipeline, and supported graph applications. The automated machine learning pipeline further contains five functional modules: auto feature engineering, neural architecture search, hyper-parameter optimization, model training, and auto ensemble, covering the majority of existing AutoML methods on graphs. For each module, we provide numerous state-of-the-art methods and flexible base classes and APIs, which allow easy usage and customization. We further provide experimental results to showcase the usage of our AutoGL library. We also present AutoGL-light, a lightweight version of AutoGL to facilitate customizing pipelines and enriching applications, as well as benchmarks for graph neural architecture search. The codes of AutoGL are publicly available at https://github.com/THUMNLab/AutoGL.
title AutoGL: A Library for Automated Graph Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2104.04987