Correlations of non-affine displacements in metallic glasses through the yield transition