cs.ROOct 26, 2026

HIDRA: Hierarchical Dual-Routing Attention for Replay-Free Lifelong Imitation Learning

Authors: Fanqi Yu, Matteo Tiezzi, Cigdem Beyan, Tommaso Apicella, Vittorio Murino

Organizations: AI for Good (AIGO) Research Unit, Istituto Italiano di Tecnologia, Genoa, Italy · DITEN, University of Genoa, Genoa, Italy · Pattern Analysis and Computer Vision (PAVIS), Istituto Italiano di Tecnologia, Genoa, Italy · Department of Computer Science, University of Verona, Verona, Italy

Abstract

Robotic agents operating in real-world environments must adapt continuously from a stream of multimodal demonstrations, often under constraints that preclude storing or revisiting past data. In this replay-free, single-pass lifelong imitation learning setting, sequential updates can progressively distort the representation space, making expert-based approaches particularly sensitive to routing interference, causing expert selection to degrade over time. We propose HIDRA, a structured routing framework that mitigates this issue through a hierarchical dual-attention mechanism, which decouples instruction-level expert retrieval from context-dependent refinement. To further stabilize routing under representation drift, HIDRA introduces a key-level geometric regularization that enforces both alignment within tasks and separation across tasks. The resulting approach enables reliable expert selection while supporting an expandable set of experts for continual adaptation. We evaluate HIDRA on several LIBERO benchmark suites, including more challenging variants with heterogeneous task sequences and paraphrased language instructions. Our approach consistently outperforms both replay-free and replay-based baselines, improving AUC and forward transfer while maintaining low forgetting, with stronger gains under high interference and language variation. The code is available at: https://github.com/yfqi/HIDRA.

Figures & tables

Appendix figures & tables15 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Lifelong Imitation Learning with Multimodal Latent Replay and Incremental Adjustment

    Mar 11, 2026Fanqi Yu, Matteo Tiezzi, Tommaso Apicella +2Imitation LearningRobot Imitation Learning

  2. SPREAD: Subspace Representation Distillation for Lifelong Imitation Learning

    Mar 9, 2026Kaushik Roy, Giovanni D'urso, Nicholas Lawrance +2Imitation LearningRobot Imitation Learning

  3. LAR-MoE: Latent-Aligned Routing for Mixture of Experts in Robotic Imitation Learning

    Mar 9, 2026Ariel Rodriguez, Chenpan Li, Lorenzo Mazza +5Robot Imitation LearningAgent Skill Learning