Skip to main content
. Author manuscript; available in PMC: 2023 Aug 1.
Published in final edited form as: IEEE Trans Med Imaging. 2022 Aug 1;41(8):1990–2003. doi: 10.1109/TMI.2022.3153322

Algorithm 1.

ImageGCN batch propagation algorithm.

Require:
graph node set V and the mini-batch B;
relation adjacency matrix Ar, rR;
input image features X (can be stored externally);
network depth K;
number of neighbors to sample for each node and each relation n.
Ensure: The representation of all samples in B
1: B(K)B
2: for k = K · · · 1 do
3: B(k−1)B(k)
4: for rR do
5:   for vB(k) do
6:    Nr (v) ← the neighbor set of v based on Ar
7:    n ← min(n, |Nr(v)|)
8:    S ← random n samples from Nr(v) without replacement
9:    B(k−1)B(k−1)S
10:   end for
11: end for
12: end for
13: load the features H(0) for B(0) from X
14: for rR do
15: ArB ← the sub-matrix corresponding to B(0) in the adjacency matrix Ar
16: end for
17: execute Eq. 8 with Ar=ArB, for k = 0, · · ·, K − 1
18: extract the representations of samples in B from H(K)