publications

A list of published works in which I was involved.

2025

  1. tdjepa.png
    TD-JEPA: Latent-predictive Representations for Zero-Shot Reinforcement Learning
    Marco Bagatella, Matteo Pirotta, Ahmed Touati, and 2 more authors
    In arXiv, 2025
  2. optibfm.png
    Optimistic Task Inference for Behavior Foundation Models
    Thomas Rupf, Marco Bagatella, Marin Vlastelica Pogancic, and 1 more author
    In arXiv, 2025
  3. gcttt.png
    Test-time Offline Reinforcement Learning on Goal-related Experience
    Marco Bagatella, Mert Albaba, Jonas Hübotter, and 2 more authors
    In European Workshop in Reinforcement Learning, 2025
  4. discover.png
    DISCOVER: Automated Curricula for Sparse-Reward Reinforcement Learning
    Leander Diaz Bone, Marco Bagatella, Jonas Hübotter, and 1 more author
    In Advances in Neural Information Processing Systems, 2025
  5. amf.png
    Active Fine-Tuning of Generalist Policies
    Marco Bagatella, Jonas Hübotter, Georg Martius, and 1 more author
    In Forty-second International Conference on Machine Learning, 2025
  6. zilot.png
    Zero-shot Offline Imitation Learning via Optimal Transport
    Thomas Rupf, Marco Bagatella, Nico Gürtler, and 2 more authors
    In Forty-second International Conference on Machine Learning, 2025
  7. drl2.png
    Directed Exploration in Reinforcement Learning from Linear Temporal Logic
    Marco Bagatella, Andreas Krause, and Georg Martius
    In Transactions on Machine Learning Research, J2C Track, 2025

2024

  1. pst.png
    Problem Space Transformations for Generalisation in Behavioural Cloning
    Kiran Doshi, Marco Bagatella, and Stelian Coros
    In CoRL Workshop on Safe and Robust Robot Learning for Operation in the Real World, 2024
  2. micrognn.png
    Modelling Microbial Communities with Graph Neural Networks
    Albane Ruaud, Cansu Sancaktar, Marco Bagatella, and 2 more authors
    In Forty-first International Conference on Machine Learning, 2024
  3. caiac.png
    Causal Action Influence Aware Counterfactual Data Augmentation
    Núria Armengol Urpí, Marco Bagatella, Marin Vlastelica, and 1 more author
    In Forty-first International Conference on Machine Learning, 2024

2023

  1. gcopfce.png
    Goal-conditioned Offline Planning from Curious Exploration
    Marco Bagatella, and Georg Martius
    In Advances in Neural Information Processing Systems, 2023
  2. elfp.png
    Efficient Learning of High Level Plans from Play
    Núria Armengol Urpí, Marco Bagatella, Otmar Hilliges, and 2 more authors
    In International Conference on Robotics and Automation, 2023

2022

  1. sfp.png
    SFP: State-free Priors for Exploration in Off-Policy Reinforcement Learning
    Marco Bagatella, Sammy Joe Christen, and Otmar Hilliges
    Transactions on Machine Learning Research, 2022

2021

  1. ppgs.png
    Planning from Pixels in Environments with Combinatorially Hard Search Spaces
    Marco Bagatella, Miroslav Olšák, Michal Rolinek, and 1 more author
    In Advances in Neural Information Processing Systems, 2021