“…These models assume each context acts as a pointer to a latent structure, and generalizing task statistics requires inference over which structure the current context belongs to. This form of Bayesian non-parametric clustering and generalization can be approximately implemented in corticostriatal gating networks endowed with hierarchical structure (Collins & Frank, 2013) and have been used to explain human generalization behavior and neural correlates thereof in a number of reinforcement learning tasks Collins, 2017;Collins, Cavanagh, & Frank, 2014;Collins & Frank, 2013, 2016Schulz, Franklin, & Gershman, 2018;Tomov, Dorfman, & Gershman, 2018).…”