Neural Relational Inference with Efficient Message Passing Mechanisms