I think maybe CLEVR[0] dataset is what you are talking about?

Keep in mind that a most of the current ML systems have diverged from biology. A majority of the recent breakthroughs come from mathematics, the rational is that just because human brain does it in a certain way does not necessarily mean it is the only way to do it.

[0] https://cs.stanford.edu/people/jcjohns/clevr/