← Search

Wataru Kumagai

7 accepted papers

2025

Near-Optimal Policy Identification in Robust Constrained Markov Decision Processes via Epigraph Form

ICLR 2025poster

Designing a safe policy for uncertain environments is crucial in real-world control systems. However, this challenge remains inadequately addressed within the Markov decision process (MDP) framework. This paper presents the first algorithm guaranteed to identify a near-optimal policy in a robust con…

2025

Provably Efficient RL under Episode-Wise Safety in Constrained MDPs with Linear Function Approximation

NeurIPS 2025spotlight

We study the reinforcement learning (RL) problem in a constrained Markov decision process (CMDP), where an agent explores the environment to maximize the expected cumulative reward while satisfying a single constraint on the expected total utility value in every episode. While this problem is well u…

Cited by 0SourceScholar
2023

Regularization and Variance-Weighted Regression Achieves Minimax Optimality in Linear MDPs: Theory and Practice

ICML 2023poster

Mirror descent value iteration (MDVI), an abstraction of Kullback-Leibler (KL) and entropy-regularized reinforcement learning (RL), has served as the basis for recent high-performing practical RL algorithms. However, despite the use of function approximation in practice, the theoretical understandin…

2022

Langevin Autoencoders for Learning Deep Latent Variable Models

NeurIPS 2022accept

Markov chain Monte Carlo (MCMC), such as Langevin dynamics, is valid for approximating intractable distributions. However, its usage is limited in the context of deep latent variable models owing to costly datapoint-wise sampling iterations and slow convergence. This paper proposes the amortized Lan…

2021

Group Equivariant Conditional Neural Processes

ICLR 2021poster

We present the group equivariant conditional neural process (EquivCNP), a meta-learning method with permutation invariance in a data set as in conventional conditional neural processes (CNPs), and it also has transformation equivariance in data space. Incorporating group equivariance, such as rotati…

Cited by 31SourcePDFScholar