Active Inference in Contextual Multi-Armed Bandits for Autonomous Robotic Exploration

Abstract

Autonomous selection of optimal options for data collection from multiple alternatives is challenging in uncertain environments. When secondary information about options is accessible, such problems can be framed as contextual multi-armed bandits (CMABs). Neuro-inspired active inference has gained interest for its ability to balance exploration and exploitation using the expected free energy objective function. Unlike previous studies that showed the effectiveness of active inference based strategy for CMABs using synthetic data, this study aims to apply active inference to realistic scenarios, using a simulated mineralogical survey site selection problem. Hyperspectral data from AVIRIS-NG at Cuprite, Nevada, serves as contextual information for predicting outcome probabilities, while geologists' mineral labels represent outcomes. Monte Carlo simulations assess the robustness of active inference against changing expert preferences. Results show that active inference requires fewer iterations than standard bandit approaches with real-world noisy and biased data, and performs better when outcome preferences vary online by adapting the selection strategy to align with expert shifts.

11 pages, 12 figures, submitted to IEEE Transactions on Robotics

References in corpus (8)

• A Contextual-Bandit Approach to Personalized News Article Recommendation
• Simulation-based optimal Bayesian experimental design for nonlinear systems
• Partially Observable Markov Decision Processes (POMDPs) and Robotics
• A Novel Adaptive Controller for Robot Manipulators based on Active Inference
• An empirical evaluation of active inference in multi-armed bandits
• Data-Free/Data-Sparse Softmax Parameter Estimation with Structured Class Geometries
• Probabilistic Semantic Data Association for Collaborative Human-Robot Sensing
• Observation-Augmented Contextual Multi-Armed Bandits for Robotic Search and Exploration