Contextual inference through flexible integration of environmental features and behavioural outcomes

The ability to use the context we are in to flexibly adjust our decision-making is vital for navigating a complex world. To do this, the brain must i) use environmental features and behavioural outcomes to distinguish between different, often hidden contexts; and ii) learn how to use these inferred contexts to guide behaviour. However, how these two interacting processes can be performed simultaneously remains unclear. Within the brain it is thought that interaction between the prefrontal cortex (PFC) and hippocampus (HPC) supports contextual inference. We show that models of the HPC using environmental features readily support context-specific behaviour, but struggle to differentiate ambiguous contexts during learning. In contrast, models of the PFC using behavioural outcomes can stably differentiate contexts during periods of learning, but struggle to guide context-specific behaviour. Supporting feature-based with outcome-based strategies during learning overcomes the limitations of both approaches, allowing for the formation of distinct contextual representations that support contextual inference. Moreover, agents using this joint approach reproduce both behavioural- and cellular-level phenomena associated with the interaction between PFC and HPC. Together, these results provide insight into how the brain uses contextual information to guide flexible behaviour.