Visual Interaction Networks
Inferring mass in complex scenes by mental simulation