2025
Robust Offline Imitation Learning Through State-level Trajectory Stitching.
CoRR, March, 2025

2024
Improving world models for robot arm grasping with backward dynamics prediction.
Int. J. Mach. Learn. Cybern., September, 2024

2023
STORM: Efficient Stochastic Transformer based World Models for Reinforcement Learning.
Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023