Categorical Internalisation of Environmental Groupoids for Generalisable POMDP Solving
Organizations: City St George’s, University of London
Abstract
This paper advocates category theory as a practical framework for structuring and improving rein- forcement learning in high-dimensional, partially observable environments. We model symmetries between environmental states by partitioning the state space into equivalence classes induced by sym- metry orbits, and organise each such class as a groupoid with a designated canonical representative. This allows the agent to share what it learns across many similar environmental states simultaneously, rather than treating every orientation or position as an entirely new problem. Learning is thus carried out on a symmetry-reduced state space with each orbit represented once, preserving structure while eliminating redundancy and improving sample efficiency. We implement this framework within standard reinforcement learning pipelines and evaluate two different approaches on partially observable benchmarks, demonstrating that orbit-based partitioning yields consistent performance improvements in environments exhibiting latent symmetry. Beyond these empirical results, our approach illustrates how categorical structure provides a principled bridge between abstract reinforcement learning formulations and their computational application, thereby establishing a pathway toward more structured and scalable learning systems.
Figures & tables
| Category | Specific Parameter | RockSample (20 20) | Tag (15 15) |
|---|---|---|---|
| Environment | Grid Dimensions ( ) | ||
| Training Episodes | |||
| Reward Discount Factor ( ) | |||
| Observation Model | Exponential Decay ( ) | Adjacent Only | |
| Groupoid Agent | Learning Rate ( ) | ||
| Target Symmetry Group ( ) | Dihedral | Translation |
Reinforcement Learning
Reinforcement learning studies how agents learn sequential decisions from rewards and interaction, balancing exploration, performance, safety, and adaptation.
See all 1,739 papers