Task representations in neural networks trained to perform many cognitive tasks
Task representations in neural networks trained to perform many cognitive tasks
复制标题
DOI:
10.1038/s41593-018-0310-2
复制
发表时间:
2019-02-01
影响因子:
25
通讯作者:
Wang, Xiao-Jing
中科院分区:
文献类型:
--
作者:
Yang, Guangyu Robert;Joglekar, Madhura R.;Wang, Xiao-Jing
The brain has the ability to flexibly perform many tasks, but the underlying mechanism cannot be elucidated in traditional experimental and modeling studies designed for one task at a time. Here, we trained single network models to perform 20 cognitive tasks that depend on working memory, decision making, categorization, and inhibitory control. We found that after training, recurrent units can develop into clusters that are functionally specialized for different cognitive processes, and we introduce a simple yet effective measure to quantify relationships between single-unit neural representations of tasks. Learning often gives rise to compositionality of task representations, a critical feature for cognitive flexibility, whereby one task can be performed by recombining instructions for other tasks. Finally, networks developed mixed task selectivity similar to recorded prefrontal neurons after learning multiple tasks sequentially with a continual-learning technique. This work provides a computational platform to investigate neural representations of many cognitive tasks.