Search papers, labs, and topics across Lattice.
This paper introduces an automated prototype selection method for Prototype-Wrapper Networks (PW-Nets) in reinforcement learning, addressing the limitation of requiring manually defined prototypes that necessitate expert domain knowledge. The proposed approach uses Principal Prototype Analysis to identify optimal prototypes directly from the data, enhancing the interpretability of RL agents. Experiments on standard Gym environments show that the automated prototype selection matches the performance of existing PW-Nets and remains competitive with original black-box models.
Forget hand-crafting prototypes for interpretable RL: this method learns them directly from the data, matching the performance of expert-designed systems.
Recent years have witnessed the widespread adoption of reinforcement learning (RL), from solving real-time games to fine-tuning large language models using human preference data significantly improving alignment with user expectations. However, as model complexity grows exponentially, the interpretability of these systems becomes increasingly challenging. While numerous explainability methods have been developed for computer vision and natural language processing to elucidate both local and global reasoning patterns, their application to RL remains limited. Direct extensions of these methods often struggle to maintain the delicate balance between interpretability and performance within RL settings. Prototype-Wrapper Networks (PW-Nets) have recently shown promise in bridging this gap by enhancing explainability in RL domains without sacrificing the efficiency of the original black-box models. However, these methods typically require manually defined reference prototypes, which often necessitate expert domain knowledge. In this work, we propose a method that removes this dependency by automatically selecting optimal prototypes from the available data. Preliminary experiments on standard Gym environments demonstrate that our approach matches the performance of existing PW-Nets, while remaining competitive with the original black-box models.
