EgoReasoner: Learning Egocentric 4D Reasoning via Task-Adaptive Structured Thinking
Organizations: Northeastern University · Google Research · Google DeepMind
Abstract
Egocentric video understanding is inherently complex due to the dynamic 4D nature of the environment, where camera motion and object displacements necessitate a continuous re-evaluation of spatial relations. In this work, we target a suite of under-explored egocentric 4D reasoning tasks, including fixture interaction counting, viewpoint-relative fixture location, object movement itinerary tracking, and stationary object localization, that require fundamentally different cognitive operations: spatial anchoring, temporal tracking, and duration reasoning. We observe that these structural differences make task-agnostic approaches insufficient: generic Chain-of-Thought methods lack task-appropriate reasoning primitives, and uniform reinforcement learning actively destabilizes performance on spatial tasks. To address this, we propose EgoReasoner, a two-stage framework that aligns both the reasoning scaffold and the reward signal to each task's cognitive structure. In the first stage, Task-Adaptive Thinking Templates guide the synthesis of structured CoT traces that teach the model to reason adaptively across task types via supervised fine-tuning. In the second stage, task-aware reward functions verify entity grounding, temporal alignment, and task-adaptive logical consistency, selectively strengthening each reasoning pathway via reinforcement fine-tuning with GRPO. Our 3B-parameter model, trained on only 16K samples, achieves 37.5% average accuracy on the challenging HD-EPIC benchmark, surpassing Qwen2.5-VL-7B (25.7%) by over 10 points.
Figures & tables
| Models | Size | Avg. | 3D Perception | Object Motion | ||||
| Fixture Interaction Count | Fixture Location | Object Location | Object Movement Count | Object Movement Itinerary | Stationary Object Localization | |||
| Sample Human Baseline | – | 91.1 | 89.5 | 95.6 | 96.3 | 87.6 | 93.2 | 84.3 |
| Closed-source Models | ||||||||
| Gemini 2.5 Pro [ 6 ] | 128B | 49.2 | 36.3 | 40.4 | 63.6 | 45.5 | 60.0 | 49.5 |
| Open-source Models | ||||||||
| VideoLlama 2 [ 5 ] | 7B | 25.5 | 17.7 | 18.8 | 31.0 | 44.0 | 11.0 | 30.5 |
| Setting | Variant | 3D Perception | Object Motion | Avg. | |||||
| CoT | Task-Aware Reward | FIC | FL | OL | OMC | OMI | SOL | ||
| Baseline | - | - | 31.0 | 18.4 | 37.8 | 39.0 | 16.2 | 19.5 | 27.0 |
| +SFT | - | - | 24.3 | 22.6 | 38.4 | 42.5 | 21.6 | 20.0 | 28.2 |
| ✓ | - | 34.3 | 25.2 | 44.8 | 41.0 | 28.0 | 18.0 | 31.9 | |
| +RFT | ✓ | - | 33.7 | 24.8 | 40.6 | 37.0 | 26.2 | 24.5 | 31.1 |
| ✓ | ✓ | 37.7 | 28.0 | 48.2 | 46.5 | 36.6 | 23.5 | 36.8 | |
Egocentric Video Understanding
Egocentric video understanding studies how AI interprets first-person video to recognize actions, track interactions, reason about space and time, and assist users.
See all 203 papers