Towards Better Generalization with Flexible Representation of Multi-Module Graph Neural Networks