Graph Differentiable Architecture Search with Structure Learning

1 Introduction . In real-world applications including social networks , e-commerce , traffics , and biochemistry , a variety of relational data can be represented as graphs . This motivates the advent of graph neural networks ( GNNs ) based models such as GCN [ 1 ] , GAT [ 2 ] and GIN [ 3 ] , which are designed to learn and extract knowledge from the relational graph-structured data . To utilize information in graph structure and node features , GNNs follow a recursive message passing scheme where nodes aggregate information from their neighbors in each layer , making great success in various graph related tasks . Given that discovering ideal GNN architectures for different tasks is labor intensive and time consuming , graph architecture search [ 4 , 5 ] employs the ideas of neural architecture search ( NAS ) [ 6–8 ] to facilitate the automatic design of optimal GNN architectures so that a large number of human efforts can be saved . These automatically generated GNNs can achieve competitive or even better performance compared with manually designed GNNs on graph related tasks . 2 Nevertheless , existing works on graph architecture search mainly focus on designing search space and search strategy , ignoring the fundamental mechanism in adopting NAS for automatic GNN ∗Corresponding Authors . 2Our code will be released at https : //github.com/THUMNLab/AutoGL 35th Conference on Neural Information Processing Systems ( NeurIPS 2021 ) , virtual . structure search and failing to answer the following questions . ( i ) How does graph architecture search select its desired architectures ? and ( ii ) How optimal are the architectures selected by graph neural architecture search ? Answers to these questions can help us to understand the NAS and GNN mechanism in gathering messages , leading us to design better GNN structures for certain tasks . However , the failure of existing works in answering the questions significantly limits their capabilities of designing powerful GNN architectures . In this paper , we answer the above question through exploring how graph neural architecture search is able to select the desired GNN architectures . Given that neural architecture search approaches in literature can be mainly categorized into several groups , i.e. , gradient based ( DARTS [ 8 ] ) , reinforcement learning based , evolution algorithm based , and Bayesian optimization based methods , we only focus on gradient based architecture search approach in this work and leave investigations into the other groups as future works . We theoretically analyze DARTS behavior and find that DARTS prefers operations who can help to correct the predictions on hard data . Further analysis on graph data shows that different operations fit graphs with different amount of information in the node features and graph structures . Measurement study via designing synthetic experiments corroborates our theory . We find that DARTS on GNN is able to evaluate the usefulness of information hidden in node features and graph structure with respect to the target task , then automatically select appropriate operations based on the evaluated usefulness for GNN architecture . On the other hand , we also discover that the performance of gradient based graph architecture search ( e.g. , DARTS ) can also be deteriorated by noises inside the graphs , leading to suboptimal architectures . To solve the problem , we propose Graph differentiable Architecture Search model with Structure Optimization ( GASSO ) , which allows differentiable search of the architecture with gradient descent and is capable of searching the optimal architecture as well as adjusting graph structure adaptively through a joint optimization scheme . The graph structure adjustment in our proposed GASSO model serves as an adaptive noise reduction to enhance the architecture search performance . The framework of GASSO is shown in Figure 1 . We employ differentiable graph structure to allow us to optimize graph structure by gradient based methods during the training process , and we use hidden feature smoothness to evaluate the edge importance and constrain edge weights . Overall , we jointly optimize the parameters of neural architecture and graph structure in an iterative updating manner . We evaluate our model on several widely used graph benchmark datasets including CiteSeer , Cora and PubMed . The experimental results show that our proposed GASSO model can outperform state-of-the-art methods and demonstrate a better denoising ability than existing graph architecture search model using DARTS and other GNNs models . To summarize , this paper makes the following contributions : ( 1 ) We theoretically explore how graph neural architecture search is able to select the desired GNN architectures , to the best of knowledge , for the first time , showing that i ) gradient based graph architecture search prefers operations who can help to correct the predictions on hard data , and ii ) different operations fit graphs with different amount of information in the node features and graph structures . ( 2 ) We design measurement study with experiments to show that i ) gradient based graph architecture search is able to select operations based on the usefulness of the information in graphs , and ii ) noises in graph features and structures can deteriorate the architecture search performance . ( 3 ) We propose Graph differentiable Architecture Search model with Structure Optimization ( GASSO ) , which searches the optimal architecture as well as adjusts graph structure adaptively through a joint optimization scheme . Experimental results on several graph benchmark datasets demonstrate the superiority of our GASSO model against state-of-the-art methods . 2 Related Work . 2.1 Graph Neural Network . As an effective framework for graph representation learning , GNN [ 1–3 , 9–13 ] follows a neighborhood aggregation scheme . At each iteration , representations of nodes are generated through aggregating their neighbors ’ representations . For instance , graph convolutional neural network ( GCN ) [ 1 ] takes the average of all neighbor nodes ’ features as aggregation . Different from GCN , Graph Attention Network ( GAT ) [ 2 ] treats neighbors unequally . It calculates the attention on each neighbor . Therefore , the aggregation can emphasize representations of more important neighbor nodes with larger attentions . The vectorized representation of a given node in the graph after k iterations can capture both structural ( topological ) and semantic ( attributed ) information within the region of the target node ’ s k-hop neighborhood . As such , GNN characterizes the whole graph by aggregating node representations [ 1 , 14 ] , making themselves achieve the state-of-the-art results for tasks including node and graph classifications . Like other types of deep neural networks , GNNs are vulnerable to noise and adversarial attacks . Graph attackers usually adopt data perturbation by changing the node features or graph structure [ 15 , 16 ] . How to denoise and defend graph adversarial attacks is also a popular question [ 17–19 ] . Graph structure learning is presented in recent GNN methods [ 20–22 ] . Since graph structure plays an important role in GNN scheme , certain noise in the original graph is extremely harmful to GNNs . Graph structure learning aims to generate a clean graph during the training procedure . To achieve this , prior constraints such as low rank and smoothness are utilized . Learning a new graph structure improves the model ’ s ability of denoising , as well as its robustness to adversarial attacks of GNNs . 2.2 ( Graph ) Neural Architecture Search . Recent years have witnessed a significant surge in research on automated machine learning , including hyper-parameter optimization [ 23–26 ] and Neural Architecture Search ( NAS ) methods [ 6–8 , 27–30 ] aim at designing an neural architecture automatically for certain tasks . Since the architecture search space is discrete , reinforce learning [ 6 , 7 ] and evolution algorithm [ 31 , 32 ] are often used in NAS methods . Besides , transferring the discrete architecture search space into a differentiable space is another strategy for solving the NAS problems . DARTS [ 8 ] and SNAS [ 33 ] construct a super network where each operation is mixed by all candidate operations , making it possible to update the architecture as well as the weights simultaneously through the classical gradient descent method . NAS has also been applied to graph tasks , achieving competitive performance with the state-of-the-art performances . Existing works [ 4 , 5 , 34–36 ] mainly focus on defining a proper search space and search strategy for GNNs . However , the mechanism of NAS methods to select architectures and the relationship between NAS methods and GNN denoising ability have not been noticed . 3 Exploring Architecture Search for Graph . 3.1 Preliminaries : Differentiable Architecture Search . We use DARTS [ 8 ] , a representative NAS method with both conciseness and effectiveness as the exploration object in this paper . In DARTS , neural architecture is represented as a directed acyclic graph ( DAG ) where the input data is the source node , the output is the sink node , and the edges represent operations ( e.g . a GCN layer , or a linear layer ) adopted on the data . Following a micro search space setting , DARTS only searches which operation should be selected and how the nodes in the DAG should be connected . The calculation inside each operation is predefined . There are two phases in DARTS procedure , the searching phase and the evaluation phase . In searching phase , a super network ( shown in Figure 1 ) is constructed , where edges exist between each two nodes , i.e. , ei , j exists , ∀0 ≤ i < j ≤ N − 1 . Each edge is a mixed operation that can be calculated by ei , j ( xi ) = ∑ o∈O exp { αoi , j } ∑ o′∈O exp { αo ′ i , j } · o ( xi ) , where xi is the output of node i , O is candidate operation set , α is learnable architecture parameters . After calculating the mixed weighted sum , each node aggregates all input edges by xj = ∑ i < j ei , j ( xi ) . As such , all possible operations at all possible positions are contained in the super network . DARTS uses gradient based methods to optimize both architecture parameters α and operation parameters by a bi-level optimization scheme . Since the parameters are all trained in the super network , W of different architectures are shared with each other , the mathematical formulation is as follows : min A Lval ( W ∗ , A ) s.t . W ∗ = argminWEA∈Γ ( A ) Ltrain ( W , A ) . ( 1 ) Here , Γ ( A ) is architecture distribution learned in the searching phase . In this formulation , W and A are independently learned , improving the search efficiency . At the end of the training procedure , the operation with the maximum α of each node is chosen , and the selected operations compose the optimal architecture . Then comes to evaluation phase , a new network based on the designed architecture is constructed . This network will be trained from scratch and finally get tested after retraining . 3.2 Theoretical Analysis of Architecture Parameters . We next explore the DARTS behavior on graph task , aiming to answer ( i ) how does graph neural architecture search select its desired architectures ? and ( ii ) how optimal are the architectures selected by graph neural architecture search ? We firstly provide theoretically analysis on ( i ) . Mixed operation is the key part in DARTS , where the architecture parameters controls the operation selection . However , there is no theoretical analysis on how these architecture parameters change according to the performance of different operations . To explore how these architecture parameters change during training procedure , we firstly analyze a simple scene of a binary classification problem , and there is only one layer with two candidate operations needs searching in the super network . The mixed operation is F ( xk ) = o1f1 ( xk ) + o2f2 ( xk ) . Here , o1 = exp { α1 } exp { α1 } +exp { α2 } and o2 = exp { α2 } exp { α1 } +exp { α2 } . f1 and f2 are candidate operations . We use gradient methods to minimize binary cross entropy loss combined with a sigmoid function : H ( F ) = − ∑ k [ yk · lnσ ( F ( xk ) ) + ( 1− yk ) · ln ( 1− σ ( F ( xk ) ) ) ] . Then we can conduct following theorem with proof in Appendix A.1 . Theorem 1 The changes of operation weights on after one step of gradient descent w.r.t . architecture parameters depend on the score ∑ k [ wkfn ( xk ) ] , where wk = σ ( F ( xk ) ) − yk . Specifically , if∑ k [ wkf1 ( xk ) ] < ∑ k [ wkf2 ( xk ) ] , then o1 increases and o2 decreases , and vice versa . Theorem 1 tells us that DARTS judges different candidate operations by a score , which can be seen as a weighted sum of logits . As for the weights , we have wk ∈ ( −1 , 1 ) , whether it is positive or negative depends on the data label . If yk = 0 , the weight is a positive number σ ( F ( xk ) ) , which means those data with bad prediction given by the super network have larger weights . If only considering those scores with positive weights , the candidate operation giving better prediction has smaller logits . Summing them up will get smaller score and increase architecture weight on . We can reach a similar conclusion if we analyze data with label yk = 1 . In a nutshell , we know DARTS prefers those operations which can help to correct the predictions over hard data .

Summary:  The paper investigates the ability of gradient-based NAS method on selecting desired operators to construct GNN architectures. It shows that DARTs method can discriminate the benefits of different operators based on the graph structure but suffers from the noised in graph. Based on the findings, the authors propose to add an extra graph-structure learning term during the optimization process to denoise the graph structure and shows improved performance compared to existing baselines.