← Search

Tyler Kastner

3 accepted papers

2025

Categorical Distributional Reinforcement Learning with Kullback-Leibler Divergence: Convergence and Asymptotics

ICML 2025poster

We study the problem of distributional reinforcement learning using categorical parametrisations and a KL divergence loss. Previous work analyzing categorical distributional RL has done so using a Cramér distance-based loss, simplifying the analysis but creating a theory-practice gap. We introduce a…

Cited by 0SourcePDFScholar
2023

Distributional Model Equivalence for Risk-Sensitive Reinforcement Learning

NeurIPS 2023poster

We consider the problem of learning models for risk-sensitive reinforcement learning. We theoretically demonstrate that proper value equivalence, a method of learning models which can be used to plan optimally in the risk-neutral setting, is not sufficient to plan optimally in the risk-sensitive set…

2021

MICo: Improved representations via sampling-based state similarity for Markov decision processes

NeurIPS 2021poster

We present a new behavioural distance over the state space of a Markov decision process, and demonstrate the use of this distance as an effective means of shaping the learnt representations of deep reinforcement learning agents. While existing notions of state similarity are typically difficult to l…