← Search

Marco Bagatella

13 accepted papers

2026

Policy Search via Bayesian Optimization with Temporal Difference Gaussian Processes

ICML 2026poster

Bayesian optimization (BO) is a method commonly used for policy search in problems with low-dimensional policy parameterizations. While it is generally considered data-efficient, existing BO approaches are agnostic to the sequential structure of the optimization objective induced by policy roll-outs…

Cited by 0SourceScholar
2026

Reinforcement Learning via Self-Distillation

ICML 2026poster

Large language models are increasingly post-trained with reinforcement learning in verifiable domains such as code and math. Yet, current methods for reinforcement learning with verifiable rewards (RLVR) learn only from a scalar outcome reward per attempt, creating a severe credit-assignment bottlen…

Cited by 0SourceScholar
2026

TD-JEPA: Latent-predictive Representations for Zero-Shot Reinforcement Learning

ICLR 2026oral

Latent prediction–where agents learn by predicting their own latents–has emerged as a powerful paradigm for training general representations in machine learning. In reinforcement learning (RL), this approach has been explored to define auxiliary losses for a variety of settings, including reward-bas…

Cited by 0SourcecodeScholar
2026

Test-time Offline Reinforcement Learning on Goal-related Experience

ICML 2026poster

Foundation models compress a large amount of information in a single, large neural network, which can then be queried for individual tasks. There are strong parallels between this widespread framework and offline goal-conditioned reinforcement learning algorithms: a universal value function is train…

Cited by 0SourceScholar
2025

DISCOVER: Automated Curricula for Sparse-Reward Reinforcement Learning

NeurIPS 2025poster

Sparse-reward reinforcement learning (RL) can model a wide range of highly complex tasks. Solving sparse-reward tasks is RL's core premise — requiring efficient exploration coupled with long-horizon credit assignment — and overcoming these challenges is key for building self-improving agents with su…

Cited by 0SourcecodeScholar
2025

Zero-Shot Offline Imitation Learning via Optimal Transport

ICML 2025poster

Zero-shot imitation learning algorithms hold the promise of reproducing unseen behavior from as little as a single demonstration at test time. Existing practical approaches view the expert demonstration as a sequence of goals, enabling imitation with a high-level goal selector, and a low-level goal-…

2024

Causal Action Influence Aware Counterfactual Data Augmentation

ICML 2024poster

Offline data are both valuable and practical resources for teaching robots complex behaviors. Ideally, learning agents should not be constrained by the scarcity of available demonstrations, but rather generalize beyond the training distribution. However, the complexity of real-world scenarios typica…

2024

Modelling Microbial Communities with Graph Neural Networks

ICML 2024poster

Understanding the interactions and interplay of microorganisms is a great challenge with many applications in medical and environmental settings. In this work, we model bacterial communities directly from their genomes using graph neural networks (GNNs). GNNs leverage the inductive bias induced by t…

Cited by 2SourcePDFScholar
2023

Efficient Learning of High Level Plans from Play

ICRA 2023poster

Real-world robotic manipulation tasks remain an elusive challenge, since they involve both fine-grained environment interaction, as well as the ability to plan for long-horizon goals. Although deep reinforcement learning (RL) methods have shown encouraging results when planning end-to-end in high-di…

Cited by 5SourceScholar
2021

Planning from Pixels in Environments with Combinatorially Hard Search Spaces

NeurIPS 2021poster

The ability to form complex plans based on raw visual input is a litmus test for current capabilities of artificial intelligence, as it requires a seamless combination of visual processing and abstract algorithmic execution, two traditionally separate areas of computer science. A recent surge of int…

Cited by 8SourcePDFScholar