Researchers apply developmental psychology to AI model that predicts object relationships