Saved in:
Bibliographic Details
Main Authors: Vamshi, Bodla Krishna, Yang, Haizhao
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2603.27971
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914435216441344
author Vamshi, Bodla Krishna
Yang, Haizhao
author_facet Vamshi, Bodla Krishna
Yang, Haizhao
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2603_27971
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Principal Prototype Analysis on Manifold for Interpretable Reinforcement Learning
Vamshi, Bodla Krishna
Yang, Haizhao
Machine Learning
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.
title Principal Prototype Analysis on Manifold for Interpretable Reinforcement Learning
topic Machine Learning
url https://arxiv.org/abs/2603.27971