Wasserstein barycenter matching for graph size generalization of message passing neural networks - Chu, Xu; Jin, Yujie; Xin Wang; Zhang, Shanghang; Wang, Yasha; Zhu, Wenwu; Mei, Hong (International Conference on Machine Learning, 2023)