← Search

Artem Zholus

4 accepted papers

2026

Unraveling the Complexity of Memory in RL Agents: an Approach for Classification and Evaluation

ICLR 2026poster

The incorporation of memory into agents is essential for numerous tasks within the domain of Reinforcement Learning (RL). In particular, memory is paramount for tasks that require the use of past information, adaptation to novel environments, and improved sample efficiency. However, the term ``memor…

Cited by 0SourceScholar
2025

BindGPT: A Scalable Framework for 3D Molecular Design via Language Modeling and Reinforcement Learning

AAAI 2025technical

Generating novel active molecules for a given protein is an extremely challenging task for generative models that requires an understanding of the complex physical interactions between the molecule and its environment. This paper presents a novel generative model, BindGPT, which uses a conceptually…

Cited by 10SourcePDFScholar
2025

TAPNext: Tracking Any Point (TAP) as Next Token Prediction

ICCV 2025poster

Tracking Any Point (TAP) in a video is a challenging computer vision problem with many demonstrated applications in robotics, video editing, and 3D reconstruction. Existing methods for TAP rely heavily on complex tracking-specific inductive biases and heuristics, limiting their generality and potent…

2024

Mastering Memory Tasks with World Models

ICLR 2024oral

Current model-based reinforcement learning (MBRL) agents struggle with long-term dependencies. This limits their ability to effectively solve tasks involving extended time gaps between actions and outcomes, or tasks demanding the recalling of distant observations to inform current actions. To improv…