In applied mathematics, proto-value functions (PVFs) are automatically learned basis functions that are useful in approximating task-specific value functions, providing a compact representation of the powers of transition matrices. They provide a novel framework for solving the credit assignment problem. The framework introduces a novel approach to solving Markov decision processes (MDP) and reinforcement learning problems, using multiscale spectral and manifold learning methods. Proto-value functions are generated by spectral analysis of a graph, using the graph Laplacian.
Contents
- Motivation
- Overview
- Basis functions from graph Laplacian
- Graph construction on discrete state space
- Graph construction on continuous or large state space
- Application
- Least squares approximation using proto value functions
- References
Proto-value functions were first introduced in the context of reinforcement learning by Sridhar Mahadevan in his paper, Proto-Value Functions: Developmental Reinforcement Learning at ICML 2005.
Motivation
Value function approximation is a critical component to solving MDPs defined over a continuous state space. A good function approximator allows an RL agent to accurately represent the value of any state it has experienced, without explicitly storing its value. Linear function approximation using basis functions is a common way of constructing a value function approximation, like Radial basis functions, polynomial state encodings, and CMACs. However, parameters associated with these basis functions often require significant domain-specific hand-engineering. Proto-value functions attempts to solve this required hand-engineering by accounting for the underlying manifold structure of the problem domain.
Overview
Proto-value functions are task-independent global basis functions that collectively span the entire space of possible value functions for a given state space. They incorporate geometric constraints intrinsic to the environment. For example, states close in Euclidean distance (such as states on opposite sides of a wall) may be far apart in manifold space. Previous approaches to this nonlinearity problem lacked a broad theoretical framework, and consequently have only been explored in the context of discrete MDPs.
Proto-value functions arise from reformulating the problem of value function approximation as real-valued function approximation on a graph or manifold. This results in broader applicability of the learned bases and enables a new class of learning algorithms, which learn representations and policies at the same time.
Basis functions from graph Laplacian
In this approach, we will construct the basis functions by spectral analysis of the graph Laplacian, a self-adjoint (or symmetric) operator on the space of functions on the graph, closely related to the random walk operator.
For the sake of simplicity, assume that the underlying state space can be represented as an undirected unweighted graph
The spectral analysis of the Laplace operator on a graph consists of finding the eigenvalues and eigenfunctions which solve the equation
where
The combinatorial Laplacian is not the only operator on graphs to select from. Other possible graph operators include:
Graph construction on discrete state space
For a finite state space the graph
It is important to note that this can only be done when the state space is finite and of reasonable size.
Graph construction on continuous or large state space
For a continuous state space or simply a very large discrete state space, it is necessary to sample from the manifold in state space. Then constructing the Graph
Application
Once the PVFs are generated, they can be plugged into a traditional function approximation framework. One such method is least-squares approximation.
Least-squares approximation using proto-value functions
Let
Define the gram matrix
here
Now the we can solve for the coefficients that minimize the least squares error with the equation
A nonlinear least-squares approach is possible by using the k PVFs with the largest absolute coefficients to compute the approximation.