Minimax Optimal Q Learning with Nearest Neighbors

Markov decision process (MDP) is an important model of sequential decision making problems. Existing theoretical analysis focus primarily on finite state spaces. For continuous state spaces, a recent interesting work [1] proposes a nearest neighbor
Q
learning approach. Under the streaming setting, in shich samples are received in a sequential manner, the sample complexity of this method is Õ(|
A
|/ϵ
^d+3
(1-γ)
^d+7
) for ϵ-accurate
Q
function estimation of infinite horizon discounted MDP with discount factor γ, in which |
A
| is the size of the action space. However, the sample complexity is not optimal, and the method is suitable only for bounded state spaces. In this paper, we propose two new nearest neighbor
Q
learning methods, one for the offline setting and the other for the streaming setting. We show that the sample complexities of these two methods are Õ(|
A
|/ϵ
^d+2
(1-γ)
^d+2
) and Õ(|
A
|/ϵ
^d+2
(1-γ)
^d+3
) for offline and streaming settings respectively, which significantly improve over existing results and have minimax optimal dependence over ϵ. We achieve such improvement by utilizing samples more efficiently. In particular, the method in [1] clears up all samples after each iteration, thus these samples are somewhat wasted. On the other hand, our offline method does not remove any samples, and our streaming method only removes samples with time earlier than β
t
at time
t
, thus our methods significantly reduce the loss of information. Apart from the sample complexity, our methods also have additional advantages of better computational complexity, as well as suitability to unbounded state spaces. Finally, we extend our work to the case where both state and action spaces are continuous.