Memory guides behavior across widely varying environments and must therefore be both sufficiently specific and general. A memory too specific will be useless in even a slightly different environment, while an overly general memory may lead to suboptimal choices. Animals successfully learn to both distinguish between very similar stimuli and generalize across cues. Rather than forming memories that strike a balance between specificity and generality, Drosophila can flexibly categorize a given stimulus into different groups depending on the options available. We asked how this flexibility manifests itself in the well-characterized learning and memory pathways of the fruit fly. We show that flexible categorization in neuronal activity as well as behavior depends on the order and identity of the perceived stimuli. Our results identify the neural correlates of flexible stimulus-categorization in the fruit fly.

The ability to discriminate sensory stimuli with overlapping features is thought to arise in brain structures called expansion layers, where neurons carrying information about sensory features make combinatorial connections onto a much larger set of cells. For 50 years, expansion coding has been a prime topic of theoretical neuroscience, which seeks to explain how quantitative parameters of the expansion circuit influence sensory sensitivity, discrimination, and generalization. Here, we investigate the developmental events that produce the quantitative parameters of the arthropod expansion layer, called the mushroom body. Using Drosophila melanogaster as a model, we employ genetic and chemical tools to engineer changes to circuit development. These allow us to produce living animals with hypothesis-driven variations on natural expansion layer wiring parameters. We then test the functional and behavioral consequences. By altering the number of expansion layer neurons (Kenyon cells) and their dendritic complexity, we find that input density, but not cell number, tunes neuronal odor selectivity. Simple odor discrimination behavior is maintained when the Kenyon cell number is reduced and augmented by Kenyon cell number expansion. Animals with increased input density to each Kenyon cell show increased overlap in Kenyon cell odor responses and become worse at odor discrimination tasks.

Sparse coding is thought to improve discrimination of sensory stimuli by reducing overlap between their representations. Two factors, however, can offset sparse coding's advantages. Similar sensory stimuli have significant overlap, and responses vary across trials. To elucidate the effect of these two factors, we analyzed odor responses in the fly and mouse olfactory regions implicated in learning and discrimination --- the Mushroom Body (MB) and the Piriform Cortex (PCx). In both species, we show that neuronal responses fall along a continuum from extremely reliable across trials to extremely variable or stochastic. Computationally, we show that the range of observed variability arises from probabilistic synapses in inhibitory feedback connections within central circuits rather than sensory noise, as is traditionally assumed. We propose this coding scheme to be advantageous for coarse- and fine-odor discrimination. More reliable cells enable quick discrimination between dissimilar odors. For similar odors, however, these cells overlap, and do not provide distinguishing information. By contrast, more unreliable cells are decorrelated for similar odors, providing distinguishing information, though this requires extended training with more trials. Overall, we have uncovered a stochastic coding scheme that is conserved in vertebrates and invertebrates, and we identify a candidate mechanism, based on variability in a winner-take-all inhibitory circuit, that improves discrimination with training.