Reinforcement Learning-Guided Retrieval with Soft Fusion for Robust Multimodal Imitation Learning under Missing Modalities

Hugging Face Daily Papers Papers

Summary

RL4IL introduces a reinforcement learning-guided retrieval method that uses soft fusion over frozen demonstration libraries to handle missing sensor modalities in robotic imitation learning at inference time, achieving high success rates under complete camera dropout.

Robotic systems perceive the world through multiple input modalities -- including visual camera streams and natural language instructions -- and must select appropriate actions based on these signals. However, assuming the permanent availability of all input devices is unrealistic, as sensors may fail, become occluded, or drop out entirely during deployment. Robust handling of such missing-modality scenarios is therefore essential for real-world robot operation. This paper introduces RL4IL, a reinforcement learning guided method for imitation learning that selects the most suitable action for a given observation by identifying the most relevant expert demonstrations from a training library. A reinforcement learning policy, trained via Proximal Policy Optimisation over Breadth-First Search candidate sets, ranks candidate demonstrations and a soft cross-attention fusion head aggregates their action signals to produce the final prediction. When a modality is missing at inference time, a dedicated per-modality RL retrieval policy identifies donor demonstrations from the training library, and a soft imputation head reconstructs the missing embedding via cross-attention over the top-ranked donors -- without requiring any retraining of the system. Experiments on three LIBERO benchmark suites demonstrate that RL4IL substantially outperforms state-of-the-art imitation learning methods under sensor dropout conditions, while requiring no policy network training. The code can be found at https://github.com/h-ismkhan/Reinforcement-Learning-via-kNN-for-Robotic-Learning-with-Missing-Camera
Original Article
View Cached Full Text

Cached at: 06/18/26, 07:58 PM

Paper page - Reinforcement Learning-Guided Retrieval with Soft Fusion for Robust Multimodal Imitation Learning under Missing Modalities

Source: https://huggingface.co/papers/2606.15514 RL4IL addresses a genuinely underexplored problem in imitation learning: what happens when sensors fail at deployment time? Most IL methods silently assume all modalities are always available, which is unrealistic for real robot deployments. Our key insight is that instead of retraining the policy for every possible dropout pattern, we can retrieve the right behaviour from a frozen demonstration library using a learned RL ranking policy. A few highlights that might interest the community:

The PPO policy operates over BFS-augmented candidate sets, giving it a richer and more label-diverse pool than plain kNN Soft cross-attention fusion over top-K ranked candidates consistently outperforms hard argmax selection, especially under noisy retrieval Zero-shot missing-modality handling at inference — no retraining needed when a camera fails On LIBERO benchmarks, RL4IL reaches up to 0.733 success rate under complete camera dropout, where the strongest prior method (DisDP) reaches only 0.295

Happy to discuss the retrieval design, the imputation pipeline, or the LIBERO experimental setup with anyone interested in robust robot learning.

Similar Articles

Region-Level Policy Optimization for Fine-grained MLLM Perception

Hugging Face Daily Papers

Vision-RL² is a method for improving fine-grained perception in multimodal large language models by using region-level reinforcement learning to compress visual tokens and enhance performance across multiple benchmarks without full model fine-tuning.

LLMs help robots understand vague instructions and focus on key details

MIT News — Artificial Intelligence

MIT CSAIL researchers developed Masked Inverse Reinforcement Learning (IRL), which uses large language models to clarify ambiguous instructions for robots and focus on key environmental details, reducing the need for extensive demonstration data.