awesome-auto-graph-learning

This is a paper collection about automated graph learning, i.e., fusing AutoML and graph learning. Two special focuses are graph hyper-parameter optimization (HPO) and graph neural architecture search (NAS).

Please submit a pull request if you want to add new papers or have any suggestions!

Survey

• [IJCAI 2021] Automated Machine Learning on Graphs: A Survey (Paper)
• [Extension] Automated Graph Machine Learning: Approaches, Libraries and Directions (Paper)
• (In Chinese)[计算机学报 2023] 图神经架构搜索综述 (Paper)

Tool

• [ICLR 2021 GTRL workshop] AutoGL: A Library for Automated Graph Learning (Code) (Homepage) (Paper)
• Lightweight Auto Graph Learning (Code) (Paper)

Graph NAS

2024

• [AAAI 2024] Multimodal Graph Neural Architecture Search Under Distribution Shifts (Paper)
• [AAAI 2024] Data-Augmented Curriculum Graph Neural Architecture Search Under Distribution Shifts (Paper)

2023

• [NeurIPS 2023] Unsupervised Graph Neural Architecture Search with Disentangled Self-Supervision (Paper) (Code)
• [NeurIPS 2023] Multi-task Graph Neural Architecture Search with Task-aware Collaboration and Curriculum (Paper) (Code)
• [NeurIPS 2023] GLEMOS: Benchmark for Instantaneous Graph Learning Model Selection (Paper)
• [CIKM 2023] Node-dependent Semantic Search over Heterogeneous Graph Neural Networks (Paper) [(Code)] https://github.com/BUPT-GAMMA/NDS
• [ICML 2023] Do Not Train It A Linear Neural Architecture Search of Graph Neural Networks (Paper)
• [ICLR 2023] AutoGT Automated Graph Transformer Architecture Search (Paper)
• [WWW 2023] Auto-HeG Automated Graph Neural Network on Heterophilic Graphs (Paper)
• [WWW 2023] Search to Capture Long-range Dependency with Stacking GNNs for Graph Classification (Paper)
• [AAAI 2023] Dynamic Heterogeneous Graph Attention Neural Architecture Search (Paper) (Code)
• [AAAI 2023] Differentiable Meta Multigraph Search with Partial Message Propagation on Heterogeneous Information Networks (Paper) (Code)
• [ICDE 2023] PSP Progressive Space Pruning for Efficient Graph Neural Architecture Search (Paper)
• [TKDE 2023] Automated Graph Neural Network Search under Federated Learning Framework (Paper)
• [TKDE 2023] HGNAS++ Efficient Architecture Search for Heterogeneous Graph Neural Networks (Paper)
• [TOIS 2023] Neural Architecture Search for GNN-based Graph Classification (Paper)
• [AI 2023] AutoSTG+: An automatic framework to discover the optimal network for spatio-temporal graph prediction (Paper)
• [TETC 2023] CommGNAS: Unsupervised Graph Neural Architecture Search for Community Detection (Paper)
• [EngAppAI 2023] Meta-GNAS Meta-reinforcement learning for graph neural architecture search (Paper)
• [KAIS 2023] GM2NAS: multitask multiview graph neural architecture (Paper)
• [ASOC 2023] A surrogate evolutionary neural architecture search algorithm for graph neural networks search (Paper) (Code)
• [arXiv 2023] Efficient and Explainable Graph Neural Architecture Search via Monte-Carlo Tree Search (Paper) (Code)

2022

• [TKDD 2022] Auto-STGCN: Autonomous Spatial-Temporal Graph Convolutional Network Search (Paper)
• [ICDM 2022] Multi-Relational Graph Neural Architecture Search with Fine-grained Message Passing (Paper)
• [NeurIPS 2022] NAS-Bench-Graph: Benchmarking Graph Neural Architecture Search (Paper)(Code)
• [CIKM 2022] GraTO: Graph Neural Network Framework Tackling Over-smoothing with Neural Architecture Search (Paper) (Code)
• [ICML 2022] Large-Scale Graph Neural Architecture Search (Paper) (Code)
• [ICML 2022] Graph Neural Architecture Search Under Distribution Shifts (Paper)
• [ICML 2022] DFG-NAS: Deep and Flexible Graph Neural Architecture Search (Paper) (Code)
• [KDD 2022] Graph Neural Networks with Node-wise Architecture (Paper)
• [KDDDLG 2022] Graph Property Prediction on Open Graph Benchmark: A Winning Solution by Graph Neural Architecture Search (Paper) (Code)
• [SIGIR 2022] AutoGSR: Neural Architecture Search for Graph-based Session Recommendation (Paper)
• [TKDE 2022] GraphNAS++: Distributed Architecture Search for Graph Neural Network (Paper)
• [CVPR 2022] Automatic Relation-aware Graph Network Proliferation (Paper) (Code)
• [WWW 2022] PaSca a Graph Neural Architecture Search System under the Scalable Paradigm (Paper) (Code)
• [WWW 2022] Designing the Topology of Graph Neural Networks A Novel Feature Fusion Perspective (Paper) (Code)
• [ICDE 2022] AutoHEnsGNN Winning Solution to AutoGraph Challenge for KDD Cup 2020 (Paper) (Code)
• [TPDS 2022] Auto-GNAS A Parallel Graph Neural Architecture Search Framework (Paper)
• [WSDM 2022] Profiling the Design Space for Graph Neural Networks based Collaborative Filtering (Paper) (Code)
• [ESWA 2022] Efficient graph neural architecture search using Monte Carlo Tree search and prediction network (Paper)
• [Applied Intelligence] Automatic search of architecture and hyperparameters of graph convolutional networks for node classification (Paper)
• [arXiv 2022] AutoKE: An automatic knowledge embedding framework for scientific machine learning (Paper)
• [arXiv 2022] Enhancing Intra-class Information Extraction for Heterophilous Graphs: One Neural Architecture Search Approach (Paper)

