An individual's readiness to switch tasks (cognitive flexibility) varies over time, in part, as the result of reinforcement learning based on the statistical structure of the world around them. Consequently, the behavioral cost associated with task-switching is smaller in contexts where switching is frequent than where it is rare, but the underlying brain mechanisms of this adaptation in cognitive flexibility are not well understood. Here, we manipulated the likelihood of switches across blocks of trials in a classic cued task-switching paradigm while participants underwent fMRI. As anticipated, behavioral switch costs decreased as the probability of switching increased, and neural switch costs were observed in lateral and medial frontoparietal cortex. To study moment-by-moment adjustments in cognitive flexibility at the neural level, we first fitted the behavioral RT data with reinforcement learning algorithms and then used the resulting trial-wise prediction error estimate as a regressor in a model-based fMRI analysis. The results revealed that lateral frontal and parietal cortex activity scaled positively with unsigned switch prediction error and that there were no brain regions encoding signed (i.e., switch- or repeat-specific) prediction error. Taken together, this study documents that adjustments in cognitive flexibility to time-varying switch demands are mediated by frontoparietal cortex tracking the likelihood of forthcoming task switches.

Individuals are able to adjust their readiness to shift spatial attention, referred to as “attentional flexibility,” according to the changing demands of the environment, but the neural mechanisms underlying learned adjustments in flexibility are unknown. In the current study, we used fMRI to identify the brain structures responsible for learning shift likelihood. Participants were cued to covertly hold or shift attention among continuous streams of alphanumeric characters and to indicate the parity of target stimuli. Unbeknown to the participants, the stream locations were predictive of the likelihood of having to shift (or hold) attention. Participants adapted their attentional flexibility according to contextual demands, such that the RT cost associated with shifting attention was smallest when shift cues were most likely. Learning model-derived shift prediction error scaled positively with activity within dorsal and ventral frontoparietal regions, documenting that these regions track and update shift likelihood. A complementary inverted encoding model analysis revealed that the pretrial difference in attentional selection strength between to-be-attended and to-be-ignored locations did not change with increasing shift likelihood. The behavioral improvement associated with learned flexibility may primarily arise from a speeding of the shift process rather than from preparatory broadening of attentional selection.