TLDR: A new AI method, “attention-aware inverse planning,” infers human attentional biases from actions, addressing the flaw in traditional AI that assumes perfect rationality. Tested in simulated driving, it accurately recovers biases and outperforms standard inverse reinforcement learning, paving the way for AI systems that better understand and interact with humans.
In the realm of artificial intelligence, particularly when designing systems that interact with people, a fundamental challenge arises: humans are not always perfectly rational. Our decisions and actions are often influenced by cognitive biases, especially how we pay attention to our surroundings. This reality often clashes with traditional AI models, which typically assume people act optimally to achieve their goals.
A recent research paper, “Estimating cognitive biases with attention-aware inverse planning,” by Sounak Banerjee, Daphne Cornelisse, Deepak Gopinath, Emily Sumner, Jonathan DeCastro, Guy Rosman, Eugene Vinitsky, and Mark K. Ho, tackles this crucial disconnect. The authors introduce a novel approach called attention-aware inverse planning, designed to estimate a person’s attentional biases directly from their observed actions. This work is vital for creating autonomous systems that can better understand and anticipate human behavior in complex, real-world scenarios like driving.
The Problem with Perfect Rationality
For decades, cognitive science has shown that human behavior deviates from ideal rationality. Yet, many machine learning models, such as those based on inverse reinforcement learning (IRL), operate under the assumption that people act optimally. While this assumption simplifies algorithm development and allows for generalizable predictions, it overlooks the nuanced ways in which human cognition is limited, particularly by attention.
Consider driving: a person must constantly decide which objects to focus on (e.g., the car ahead, a child entering a blind spot) and which to disregard (e.g., a parked car far away). This prioritization of attention, influenced by various factors including immediate relevance and past experience, systematically affects how tasks are performed. The paper builds on the “value-guided construal” framework, which models how people form simplified mental representations to plan effectively with limited attentional resources.
Attention-Aware Inverse Planning: A New Lens
The core innovation of this research is the formal articulation of the attention-aware inverse planning problem. Instead of assuming optimal behavior, this approach seeks to infer the underlying attentional biases that shape a person’s decisions. It extends the value-guided construal framework by incorporating “heuristic biases” – learned strategies or predispositions that influence what a person attends to. For instance, a driver in a snowy region might have a strong bias to attend to ice on the road, while a driver in a warmer climate might ignore it.
The model works by observing a sequence of actions and then inferring the set of heuristic bias parameters that best explain that behavior. This involves considering how a decision-maker samples a “construal” (a simplified representation of the environment based on their attention) and then acts optimally within that limited view.
Testing the Approach: From Simple Grids to Real-World Roads
The researchers validated their attention-aware inverse planning method through a series of experiments. First, in a simplified “DrivingWorld” tabular domain, they simulated agents with different attentional biases (e.g., biases to ignore traffic cones, ice patches, or parked cars). They then successfully inferred these biases from the agents’ simulated trajectories, demonstrating the feasibility of the approach in a controlled setting.
Crucially, the paper highlights a significant difference from standard inverse reinforcement learning (IRL). When IRL was applied to behavior generated by an attention-limited agent, it often failed to accurately capture the nuances of the original policy, leading to noisier and less representative behaviors. This suggests that mis-specifying the human decision-making process (by assuming unlimited attention) can lead to substantial errors in modeling.
To demonstrate scalability to more complex, continuous environments, the team applied their method to real-world driving scenarios from the Waymo Open Dataset, using a high-fidelity simulator called GPUDrive. They generated synthetic driving data from agents with specific heuristic biases related to relative motion (e.g., deviation from ego heading, relative heading, deviation from ego collision). The attention-aware inverse planning algorithm was then able to reliably recover these underlying biases from the simulated driving trajectories. This showcases the potential for this method to be used in naturalistic and challenging domains.
Also Read:
- Perception Learning: Decoupling How AI Sees from How It Decides
- The Oversight Game: A New Framework for Balancing AI Autonomy and Human Control
Towards Safer and More Understanding AI
This research represents a significant step towards bridging the gap between computational cognitive science and scalable human behavior modeling. By explicitly accounting for attentional biases, attention-aware inverse planning offers a more accurate and interpretable way to understand human decision-making. This can lead to the development of safer, more trustworthy, and more effective autonomous systems that can better anticipate and interact with people, especially in critical applications like self-driving cars.
For more in-depth information, you can access the full research paper here: Estimating cognitive biases with attention-aware inverse planning.


