Learning to design protein-protein interactions with enhanced generalization