Pixel-wise MSE loss