Cognitively Inspired Energy-Based World Models