Chen, S.; Wang, J.; Li, G. Neural Relational Inference With Efficient Message Passing Mechanisms. AAAI 2021, 35, 7055-7063.