Schematic architecture of NezNet. The input features are first processed by a dense network. Message passing between the two layers through Eq. (1) is then applied to take the relative differences and global values of the features into account. Before the final dense layer, the features are summed and then reprocessed with an MLP to output the score probability of two galaxies being actual neighbours.