2021

• [NeurIPS 2021] Graph Differentiable Architecture Search with Structure Learning (Paper) (Code)
• [NeurIPS 2021] AutoGEL: An Automated Graph Neural Network with Explicit Link Information (Paper) (Code)
• [ICDM 2021] Heterogeneous Graph Neural Architecture Search (Paper)
• [IJCNN 2021] Automated Graph Representation Learning for Node Classification (Paper)
• [PRICAI 2021] ALGNN Auto-Designed Lightweight Graph Neural Network (Paper)
• [CIKM 2021] Pooling Architecture Search for Graph Classification (Paper) (Code)
• [KDD 2021] DiffMG Differentiable Meta Graph Search for Heterogeneous Graph Neural Networks (Paper) (Code)
• [KDD 2021 DLG Workshop] Learn Layer-wise Connections in Graph Neural Networks (Paper)
• [ICML 2021] AutoAttend Automated Attention Representation Search (Paper)
• [SIGIR 2021] GraphPAS Parallel Architecture Search for Graph Neural Networks (Paper)
• [CVPR 2021] Rethinking Graph Neural Network Search from Message-passing (Paper) (Code)
• [GECCO 2021] Fitness Landscape Analysis of Graph Neural Network Architecture Search Spaces (Paper) (Code)
• [EuroSys 2021 EuroMLSys workshop] Learned low precision graph neural networks (Paper)
• [WWW 2021] Autostg: Neural architecture search for predictions of spatio-temporal graphs (Paper) (Code)
• [ICDE 2021] Search to aggregate neighborhood for graph neural network (Paper) (Code)
• [AAAI 2021] One-shot graph neural architecture search with dynamic search space (Paper)
• [arXiv] Search For Deep Graph Neural Networks (Paper)
• [arXiv] G-CoS GNN-Accelerator Co-Search Towards Both Better Accuracy and Efficiency (Paper)
• [arXiv] Edge-featured Graph Neural Architecture Search (Paper)
• [arXiv] FL-AGCNS: Federated Learning Framework for Automatic Graph Convolutional Network Search (Paper)

2020

• [NeurIPS 2020] Design space for graph neural networks (Paper) (Code)
• [ICONIP 2020] Autograph: Automated graph neural network (Paper)
• [BigData 2020] Graph neural network architecture search for molecular property prediction (Paper) (Code)
• [CIKM 2020] Genetic Meta-Structure Search for Recommendation on Heterogeneous Information Network (Paper) (Code)
• [CIKM 2020 CSSA workshop] Simplifying architecture search for graph neural network(Paper) (Code)
• [BRACIS 2020] Neural architecture search in graph neural networks (Paper) (Code)
• [IJCAI 2020] Graph neural architecture search (Paper) (Code)
• [CVPR 2020] SGAS: Sequential Greedy Architecture Search (Paper) (Code)
• [AAAI 2020] Learning graph convolutional network for skeleton-based human action recognition by neural searching (Paper) (Code)
• [OpenReview 2020] Efficient graph neural architecture search (Paper)
• [OpenReview 2020] FGNAS: FPGA-Aware Graph Neural Architecture Search (Paper)
• [arXiv 2020] Evolutionary architecture search for graph neural networks (Paper) (Code)
• [arXiv 2020] Probabilistic dual network architecture search on graphs (Paper)

2019

• [arXiv 2019] Auto-gnn: Neural architecture search of graph neural networks (Paper)

Graph HPO

2023

• [TKDE 2023] Revisiting Embedding Based Graph Analyses Hyperparameters Matter (Paper)

2022

