This was one of my first thoughts as well, sort of. Alternatively, is it possible to model each neuron when there are only around 8k? Maybe estimate the visual, audio, and tactile bandwidth of the wasp and feed the simulated neurons a stream of simulated environmental data and just let the connections evolve.
There are not many neurons, but there are many (too many) possible connections. To find the right ones, nature took several billion years. Even with the fastest computers, it would take a considerable amount of time (note that the lifetime of these wasps is a few weeks at most, so evolution happens very fast).
