reinforcement learning high-dimensional data with geometric structure