• [CIKM 2022] Calibrate Automated Graph Neural Network via Hyperparameter Uncertainty (Paper)
• [KAIS 2022] Autonomous graph mining algorithm search with best performance trade-off (Paper)
• [ACL 2022] KGTuner: Efficient Hyper-parameter Search for Knowledge Graph Learning (Paper)
• [arXiv 2022] Start Small, Think Big On Hyperparameter Optimization for Large-Scale Knowledge Graph Embeddings (Paper)
• [arXiv 2022] Assessing the Effects of Hyperparameters on Knowledge Graph Embedding Quality (Paper)

2021

• [ICML 2021] Explainable Automated Graph Representation Learning with Hyperparameter Importance (Paper)
• [SIGIR 2021] Automated Graph Learning via Population Based Self-Tuning GCN (Paper)
• [PRICAI 2021] Automatic Graph Learning with Evolutionary Algorithms: An Experimental Study (Paper)
• [GECCO 2021] Which Hyperparameters to Optimise? An Investigation of Evolutionary Hyperparameter Optimisation in Graph Neural Network For Molecular Property Prediction (Paper)
• [P2PNA 2021] ASFGNN Automated separated-federated graph neural network (Paper)
• [arXiv 2021] A novel genetic algorithm with hierarchical evaluation strategy for hyperparameter optimisation of graph neural networks (Paper)
• [arXiv 2021] Jitune: Just-in-time hyperparameter tuning for network embedding algorithms (Paper)

2020

• [ICDM 2020] Autonomous graph mining algorithm search with best speed/accuracy trade-off (Paper) (Code)

2019

• [KDD 2019] AutoNE: Hyperparameter optimization for massive network embedding (Paper) (Code)

Applications

Finance

• [CIKM 2022] Explainable Graph-based Fraud Detection via Neural Meta-graph Search (Paper)

Biology

• [JBHI 2024] AutoDDI: Drug–drug Interaction Prediction with Automated Graph Neural Network (Paper)
• [Bioinformatics 2023] Cancer Drug Response Prediction With Surrogate Modeling-Based Graph Neural Architecture Search (Paper) (Code)
• [arXiv 2023] Uncertainty Quantification for Molecular Property Predictions with Graph Neural Architecture Search (Paper) (Code)
• [TCBB 2022] Multi-view Graph Neural Architecture Search for Biomedical Entity and Relation Extraction (Paper)
• [TCBB 2022] AutoMSR: Auto Molecular Structure Representation Learning for Multi-label Metabolic Pathway Prediction (Paper)
• [AILSCI 2022] AutoGGN: A gene graph network AutoML tool for multi-omics research (Paper)
• [BIBM 2021] Multi-label Metabolic Pathway Prediction with Auto Molecular Structure Representation Learning (Paper)

Knowledge Graph Embedding

• [arXiv 2021] AutoSF+: Towards Automatic Scoring Function Design for Knowledge Graph Embedding (Paper)
• [ICDE 2020] AutoSF: Searching Scoring Functions for Knowledge Graph Embedding (Paper) (Code)

Others

• [arXiv 2024] AutoGCN -- Towards Generic Human Activity Recognition with Neural Architecture Search (Paper)
• [CIKM 2023] Retrieving GNN Architecture for Collaborative Filtering (Paper)
• [CASES 2023] MaGNAS: A Mapping-Aware Graph Neural Architecture Search Framework for Heterogeneous MPSoC Deployment (Paper)
• [TIST 2023] Dual Graph Convolution Architecture Search for Travel Time Estimation (Paper)
• [WWW 2023] Automated Self-Supervised Learning for Recommendation (Paper)

Miscellaneous

Self-supervised Learning

• [LOGS 2022] AutoGDA Automated Graph Data Augmentation for Node Classification (Paper)
• [ICLR 2022] Automated Self-Supervised Learning for Graphs (Paper) (Code)
• [AAAI 2022] AutoGCL Automated Graph Contrastive Learning via Learnable View Generators (Paper) (Code)
• [ICML 2021] Graph Contrastive Learning Automated (Paper) (Code)

Others

• [ICLR 2023] AutoTransfer AutoML with Knowledge Transfer - An Application to Graph Neural Networks (Paper)
• [arXiv 2022] AutoGML Fast Automatic Model Selection for Graph Machine Learning (Paper)
• [TKDE 2021] Automated Unsupervised Graph Representation Learning (Paper) (Code)
• [arXiv 2022] Bridging the Gap of AutoGraph between Academia and Industry: Analysing AutoGraph Challenge at KDD Cup 2020 (Paper)

Cite

Please consider citing our survey paper if you find this repository helpful:

@inproceedings{zhang2021automated,
  title={Automated Machine Learning on Graphs: A Survey},
  author={Zhang, Ziwei and Wang, Xin and Zhu, Wenwu},
  booktitle = {Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence, {IJCAI-21}},
  year={2021},
  note={Survey track}
}