← Search

Sergey Kolesnikov

8 accepted papers

2024

Emergence of In-Context Reinforcement Learning from Noise Distillation

ICML 2024poster

Recently, extensive studies in Reinforcement Learning have been carried out on the ability of transformers to adapt in-context to various environments and tasks. Current in-context RL methods are limited by their strict requirements for data, which needs to be generated by RL agents or labeled with…

2024

In-Context Reinforcement Learning for Variable Action Spaces

ICML 2024poster

Recently, it has been shown that transformers pre-trained on diverse datasets with multi-episode contexts can generalize to new reinforcement learning tasks in-context. A key limitation of previously proposed models is their reliance on a predefined action space size and structure. The introduction…

2024

XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAX

NeurIPS 2024poster

Inspired by the diversity and depth of XLand and the simplicity and minimalism of MiniGrid, we present XLand-MiniGrid, a suite of tools and grid-world environments for meta-reinforcement learning research. Written in JAX, XLand-MiniGrid is designed to be highly scalable and can potentially run on GP…

2023

Anti-Exploration by Random Network Distillation

ICML 2023poster

Despite the success of Random Network Distillation (RND) in various domains, it was shown as not discriminative enough to be used as an uncertainty estimator for penalizing out-of-distribution actions in offline reinforcement learning. In this paper, we revisit these results and show that, with a na…

2023

CORL: Research-oriented Deep Offline Reinforcement Learning Library

NeurIPS 2023poster

CORL is an open-source library that provides thoroughly benchmarked single-file implementations of both deep offline and offline-to-online reinforcement learning algorithms. It emphasizes a simple developing experience with a straightforward codebase and a modern analysis tracking tool. In CORL, we…

Cited by 151SourcePDFScholar
2023

Katakomba: Tools and Benchmarks for Data-Driven NetHack

NeurIPS 2023poster

NetHack is known as the frontier of reinforcement learning research where learning-based methods still need to catch up to rule-based solutions. One of the promising directions for a breakthrough is using pre-collected datasets similar to recent developments in robotics, recommender systems, and mor…

2023

Revisiting the Minimalist Approach to Offline Reinforcement Learning

NeurIPS 2023poster

Recent years have witnessed significant advancements in offline reinforcement learning (RL), resulting in the development of numerous algorithms with varying degrees of complexity. While these algorithms have led to noteworthy improvements, many incorporate seemingly minor design choices that impact…

2022

Showing Your Offline Reinforcement Learning Work: Online Evaluation Budget Matters

ICML 2022spotlight

In this work, we argue for the importance of an online evaluation budget for a reliable comparison of deep offline RL algorithms. First, we delineate that the online evaluation budget is problem-dependent, where some problems allow for less but others for more. And second, we demonstrate that the pr…

Cited by 26SourcePDFScholar