Chu X, Jin Y, Wang X, Zhang S, Wang Y, Zhu W, Mei H. Wasserstein barycenter matching for graph size generalization of message passing neural networks, in International Conference on Machine Learning. PMLR; 2023:6158–6184.