where \(\mathbf {W}\)  is the learnable weight, and the MLPs following the concatenate decoder [1]}, [2]} (short as ConcatMLP decoder):
\(s_{ij} = \mathrm {MLP}(\mathbf {h}_i \Vert  \mathbf {h}_j)\) 
