Attention schema in neural agents
Open accessAbstract
Attention has become a common ingredient in deep learning architectures. It adds a dynamical selection of information on top of the static selection of information supported by weights. In the same way, we can imagine a higher-order informational filter built on top of attention: an Attention Schema (AS), namely, a descriptive and predictive model of attention. In cognitive neuroscience, Attention Schema Theory (AST) supports this idea of distinguishing attention from AS. A strong prediction of this theory is that an agent can use its own AS to also infer the states of other agents' attention and consequently enhance coordination with other agents. As such, multi-agent reinforcement learning would be an ideal setting to experimentally test the validity of AST. We explore different ways in which attention and AS interact with each other. Our preliminary results indicate that agents that implement the AS as a recurrent internal control achieve the best performance. In general, these exploratory experiments suggest that equipping artificial agents with a model of attention can enhance their social intelligence.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
Liu et al.'s system performed better than the others with different architectures made up of the same components on the multi-agent RL tasks.
“This system performed better than the others, with different architectures made up of the same components, which were tested on the multi-agent RL tasks.”
From Consciousness in Artificial Intelligence: Insights from the Science of Consciousness (2023) · cites this paper as Liu et al. (2023)
Section 6.1 reports that the attention-schema architecture achieved the best performance among five architectures in the GhostRun multi-agent reinforcement-learning task; Section 6.3 reports the best performance for its 5.4 variant in Maze-Cleaners. The claim that the compared architectures were all made up of the same components is too broad: Section 4.1 says components were kept the same where shared, while Section 5 describes variants that remove or rearrange components. (LLM, conf=0.83)
Experimental — accuracy not yet measuredRecent studies have used AI to test predictions of AST and GWT.
“For example, recent studies have used AI to test predictions of AST [ 75,76] and GWT [ 77,78].”
From Identifying indicators of consciousness in AI systems (2026) · cites this paper as [75]
Attention schema-like models which enable the effective control of attention could be valuable for the increasing number of AI systems which employ attention.
“Attention schema-like models which enable the effective control of attention could be valuable for the increasing number of AI systems which employ attention-here understood as active control of information flow (Liu et al. 2023).”
From Consciousness in Artificial Intelligence: Insights from the Science of Consciousness (2023) · cites this paper as Liu et al. 2023
Two notable studies have implemented simple attention schemas in artificial systems.
“Two notable studies have implemented simple attention schemas in artificial systems.”
From Consciousness in Artificial Intelligence: Insights from the Science of Consciousness (2023) · cites this paper as Liu et al. (2023)
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Kaiser, L., & Polosukhin, I. (2017) · arXiv · doi:10.48550/arxiv.1706.03762 · Cited by 120,404 · cited 2× · see citing works
“Attention modules are implemented as multi-head attention layers similar to the kind implemented by Transformers (Vaswani et al., 2017); they dynamically select information from the observation space or, in the case of hypotheses 2 and 4, from the output of the internal control module.”
BackgroundMethodPyTorch: An Imperative Style, High-Performance Deep Learning Library
Paszke, A., Gross, S., Massa, F., Lerer, A., Bradbury, J., Chanan, G., Killeen, T., Lin, Z., Gimelshein, N., Antiga, L., et al. (2019) · arXiv · doi:10.48550/arxiv.1912.01703 · Cited by 31,473 · see citing works
An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., Gelly, S., et al. (2020) · arXiv · doi:10.48550/arxiv.2010.11929 · Cited by 21,793 · cited 1× · see citing works
“Transformers have particularly demonstrated how attention may be all we need , from sequence learning (Vaswani et al., 2017) to visual processing (Dosovitskiy et al., 2020) and time series forecasting (Lim et al., 2021).”
BackgroundNeural Machine Translation by Jointly Learning to Align and Translate
Bahdanau, D., Cho, K., & Bengio, Y. (2014) · arXiv · doi:10.48550/arxiv.1409.0473 · Cited by 16,213 · cited 1× · see citing works
“In the last decade, the scope of attention mechanisms has grown from the first implementation in RNNSearch (Bahdanau et al., 2014) to the recent large-scale models currently dominating natural language processing and text-to-image generation (Santana & Colombini, 2021).”
BackgroundSelf-citationA feature-integration theory of attention.
Treisman, A. M., & Gelade, G. (1980) · Cognitive Psychology · doi:10.1016/0010-0285(80)90005-5 · Cited by 12,366 · cited 1× · see citing works
“…of a representation, which makes it impossible for the representation of a stimulus to be bound to the information contained in its internal control (Treisman & Gelade, 1980) b) without attention, the representation of a stimulus has lower signal strength and is less likely to have an influence on…”
BackgroundCategorical Reparameterization with Gumbel-Softmax
Jang, E., Gu, S., & Poole, B. (2016) · arXiv · doi:10.48550/arxiv.1611.01144 · Cited by 3,616 · cited 1× · see citing works
“The mask is generated from a pair of MLPs inspired by a neural activator and suppressor and followed by a Gumbel softmax to binarize the value pairs (Jang et al., 2017), where the activator and suppressor work together to generate a binary mask on attention scores, in which a ”1” indicates that the…”
MethodPytorch (1998) · Trends in Cognitive Sciences · doi:10.1016/s1364-6613(97)82741-6 · Cited by 3,307
Temporal Fusion Transformers for interpretable multi-horizon time series forecasting
Lim, B., Arık, S. Ö., Loeff, N., & Pfister, T. (2021) · International Journal of Forecasting · doi:10.1016/j.ijforecast.2021.03.012 · Cited by 3,291 · cited 1× · see citing works
“Transformers have particularly demonstrated how attention may be all we need , from sequence learning (Vaswani et al., 2017) to visual processing (Dosovitskiy et al., 2020) and time series forecasting (Lim et al., 2021).”
BackgroundLearning to Communicate with Deep Multi-Agent Reinforcement Learning
Foerster, J. N., Assael, Y. M., de Freitas, N., & Whiteson, S. (2016) · arXiv · doi:10.48550/arxiv.1605.06676 · Cited by 1,217 · cited 1× · see citing works
“…the state of other agents’ attention will also be useful for settings in which modeling opponents (Albrecht et al., 2017; He et al., 2016), value alignment (Fisac et al., 2020; Hadfield-Menell et al., 2016) and communication (OroojlooyJadid & Hajinezhad, 2021; Foerster et al., 2016) play a key role.”
BackgroundEvery good regulator of a system must be a model of that system †
CONANT, R. C., & ROSS ASHBY, W. (1970) · International Journal of Systems Science · doi:10.1080/00207727008920220 · Cited by 1,063 · see citing works
Every Good Regulator of a System Must Be a Model of That System
Conant, R. C., & Ashby, W. R. (1991) · Facets of Systems Science · doi:10.1007/978-1-4899-0718-9_37 · Cited by 712 · see citing works
Actor-Attention-Critic for Multi-Agent Reinforcement LearningPreprint
Iqbal, S., & Sha, F. (2018) · arXiv (Cornell University) · doi:10.48550/arxiv.1810.02912 · Cited by 629 · cited 1× · see citing works
“Shared central critics (Iqbal & Sha, 2019) and joint attention (Lee et al., 2021) have been shown to promote robust and scalable learning in complex environments.”
BackgroundA review of cooperative multi-agent deep reinforcement learning
Oroojlooy, A., & Hajinezhad, D. (2022) · Applied Intelligence · doi:10.1007/s10489-022-04105-y · Cited by 607 · cited 1× · see citing works
“…the state of other agents’ attention will also be useful for settings in which modeling opponents (Albrecht et al., 2017; He et al., 2016), value alignment (Fisac et al., 2020; Hadfield-Menell et al., 2016) and communication (OroojlooyJadid & Hajinezhad, 2021; Foerster et al., 2016) play a key role.”
BackgroundParaskeva, C., & Williams, A. C. (1990) · British Journal of Cancer · doi:10.1038/bjc.1990.185 · Cited by 524
Emergence of grounded compositional language in multi-agent populations
Mordatch, I., & Abbeel, P. (2018) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v32i1.11492 · Cited by 440 · cited 1× · see citing works
“The five hypotheses were implemented using PyTorch (Paszke et al., 2019), and tested on two tasks from two different MARL benchmarks: the GhostRun environment (Jiang, 2019) and the Multi-Agent Particle environment (Mordatch & Abbeel, 2017).”
MethodCooperative Inverse Reinforcement Learning
Hadfield-Menell, D., Dragan, A., Abbeel, P., & Russell, S. (2016) · arXiv · doi:10.48550/arxiv.1606.03137 · Cited by 413 · cited 1× · see citing works
“…the state of other agents’ attention will also be useful for settings in which modeling opponents (Albrecht et al., 2017; He et al., 2016), value alignment (Fisac et al., 2020; Hadfield-Menell et al., 2016) and communication (OroojlooyJadid & Hajinezhad, 2021; Foerster et al., 2016) play a key role.”
BackgroundMachine Theory of MindPreprint
Rabinowitz, N. C., Perbet, F., Song, H. F., Zhang, C., Eslami, S. M. A., & Botvinick, M. (2018) · arXiv (Cornell University) · doi:10.48550/arxiv.1802.07740 · Cited by 259 · cited 1× · see citing works
“With the theory of mind, the agent generically attributes mental states to other agents, including their desires, beliefs, and intentions (Rabinowitz et al., 2018).”
BackgroundThe attention schema theory: a mechanistic account of subjective awareness
Graziano, M. S. A., & Webb, T. W. (2015) · Frontiers in Psychology · doi:10.3389/fpsyg.2015.00500 · Cited by 224 · cited 5× · see citing works
“In humans, there is evidence of cases where attention unfolds without its internal control being active, but this seems to correlate with poorer performance (Graziano & Webb, 2015b).”
BackgroundMethodInfluentialHuman consciousness and its relationship to social neuroscience: A novel hypothesis
Graziano, M. S. A., & Kastner, S. (2011) · Cognitive Neuroscience · doi:10.1080/17588928.2011.565121 · Cited by 195 · see citing works
Opponent Modeling in Deep Reinforcement LearningPreprint
He, H., Boyd-Graber, J. L., Kwok, K., & Daumé, H. (2016) · arXiv (Cornell University) · doi:10.48550/arxiv.1609.05559 · Cited by 156 · cited 2× · see citing works
“…attention schema to infer the state of other agents’ attention will also be useful for settings in which modeling opponents (Albrecht et al., 2017; He et al., 2016), value alignment (Fisac et al., 2020; Hadfield-Menell et al., 2016) and communication (OroojlooyJadid & Hajinezhad, 2021; Foerster et…”
BackgroundThe Attention Schema Theory: A Foundation for Engineering Artificial Consciousness
Graziano, M. S. A. (2017) · Frontiers in Robotics and AI · doi:10.3389/frobt.2017.00060 · Cited by 79 · cited 2× · see citing works
“…a descriptive and predictive model of attention (referred to as ”Internal Control” in Figure 1) could help the dynamical control of attention and therefore maximize the efficiency with which resources are strategically devoted to different elements of an ever-changing environment (Graziano, 2017).”
BackgroundMittal, S., Lamb, A., Goyal, A., et al. (2020) · International Conference on Machine Learning · doi:10.48550/arxiv.2006.16981 · Cited by 76 · cited 1×
“In deep learning, attention can be understood as a dynamical control of information flow (Mittal et al., 2020).”
BackgroundSelf-citationPragmatic-Pedagogic Value Alignment
Fisac, J. F., Gates, M. A., Hamrick, J. B., Liu, C., Hadfield-Menell, D., Palaniappan, M., Malik, D., Sastry, S., Griffiths, T. L., & Dragan, A. D. (2019) · Springer proceedings in advanced robotics · doi:10.1007/978-3-030-28619-4_7 · Cited by 56 · see citing works
The Distracting Control Suite -- A Challenging Benchmark for Reinforcement Learning from Pixels
Stone, A., Ramirez, O., Konolige, K., & Jonschkowski, R. (2021) · arXiv · doi:10.48550/arxiv.2101.02722 · Cited by 31 · cited 1× · see citing works
“Learning a rich and schematic model of one’s own attention will lead to better decision making, especially in contexts where there are many distractions and deep focus is hard to achieve (Stone et al., 2021).”
BackgroundSocial Neuro AI: Social Interaction as the “Dark Matter” of AI
Bolotta, S., & Dumas, G. (2022) · Frontiers in Computer Science · doi:10.3389/fcomp.2022.846440 · Cited by 31 · cited 1× · see citing works
“The long-term plan is to test those architectures in richer environments, from simple 3D environments to photorealistic video game engines, and finally with robots (Bolotta & Dumas, 2022).”
BackgroundSelf-citationWilterson, A. I., & Graziano, M. S. A. (2021) · Proceedings of the National Academy of Sciences · doi:10.1073/pnas.2102421118 · Cited by 26 · cited 2× · see citing works
“In fact, the performance of an artificial agent in solving a simple sensorimotor task has been shown to be greatly enhanced by a simple model of attention, but greatly reduced when such a model was not available (Wilterson & Graziano, 2021).”
BackgroundMethodNeural Attention Models in Deep Learning: Survey and Taxonomy
Correia, A. D. S., & Colombini, E. (2021) · arXiv.org · doi:10.48550/arxiv.2112.05909 · Cited by 24
Special issue on multiagent interaction without prior coordination: guest editorial
Albrecht, S. V., Liemhetcharat, S., & Stone, P. (2016) · Autonomous Agents and Multi-Agent Systems · doi:10.1007/s10458-016-9358-0 · Cited by 16 · cited 1×
“Using an attention schema to infer the state of other agents’ attention will also be useful for settings in which modeling opponents (Albrecht et al., 2017; He et al., 2016), value alignment (Fisac et al., 2020; Hadfield-Menell et al., 2016) and communication (OroojlooyJadid & Hajinezhad, 2021;…”
Joint Attention for Multi-Agent Coordination and Social LearningPreprint
Lee, D. M., Jaques, N., Kew, J. C., Wu, J., Eck, D., Schuurmans, D., & Faust, A. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2104.07750 · Cited by 6 · cited 1× · see citing works
“Shared central critics (Iqbal & Sha, 2019) and joint attention (Lee et al., 2021) have been shown to promote robust and scalable learning in complex environments.”
BackgroundWhat is it like to be a human?
Leopold, D. A. (2011) · Cognitive Neuroscience · doi:10.1080/17588928.2011.585235 · cited 1×
“The second prediction is that an attention schema is also useful to model the attention of other agents —since the machinery that computes information about the attention of other people is the same machinery that computes information about our own attention (Graziano & Kastner, 2011).”
Background
Flagged first, then by citations
1 of the 31 references couldn’t be matched to a known paper.
Research works printed but not linked: 6
- Curran As- sociates, Inc., 2016. URL https://proceedings. neurips.cc/paper/2016/hash/ c7635bfd99248a2cdef8249ef7bfbef4-Abstract. html.
- Graziano, M. S. and Webb, T. W. The attention schema theory: a mechanistic account of subjective awareness. Frontiers in psychology, pp. 500, 2015a.
- PMLR, May 2019. URL https://proceedings. mlr.press/v97/iqbal19a.html.
- PMLR, 2020.
- Moerland, T. M., Broekens, J., and Jonker, C. M. Model-based reinforcement learning: A survey. arXiv:2006.16712 [cs, stat], Feb 2021. URL http://arxiv.org/abs/2006.16712. arXiv: 2006.16712. · arXiv
- Webb, T. W. and Graziano, M. S. The attention schema theory: a mechanistic account of subjective awareness. Frontiers in psychology, 6:500, 2015. URL https:// doi.org/10.3389/fpsyg.2015.00500. · DOI · Paper page
Cited by
Sorted by citations
Consciousness in Artificial Intelligence: Insights from the Science of Consciousness
Butlin, P., Long, R., Elmoznino, E., Bengio, Y., Birch, J., Constant, A., Deane, G., Fleming, S. M., Frith, C., Ji, X., et al. (2023) · arXiv · doi:10.48550/arxiv.2308.08708 · Cited by 196 · see citing works
Self-citationIdentifying indicators of consciousness in AI systems
Butlin, P., Long, R., Bayne, T., Bengio, Y., Birch, J., Chalmers, D., Constant, A., Deane, G., Elmoznino, E., Fleming, S. M., et al. (2026) · Trends in Cognitive Sciences · doi:10.1016/j.tics.2025.10.011 · Cited by 56 · see citing works
Self-citation
- Authors
- Liu, D., Bolotta, S., Zhu, H., Bengio, Y., & Dumas, G.
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2023
- ISSN
- 2331-8422 (ISSN-L)
- arXiv
- 2305.17375
- OpenAlex
- W4378770540
- Semantic Scholar
- 72a77555c6723b7ec2a7e4d1d36ef25107a8a321
- Cited by
- 1
- References
- 31
- Citations by year
- 2023: 1
- Type
- posted content
- Language
- en
- Open access
- green
- arxiv.org
- Licence
- Locations
- arXiv (Cornell University) (submittedVersion, cc-by)
- arXiv (Cornell University) (cc-by)
- Funders
- Microsoft Research
- DeepMind
- Topics
- Physical Sciences › Computer Science › Artificial Intelligence › Psychiatry, Mental Health, Neuroscience
- Physical Sciences › Computer Science › Artificial Intelligence › Reinforcement Learning in Robotics
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Visual Attention and Saliency Detection
- Keywords
- attention schema theory, attention, multi-agent reinforcement learning, social intelligence, agent coordination
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review