ORCAID: Oblique Rule-Based Continuous-Action Interpretation for Deep RL Policies
About
Explainability remains a key issue in reinforcement learning (RL). Distilling an interpretable policy from an agent trained in a complex environment is particularly challenging when the action space is continuous. We introduce ORCAID, a novel method for extracting interpretable rule-based policies from RL agents operating in mixed continuous-discrete environments with continuous action spaces. Our main contribution is an efficient oblique decision tree training algorithm that partitions the state space by hyperplanes and fits local linear models. The key idea lies in a three-stage split search: efficient random initialization, local refinement, and backward elimination. Finally, adjacent leaves are merged to yield a concise set of interpretable rules describing a given deep RL policy. We evaluate ORCAID across multiple RL environments, demonstrating that the extracted rule-based policies maintain strong performance with a low number of parameters and can even be used to improve the performance of the original deep RL policy.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Reinforcement Learning Surrogate Modeling | Pendulum (P) (test) | Reward Ratio92 | 8 | |
| Policy improvement | InvertedPendulum (IP) | Rate17 | 4 | |
| Policy improvement | InvertedDoublePendulum (IDP) | Success Rate38 | 4 | |
| Reinforcement Learning Surrogate Modeling | LunarLander (LL) (test) | Reward Ratio90 | 4 | |
| Reinforcement Learning Surrogate Modeling | InvertedDoublePendulum (IDP) (test) | Reward Ratio (%)103 | 4 | |
| Reinforcement Learning Surrogate Modeling | Hopper H (test) | Reward Ratio89 | 4 | |
| Reinforcement Learning Surrogate Modeling | MountainCar (MC) (test) | Reward Ratio98 | 4 | |
| Reinforcement Learning Surrogate Modeling | Reacher (R) (test) | Reward Ratio80 | 4 | |
| Reinforcement Learning Surrogate Modeling | HalfCheetah (HC) (test) | Reward Ratio78 | 4 | |
| Reinforcement Learning Surrogate Modeling | Swimmer (S) (test) | Reward Ratio90 | 4 |