Descent Steps of a Relation-Aware Energy Produce Heterogeneous Graph Neural Networks
Abstract
Heterogeneous graph neural networks (GNNs) achieve strong performance on node classification tasks in a semi-supervised learning setting.
However, as in the simpler homogeneous GNN case, message-passing-based heterogeneous GNNs may struggle to balance between resisting the oversmoothing that may occur in deep models, and capturing long-range dependencies of graph structured data.
Moreover, the complexity of this trade-off is compounded in the heterogeneous graph case due to the disparate heterophily relationships between nodes of different types.
To address these issues, we propose a novel heterogeneous GNN architecture in which layers are derived from optimization steps that descend a novel relation-aware energy function.
The corresponding minimizer is fully differentiable with respect to the energy function parameters, such that bilevel optimization can be applied to effectively learn a functional form whose minimum provides optimal node representations for subsequent classification tasks.
In particular, this methodology allows us to model diverse heterophily relationships between different node types while avoiding oversmoothing effects.
Experimental results on 8 heterogeneous graph benchmarks demonstrates that our proposed method can achieve competitive node classification accuracy.
Introduction
Graph structured data, which contains information related to entities as well as the interactions between them, arise in various applications such as social networks or movie recommendations.
In a real world scenario, these entities and interactions are often multi-typed, such as actor , director , keyword , and movie in a movie review graph [17] or authors , papers , terms and venue in an academic network [28].
Hence the relationships between entities can potentially be much more complex than within a traditional homogeneous graph scenario in which only a single node and edge type exists.
To this end, heterogeneous graphs have been proposed as a practical and effective tool to systematically deal with such multi-typed graph-structured data.
With the advent of graph neural networks (GNN) for instantiating deep models that are sensitive to complex relationships between data instances, it is natural that numerous variants have been developed to explicitly handle heterogeneous graphs [4, 6, 9, 30, 26, 15], with promising performance on downstream tasks such as node classification or link prediction.
In this vein, message-passing GNNs represent one of the most widely-adopted heterogeneous architectures, whereby a sequence of type-specific graph propagation operators allow neighboring nodes to share and aggregate heterogeneous information [6, 26, 30].
However, in some cases where long-range dependencies exist between nodes across the graph, stacking too many layers to reflect a larger neighborhood may contribute to a well-known oversmoothing problem, in which node features converge to similar, non-discriminative embeddings [14, 19].
In the case of homogeneous graphs, this oversmoothing problem has been addressed (among other ways) using modified GNN architectures with layers patterned after the unfolded descent steps of some graph-regularized energy [7, 32].
Provided minimizers of this energy remain descriminative for the task at hand, arbitrarily deep networks can be trained without risk of oversmoothing, with skip connections between layers organically emerging to favor node-specific embeddings.
However, extending this approach to handle heterogeneous graphs is not straightforward, largely because node/edge heterogeneity ( e.g. , nodespecific numbers of classes or feature dimensions, multiple types of relationships between nodes) is not easily reflected in the vanilla energy functions adopted thus far for the homogeneous case.
In particular, existing energy functions are mostly predicated on the assumption of graph homophily, whereby neighboring nodes have similar labels.
But, in the heterogeneous case, this assumption is no longer realistic because the labels between nodes of different types may exhibit complex relationships.
To this end, we first introduce a novel heterogeneous GNN architecture with layers produced by the minimization of a relation-aware energy function.
This energy produces regularized heterogeneous node embeddings, relying on relation-dependent compatibility matrices as have been previously adopted in the context of belief propagation on heterogeneous graphs, e.g., ZooBP [5].
Secondly, we derive explicit propagation rules for our model layers by computing gradient descent steps along the proposed energy, with quantifiable convergence conditions related to the learning rate being considered as well.
Furthermore, we provide interpretations of each component of the resulting propagation rule, which consists of skip-connections, linearly transformed feature aggregation, and a self-loop transformation.
With respect to experiments, our contributions are twofold.
First, we show that our algorithm outperforms state-of-the-art baselines on various benchmark datasets.
And secondly, we analyze the the effectiveness of using the compatibility matrix and the behavior of the unfolding step as the number of propagations increases.
Existing Homogeneous GNNs from Unfolded Optimization
Consider a homogeneous graph G = {V , E} , with n = |V| nodes and m = |E| edges.
We define
L ∈ R n × n as the Laplacian of G , meaning L = D − A , in which D and A are degree and adjacency matrices, respectively.
Unfolded GNNs incorporate graph structure by attaining optimized embeddings Y ∗ ∈ R n × d that are functions of some adjustable weights W , i.e., Y ∗ ≡ Y ∗ ( W ) ,where ∂ Y ∗ ( W ) /∂
W is computable by design.
These embeddings are then inserted within an application-specific loss
[Equation]
in which g : R d → R c is some differentiable node-wise function with parameters θ and c -dimensional output tasked with predicting ground-truth node-wise targets t i ∈ R c .
Additionally, y ∗ i ( W ) is the i -th row of Y ∗ ( W ) , n ′ <
n is the number of labeled nodes (we assume w.l.o.g. that the first n ′ nodes are labeled), and D is a discriminator function, e.g. , cross-entropy for node classification, as will be our focus.
Given that Y ∗ ( W ) is differentiable by construction, we can optimize ℓ θ ( θ , W ) via gradient descent (over both θ and W if desired) to obtain our final predictive model.
At a conceptual level, what differentiates GNNs inspired by unfolded optimization is how Y ∗ ( W ) is obtained.
Specifically, these node embeddings are chosen to be the minimum of a lower-level, graph-regularized energy function [2, 16, 20, 32, 33, 35].
Orignally inspired by [34], the most common selection for this energy is given by
[Equation]
2) where λ > 0 is a trade-off parameter, Y ∈ R n × d is a matrix of trainable embeddings, meaning d -dimensional features across n nodes, and f ( X ; W ) is a base model parameterized by W that produces initial target embeddings using the d 0 -dimensional node features X ∈ R n × d 0 .
Minimization of (2) over Y results in new node embeddings Y ∗ characterized by a balance between similarity with f ( X ; W ) and smoothness across graph edges as favored by the trace term tr (cid:2) Y ⊤ LY (cid:3)
= (cid:80) { i,j }∈E ∥ y
i − y j ∥ 22 , where y i is i -th row of Y .
While (2) can be solved in a closed-form as
[Equation]
for large graphs this analytical solution is not practically computable.
Instead, we may form initial node embeddings Y (0) =
f ( X ; W ) , and then apply gradient descent to converge towards the minimum, with the k -th iteration given by
[Equation]
in which α 2 is the step size.
Given that L is generally sparse, computation of (4) can leverage efficient sparse matrix multiplications.
Combined with the fact that the loss is strongly convex, and provided that α is chosen suitably small, we are then guaranteed to converge towards the unique global optima with efficient gradient steps, meaning that for some K sufficiently large Y ( K ) ( W ) ≈ Y ∗ ( W ) .
Critically, Y ( K ) ( W ) remains differentiable with respect to W , such that we may substitute this value into (1) in place of Y ∗ ( W ) to obtain an entire pipeline that is end-to-end differentiable w.r.t.
W .
Moreover, per the analyses from [16, 20, 32, 33, 35], the k -th unfolded iteration of (4) can be interpreted as an efficient form of GNN layer.
In fact, for certain choices of the step size, and with the incorporation of gradient pre-conditioning and other reparameterization factors, these iterations can be exactly reduced to popular canonical GNN layers such as those used by GCN [12], APPNP [7], and others.
However, under broader settings, we obtain unique GNN layers that are naturally immune to oversmoothing, while nonetheless facilitating long-range signal propagation across the graph by virtue of the anchoring effect of the underlying energy function.
In other words, regardless of how large K is, these iterations/layers will converge to node embeddings that adhere to the design criteria of (2).
This is the distinct appeal of GNN layers motivated by unfolded optimization, at least thus far in the case of homogeneous graphs.
New Heterogeneous GNN Layers via Unfolding
In this section, we explore novel extensions of unfolded optimization to handle heterogeneous graphs, which, as mentioned previously, are commonly encountered in numerous practical application domains.
After introducing our adopted notation, we first describe two relatively simple, intuitive attempts to accommodate heterogeneous node and edge types.
We then point out the shortcomings of these models, which are subsequently addressed by our main proposal: a general-purpose energy function and attendant proximal gradient steps that both (i) descend the aforementioned energy, and (ii) in doing so facilitate flexible node- and edge-type dependent message passing with non-linear activations where needed.
The result is a complete, interpretable heterogeneous GNN (HGNN) architecture with layers in one-to-one correspondence with the minimization of a heterogeneous graph-regularized energy function.
Before we begin, we introduce some additional notation.
An undirected heterogeneous graph G =
( V , E ) is a collection of node types S and edge types T such that V =
(cid:83) s ∈S V s and E
= (cid:83) t ∈T E t .
Here V s denotes a set of n s =
|V s | nodes of type s , while E t represents a set of edge type t .
Furthermore, we use T ss ′ to refer to the set of edge types connecting node types s and s ′ .
Note that T includes both the canonical direction, t ∈ T ss ′ , and the inverse direction, t inv ∈ T s ′ s , that corresponds with a type t edge.
3.1 First Attempts at Heterogeneous Unfolding
A natural starting point for extending existing homogeneous models to the heterogeneous case would be to adopt the modified energy
[Equation]
in which L t is the graph Laplacian involving all edges of relation type t and W s parameterizes a type-specific transformation of node representations.
Here, we have simply introduced type-specific transformations of the initial node representations, as well as adding a separate graph regularization term for each relation type t .
However, this formulation is actually quite limited as it can be reduced to the equivalent energy
[Equation]
in which (cid:101) X represents the original node features concatenated with the node type, and (cid:101)
f is some function of the augmented features parameterized by W ≜ { W s } s ∈S .
Note that by construction, C will necessarily be positive semi-definite, and hence we may conclude through (6) that (5) defaults to the form of a standard quadratically-regularized loss, analogous to (2), that has often been applied to graph signal processing problems [11].
Therefore the gradient descent iterations of this objective will closely mirror the existing homogeneous unfolded GNN architectures described in Section 2, and fail to capture the nuances of heterogeneous data.
To break the symmetry that collapses all relation-specific regularization factors into tr (cid:0) Y ⊤ CY (cid:1) , a natural option is to introduce t -dependent weights as in
[Equation]
in which the set { M t } is trainable.
While certainly more expressive than (5), this revision is nonetheless saddled with several key limitations: (i) Unless additional constraints are included on { M t } (e.g., PSD, etc.), (7) may be unbounded from below; (ii) While M t may be asymmetric, w.l.o.g.
the penalty can be equivalently expressed with symmetric weights, and hence the true degrees of freedom are limited; (iii) This model could be prone to overfitting, since during training it could be that M t → 0 in which case the graph-based regularization is turned off; 5 and (iv) Because tr (cid:2) Y ⊤ L t YM t (cid:3)
=
(cid:80) ( i,j ) ∈E t ( y i − y j )
⊤ M t ( y i − y j ) , the energy function (7) is symmetric w.r.t.
the order of the nodes in
the
regularization term, and therefore any derived message passing
must
also
be symmetric, i.
e
we
cannot
exploit asymmetric penalization aligned with heterogeneous relationships in the data
Given then that both (5) and (7) have notable shortcomings, it behooves us to consider revised criteria for selecting a suitable heterogeneous energy function.
We explore such issues next.
3.2 A More Expressive Alternative Energy Function
As an entry point for developing a more flexible class of energy functions that is sensitive to nuanced relationships between different node types, we consider previous work developing various flavors of both label and belief propagation [5, 24, 31, 34].
In the more straightforward setting of homogeneous graphs under homophily conditions, label propagation serves as a simple iterative process for smoothing known training labels across edges to unlabeled nodes in such a way that, for a given node i , the predicted node label ˆ t i will approximately match the labels of neighboring nodes
N i , i.e. , ˆ t
i ≈
ˆ
t j when j ∈ N i .
An analogous relationship holds when labels are replaced with beliefs [24].
However, in broader regimes with varying degrees of heterophily, it is no longer reasonable for such a simple relationship to hold, as neighboring nodes may be more inclined to have dissimilar labels.
To address this mismatch, a compatibility matrix H must be introduced such that now label propagation serves to instantiate ˆ t i
≈
ˆ t j H for j ∈ N i [31] (here we treat each ˆ t j as a row vector).
If each predicted label ˆ t i is approximately a one-hot vector associated with class membership probabilities, then the ( k, l ) -th element of H roughly determines how a node of class k influences neighbors of class l .
And if we further extend to heterogeneous graphs as has been done with various forms of belief propagation [5], it is natural to maintain a unique (possibly non-square) compatibility matrix
H t for each relation type t , with the goal of finding predicted labels (or beliefs) of each node type satisfying ˆ t si ≈ ˆ t s ′
j H t for s, s ′ ∈ S , t ∈ T ss ′ , and ( i,
j ) ∈ E t .
From a practical standpoint, each such H t can be estimated from the data using the statistics of the labels (or beliefs) of nodes sharing edges of type t .
Returning to our original goal, we can use similar intuitions to guide the design of an appropriate regularization factor for learning heterogenerous graph node representations (as required by HGNNs).
Specifically, given s, s ′ ∈ S , t ∈ T ss ′ , and ( i,
j ) ∈
E t , we seek to enforce y si H t
≈ y s ′ j , which then naturally motivates the energy
[Equation]
|
[Equation]
y
[Equation]
H
[Equation]
− y
[Equation]
||  , (8) where λ denotes a trade-off parameter, X s ∈ R n s × d 0 s represents d 0 s -dimensional initial features and Y s ∈ R n s × d s is the embedding of d s -dimensional features on n s nodes for node type s .
Note that y si and y s ′
j are the i -th and j -th row of Y s and Y s ′ , respectively.
Additionally, H t ∈ R d s × d s ′ denotes a compatibility matrix that matches the dimension of two different node embeddings ( i.e. , y si and y s ′ j ) via linear transformation for edge type t .
The energy function (8) exhibits the following two characteristics:
• The first term prefers that the embedding Y s should be close to that from f ( X s ; W s ) .
• The second term prefers that for two nodes of type s and s ′ connection by an edge of type t , the embeddings y si H t and y s ′
j should be relatively close to one another.
Even when the dimensions of two embeddings are the same ( i.
e.
, d s = d s ′ ), H t still serves an important role in allowing the embeddings to lie within different subspaces to obtain compatibility.
3.3 Analysis and Derivation of Corresponding Descent Steps
The objective (8) can be written as a matrix form as follows:
ℓ Y
[Equation]
where A t denotes the adjacency matrix for an edge type t , and D st denotes the type- t degree matrix of type- s nodes.
The closed-form optimal point of (9) is provided by the following result; see the Supplementary
Materials for the proof.
Lemma 3.1.
The unique solution Y ∗ ( W , H ) minimizing (9) satisfies
[Equation]
where vec ( B ) =
[ b ⊤ 1 , ..., b ⊤ n ] ⊤ for matrix B ,
H ≜
{ H t } t ∈T denotes the set of all compatibility matrices, and we have defined the matrices P , Q , and D as
[Equation]
Here ⊗ denotes the Kronecker product, (cid:76) ni
=
A i = diag ( A 1 , ..., A n ) denotes a direct sum of n square matrices
A 1 , ..., A n , and D s ≜ (cid:80) s ′ ∈S (cid:80) t ∈T ss
′ D s ′ t represent a sum of degree matrices over all node types s ′ ∈ S and all edge types t ∈ T ss ′ .
We can interpret this solution as transforming (cid:101) f
( (cid:101) X ; W ) into an embedding that has both local and global information of the graph structure [34].
Therefore, this can be treated as an appropriate graph-aware embedding for carrying out a downstream task such as node classification.
However, for practically-sized graph data, computing the inverse ( I + λ ( Q − P + D )) − 1 in (10) is prohibitively expensive.
To resolve this issue, similar to the homogeneous case we instead apply gradient descent to (9) over Y s for all node types s ∈ S to approximate Y ∗ ( W , H ) .
Starting from the initial point Y (0) s =
f
( X s ; W s ) for each s ∈ S , several steps of gradient descent are performed to obtain an appropriate embedding for carrying out a downstream task.
Since all the intermediate steps of gradient descent are differentiable features w.r.t W and H , we can just plug-in the result of performing K iterations of gradient descent, Y ( K ) s , into the (18).
In order to get Y ( K ) s using gradient descent, we first compute the gradient of (9) w.r.t Y s :
[Equation]
Note, if the condition number of (11) is large, the convergence speed of gradient descent can be slow [18].
To reduce the convergence time effectively, we use the Jacobi preconditioning technique [1], rescaling the gradient step using (cid:101) D − 1 s ≜ ( I + λ D s ) − 1 .
Then iteration k + 1 of gradient descent on
Y s can be computed as Y
[Equation]
(cid:125) ( c ) − D st Y ( k ) s
( H t H ⊤ t ) (cid:124) (cid:123)(cid:122) (cid:125) ( d ) (cid:1)(cid:3) , (12) where α denotes a step size.
The above expression can be considered as a forward propagation rule for the k -th layer in a GNN model.
In (12), the terms (a), (b), (c), and (d) can be interpreted as follows:
Terms (a) & (b)
Each term can be treated as a skip connection from the previous layer Y ( k ) s and the input features f ( X s ; W s ) , respectively.
Unlike some previous work [13, 15] that adopted skip-connections as a heuristic solution to resolve the oversmoothing problem [14, 19], similar to unfolded GNNs in the homogeneous case [7, 32], the skip connections in our expression are naturally derived from the gradient descent step for minimizing the energy function.
Term (c) This term accumulates the propagated feature vectors of neighboring nodes.
The matrix
H ⊤ t + H t inv serves as edge-type specific transformations in a bi-directional way.
Note that all matrices in H are shared across all propagation layers, and as a result, the number of parameters does not increase as the propagation proceeds.
Term (d) This term is a self-loop transformation that depends on the edge type t .
The matrix H t H ⊤ t introduces an edge-type specific transformation in a uni-directional way, and the matrix D st reflects the strength of the self-loop transformation relying on edge type t .
Note that, unlike prior work [15, 26], we use edge-type specific degree and transformation matrices, and these are also shared across all GNN layers.
We reiterate that all components of (12) have naturally emerged from the gradient descent procedure for minimizing the underlying energy function.
And in terms of convergence, we have the following result; again, the proof is deferred to the Supplementary Materials:
Theorem 3.2.
The iterations (12) are guaranteed to monotonically converge to the unique global minimum of (8) provided that
[Equation]
where d min is the minimum diagonal element of D and σ max is a maximum eigenvalue of matrix ( Q − P ) .
Time Complexity Although seemingly complicated, the time complexity of executing (12) once is O
( md + nd 2 ) , where d = max { d s | s ∈ S} .
In this way the complexity is on the same level as
R-GCN [26], a commonly-used heterogeneous GNN variant, and it can be efficiently implemented through off-the-shelf sparse matrix libraries.
3.4 Generalization to Nonlinear Activations
In various message-passing GNNs [8, 12, 15, 26], node-wise ReLU activations are applied to the intermediate features during the forward propagation step.
In our setting, by adding additional regularization or constraints to our original energy function and using proximal gradient descent methods
[21], such nonlinear activations can also be naturally derived without any heuristic modifications on our propagation step.
Similar ideas have been applied to simpler homogeneous GNN models [32].
More specifically, let ϕ s : R d s → R be an arbitrary convex function of node embeddings.
Then our optimization objective can be modified the minimization of
[Equation]
Instead of directly minimizing (14) via vanilla gradient descent, we employ the proximal gradient descent (PGD) method [21] as follows.
We first denote the relevant proximal operator as
[Equation] [Equation]
Then PGD iteratively minimizes (14) by computing
[Equation]
For our purposes, we set ϕ s ( y ) ≜ (cid:80)
d s i
=1 I ∞
[ y i < 0] , in which I ∞ is an indicator function that assigns an infinite penalty to any
y i < 0 .
In this way, the corresponding proximal operator becomes prox ϕ s ( v ) =
ReLU ( v ) = max ( 0 , v ) , in which the maximum is applied to each dimension independently.
3.5 The Overall Algorithm
After performing K iterations of (12) with the proximal operator in (17), we obtain Y ( K ) ( W , H ) , and analogous to (1), the meta-loss function for downstream tasks becomes
[Equation]
18) where S ′ denotes a set of labeled node types, g s : R d s → R c s denotes a node-wise function parameterized by θ s , Θ ≜ { θ s } s ∈S ′ denotes a set of parameters of g s , and t si ∈ R c
s is a groundtruth node-wise target.
Finally, we optimize (18) over W , H , and Θ via typical iterative optimization method such as stochastic gradient descent (SGD).
We dub the model obtained by the above procedure as HALO for H eterogeneous A rchitecture
L everaging O ptimization , and Algorithm 1 summarizes the overall process.
Require:
G : Heterogeneous graph dataset
Require: λ , α : Hyperparamters, K : Number of unfolding steps, E : Number of epochs
Randomly initialize H t ∈ H , W s ∈ W , and θ s ∈ Θ for e = 1 , · · · , E do
Set Y (0) s
= f
( X s ; W s ) for all s ∈ S for k = 0 , · · · , K − 1 do
Compute (16) using (12), and (17) to get Y ( k +1) s for each s ∈ S end for
Compute ℓ Θ (Θ , W , H ) using Y ( K ) s
Optimize all W s ∈ W , H t ∈ H , and θ s ∈ Θ using SGD.
end for
Experiments
We evaluate HALO on node or entity classification tasks from the heterogeneous graph benchmark (HGB) [15] as well as several knowledge graph datasets [26], and compare with state-of-the-art baselines.
Furthermore, we later evaluate w.r.t.
ZooBP under settings that permit direct comparisons, and we carry out additional empirical analyses and ablations.
[Table]
4.1 Node Classification on Benchmark Datasets
The dataset descriptions are as follows:
HGB contains 4 node classification datasets, which is our focus herein.
These include: DBLP, IMDB,
ACM, and Freebase.
DBLP and ACM are citation networks, IMDB is a movie information graph, and Freebase is a large knowledge graph.
Except for Freebase, all datasets have node features (for
Freebase, we set the node types as the node features).
For more details, please refer to [15].
The knowledge graph benchmarking proposed in [26] is composed of 4 datasets: AIFB, MUTAG,
BGS, AM.
The task is to classify the entities of target nodes.
AIFB and MUTAG are relatively small-scale knowledge graphs, while BGS and AM are larger scale, e.g. , the AM dataset has more than 1M entities.
For more details, please refer to [26].
In every experiment, we chose f ( X s ; W s ) and
g s ( y s , θ s ) as linear functions for all s ∈ S , and when the dimension d s 0 is different across node types, f ( X s ; W s ) is naturally constructed to align the dimensions of each feature.
All models and experiments were implemented using PyTorch [22] and the Deep Graph Library (DGL) [29].
For the details on the hyperparameters and other experimental settings, please see the Supplementary Materials.
Table 1 shows the results on node classification tasks, where the best performance is highlighted in bold.
The left side of the table shows the results on HGB, including baselines R-GCN [26], HAN [30],
HGT [10], and Simple-HGN [15].
The right side shows the analogous results on the knowledge graph datasets, where we also include comparisons with R-GCN [26], hand-designed feature extractors (Feat) [23], Weisfeiler-Lehman kernels (WL), and RDF2Vec embeddings [25].
The Avg. column in each table represents the average of the results across the four datasets.
In the table, HALO achieves the highest average accuracy, and outperforms the baselines on 7 of 8 datasets.
4.2 Comparison with ZooBP
We next compare HALO with ZooBP [5], which provided motivation for our use of compatibility matrices and represents a natural baseline to evaluate against when possible.
ZooBP is a belief propagation method for heterogeneous graphs, and therefore, the number of classes per node type should be pre-defined for every node.
Consequently, running ZooBP on HGB or the knowledge graph datasets in which only a single node type has labels (and therefore defined classes), e.g. , DBLP, is not feasible.
To that end, to compare with ZooBP, we modified the DBLP and Academic [28] datasets so that all node types have labels.
Namely, in both graphs, we removed the “Term” and “Venue” node types and only utilized the “Paper” and “Author” node types.
Moreover, since the original datasets only have the class labels for the “Author” type, we set the class labels for the “Paper” type as the corresponding “Venue” of the paper. Thus, the 20 and 18 Venues (specified in the
Supplementary Materials) become the class labels for the “Paper” nodes in DBLP and Academic, respectively. Furthermore, we also considered a simpler setting in which the number of class labels for the “Paper” nodes (denoted as k ) is four, by using the 4 categories ( i.e., ML, DB, DM, IR ) of the Venues as the labels for the nodes.
We note the number of class labels for the “Author” nodes was always set to 4 (corresponding to the Venue categories), and the resulting datasets are denoted as DBLP-reduced and Academic-reduced, respectively 7 .
For choosing the fixed compatibility matrix for ZooBP (as a label propagation-like method, ZooBP has no mechanism for learning), when k = 4 , we set H t
=
I k × k − 14 1 k × k , where 1 k is a k × k square matrix in which all elements are 1.
When k = 18 or k = 20 , we set the matrix considering the correlation between the “Author” and “Paper” labels. For example, for the “ AAAI ” Paper class, we set the matrix element value corresponding to “ ML ” Author class higher than other Author classes, since [Table] “ AAAI ” venue and “ ML ” category are highly correlated.
For further details about reducing the number of classes for the “Paper” nodes, setting the compatibility matrices, and the exact compatibility matrices that we used, please refer to the Supplementary Materials.
Moreover, for setting the output layer g s ( y s , θ s ) in our case, we set g s ( y s , θ s ) as an identity function, i.e. , θ s =
I , and we treat y s as a score vector on each class.
Table 2 shows the results of ZooBP and HALO on the DBLP-bipartite and Academic datasets as described above.
We also report the itemized classification accuracy on “Author” and “Paper” node types.
From these results, we observe that HALO outperforms ZooBP in all settings.
For large k , the gap is generally more significant.
The main reason for this phenomenon is that while HALO utilizes both node features and trainable compatibility matrices, ZooBP only uses training labels with fixed compatibility matrices.
To the best of our knowledge, HALO is the only method designed to exploit a trainable compatibility matrix.
[Table]
[Table]
[Figure]
4.3 Further HALO Analysis
In this section, we first carry out an ablation study to assess the importance of learnable compatibility matrices H t ∈ H , as well as the value of non-linear graph propagation operators instantiated through
PGD.
Later, we also show the accuracy and energy values of HALO versus the number of propagation steps to demonstrate convergence of the lower-level HALO optimization process, which naturally mitigates oversmoothing.
Ablation Study To examine the effectiveness of trainable compatibility matrices, we enforce H t
= I for all t , using the same embedding dimension across all node types.
Other training conditions remain unchanged.
Table 3 compares the resulting accuracy on DBLP, MUTAG, and BGS datasets, where we observe that HALO performs significantly better, especially on MUTAG and BGS.
These results show that using trainable compatibility matrices to quantify ambiguous relationships between neighboring nodes can be an effective strategy when performing node classification with heterogeneous graphs.
Note that even when the embedding dimensions are same (as in these experiments), there can still be relation-specific mismatches between different node types that exhibit heterophily characteristics, which trainable compatibility matrices can help to resolve.
Turning to the non-linear activations, we also re-run the HALO experiments with (17) removed from the algorithm, meaning PGD defaults to regular SGD as no proximal step is used.
As shown in Table 3, HALO without PGD performs considerably worse, implying that the inclusion of the additional non-linear activation leads to better generalization.
Results Varying Propagation Steps Figure 1 shows the results of HALO with various numbers of propagation steps.
Specifically, the left plot shows the training and validation accuracy on the
ACM dataset, and the right plot shows the energy function value (9) during the unfolding steps.
Notably, when we increase the number of propagation steps with a suitable α , both training and validation accuracies are gradually increasing, and the energy function value steadily decreasing as expected.
These results suggest that as the gradient descent steps gradually decrease the low-level energy function, the corresponding node embeddings Y s that emerge are effective in performing the downstream classification task.
We have also repeated this experiment using the AIFB dataset and the overall trend is same; for more details please refer to the Supplementary Materials.
Conclusion
In this paper, we have proposed HALO, a novel HGNN architecture derived from the minimization steps of a relation-aware energy function, the latter consisting of trainable relation-dependent compatibility matrices that can resolve the mismatch between two different node types.
Because of intrinsic properties of this unfolding framework, HALO is naturally robust to oversmoothing problems, and outperforms SOTA models on various benchmark datasets.
For future work, extending
HALO to spatio-temporal graphs in which the node and edge information change continuously can be considered.
Acknowledgments
This work was supported in part by New Faculty Startup Fund from Seoul National University, NRF grants [NRF-2021R1A2C2007884, NRF-2021M3E5D2A01024795] and IITP grants [No.2021-001343, No.2021-0-02068, No.2022-0-00959, No.2022-0-00113] funded by the Korean government, and SNU-NAVER Hyperscale AI Center.