cs.ROJul 21, 2026

Beyond Transformers: Linear Attention Policy for Open-Vocabulary Object Goal Navigation

Authors: Jiahong ZhangYifan LinYandong ZhangSijun ShenKexin WangYuqi PanHongjuan PeiWei Wang+1 more

Organizations: Institute of Automation, Chinese Academy of Sciences, Beijing 100045, China · State Key Laboratory of Media Convergence and Communication, Communication University of China, Beijing 100024, China · Beihang University, Beijing 100191, China · School of Artificial Intelligence, University of Chinese Academy of Sciences, Beijing 100049, China · University of Chinese Academy of Sciences, No.19A Yuquan Road, Beijing 100049, China

Abstract

Open-Vocabulary Object Goal Navigation (OVON) requires agents to operate under partial observability, making effective internal state updates critical for navigation performance. This update is implemented by the policy network, where recent approaches adopt Transformer-based backbones with self-attention over a context window to integrate temporal information. However, our controlled experiments show that performance does not scale with context length under Transformer-based policies, questioning the suitability of self-attention for state integration in navigation. To this end, we propose Linear Attention-based Navigation (LANav), which adopts linear attention (LA) as the policy backbone to maintain a structured state update rather than self-attention over the context window. Across multiple LA variants evaluated under identical settings, LANav consistently outperforms Transformer-based baselines. Performance improves as state update mechanisms become more structured and regulated, highlighting the importance of state update design. To improve state update effectiveness, we introduce Weighted State-Expansion Linear Attention (WSLA), which expands each attention head's state into multiple sub-states and uses learnable weighted readout to aggregate expanded sub-states. Equipped with WSLA, LANav achieves 36.4% average success rate (SR) on HM3D-OVON, outperforming Transformer-based counterparts by 6.3 percentage points in macro-averaged SR, while maintaining computational efficiency. Distance-stratified results show larger gains in long-distance episodes, while HSSD transfer and fine-tuning demonstrate robustness across scene distributions. Real-world deployment on a Unitree Go2 further achieves an 82% success rate over 50 trials, supporting the practical feasibility and sim-to-real transfer of LANav.

Explore similar work

CardsList