Learning invariances via correlation and marginal alignments for out-of-distribution generalization | Synapse