Interaction Locality in Hierarchical Recursive Reasoning
Summary
"Interaction locality" is a new task-geometry-aware framework designed to measure whether information flow in spatial reasoning models remains within nearby cells or semantic segments, or crosses them. This framework, instantiated with sparse-autoencoder feature ablations and finite-noise activation patching, was applied to HRM and TRM, two compact hierarchical and recursive reasoning models, on Maze-Hard, Sudoku Extreme, and ARC-AGI benchmarks. Findings indicate that high-level recurrent states in these models primarily write information within nearby cells, with repeated recursive updates accumulating these local writes into broader solution structures. This pattern was strongest in TRM. When extended to MTU3D, a large-scale embodied 3D scene-grounding model, causal spatial locality was observed mainly at the transition where visual scene features are handed to the grounding module, suggesting a difference in how local-to-global handoffs occur between explicit recursive reasoning and embodied 3D models.
Key takeaway
For AI Architects designing or debugging spatial reasoning models, understanding interaction locality is crucial. You should apply this framework to analyze information flow within your models, especially when integrating recursive components or embodied perception. This helps pinpoint if local computations build global solutions. It also reveals if causal spatial structure bottlenecks at module interfaces, guiding architectural refinements.
Key insights
Interaction locality quantifies information flow in spatial reasoning, revealing local-to-global accumulation in recursive models.
Principles
- High-level recurrent states write information locally.
- Recursive updates accumulate local writes into global structure.
- Embodied 3D models concentrate locality at module boundaries.
Method
Interaction locality is instantiated using sparse-autoencoder feature ablations and finite-noise activation patching, complemented by structural Jacobian and attention checks.
In practice
- Apply framework to analyze spatial reasoning models.
- Identify local-to-global information flow patterns.
- Pinpoint causal spatial structure at module boundaries.
Topics
- Interaction Locality
- Spatial Reasoning
- Recursive Neural Networks
- Embodied AI
- Information Flow
- Activation Patching
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Architect
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.