Unsupervised Graph Neural Architecture Search with Disentangled Self-supervision

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Zeyang, Wang, Xin, Zhang, Ziwei, Shen, Guangyao, Shen, Shiqi, Zhu, Wenwu
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929268604272640
author Zhang, Zeyang
Wang, Xin
Zhang, Ziwei
Shen, Guangyao
Shen, Shiqi
Zhu, Wenwu
author_facet Zhang, Zeyang
Wang, Xin
Zhang, Ziwei
Shen, Guangyao
Shen, Shiqi
Zhu, Wenwu
contents The existing graph neural architecture search (GNAS) methods heavily rely on supervised labels during the search process, failing to handle ubiquitous scenarios where supervisions are not available. In this paper, we study the problem of unsupervised graph neural architecture search, which remains unexplored in the literature. The key problem is to discover the latent graph factors that drive the formation of graph data as well as the underlying relations between the factors and the optimal neural architectures. Handling this problem is challenging given that the latent graph factors together with architectures are highly entangled due to the nature of the graph and the complexity of the neural architecture search process. To address the challenge, we propose a novel Disentangled Self-supervised Graph Neural Architecture Search (DSGAS) model, which is able to discover the optimal architectures capturing various latent graph factors in a self-supervised fashion based on unlabeled graph data. Specifically, we first design a disentangled graph super-network capable of incorporating multiple architectures with factor-wise disentanglement, which are optimized simultaneously. Then, we estimate the performance of architectures under different factors by our proposed self-supervised training with joint architecture-graph disentanglement. Finally, we propose a contrastive search with architecture augmentations to discover architectures with factor-specific expertise. Extensive experiments on 11 real-world datasets demonstrate that the proposed model is able to achieve state-of-the-art performance against several baseline methods in an unsupervised manner.
format Preprint
id arxiv_https___arxiv_org_abs_2403_05064
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Unsupervised Graph Neural Architecture Search with Disentangled Self-supervision
Zhang, Zeyang
Wang, Xin
Zhang, Ziwei
Shen, Guangyao
Shen, Shiqi
Zhu, Wenwu
Machine Learning
Artificial Intelligence
The existing graph neural architecture search (GNAS) methods heavily rely on supervised labels during the search process, failing to handle ubiquitous scenarios where supervisions are not available. In this paper, we study the problem of unsupervised graph neural architecture search, which remains unexplored in the literature. The key problem is to discover the latent graph factors that drive the formation of graph data as well as the underlying relations between the factors and the optimal neural architectures. Handling this problem is challenging given that the latent graph factors together with architectures are highly entangled due to the nature of the graph and the complexity of the neural architecture search process. To address the challenge, we propose a novel Disentangled Self-supervised Graph Neural Architecture Search (DSGAS) model, which is able to discover the optimal architectures capturing various latent graph factors in a self-supervised fashion based on unlabeled graph data. Specifically, we first design a disentangled graph super-network capable of incorporating multiple architectures with factor-wise disentanglement, which are optimized simultaneously. Then, we estimate the performance of architectures under different factors by our proposed self-supervised training with joint architecture-graph disentanglement. Finally, we propose a contrastive search with architecture augmentations to discover architectures with factor-specific expertise. Extensive experiments on 11 real-world datasets demonstrate that the proposed model is able to achieve state-of-the-art performance against several baseline methods in an unsupervised manner.
title Unsupervised Graph Neural Architecture Search with Disentangled Self-supervision
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2403.05064