I work on probabilistic machine learning, vision-language models, and sample-efficient reinforcement learning.
Selected publications
Reinforcement Learning via Self-Distillation
ICML 2026 ·*equal contribution
Jonas Hübotter, Frederike Lübeck*, Lejs Behric*,
Anton Baumann*, Marco Bagatella, Daniel Marta,
Ido Hakimi, Idan Shenfeld, Thomas Kleine Buening, Carlos Guestrin, Andreas Krause
Best Paper|ICLR 2026 Workshop on Test-Time Updates
Oral|ICLR 2026 Workshop on Scaling Post-training for LLMs
Post-hoc Probabilistic Vision-Language Models
ICLR 2026
Anton Baumann, Rui Li, Marcus Klasson, Santeri Mentu,
Shyamgopal Karthik, Zeynep Akata, Arno Solin, Martin Trapp
Spotlight|NeurIPS 2024 Workshop on Bayesian Decision-making and Uncertainty
Probabilistic MIMO U-Net
ICCV 2023 · UnCV workshop
Anton Baumann, Thomas Roßberg, Michael Schmitt
Oral|ICCV 2023 Workshop on Uncertainty in Computer Vision
Experience
Aalto University, Helsinki
Summer Research Intern, Machine Learning Research Group · May 2024 – May 2025
Researched probabilistic ML for vision-language models in Arno Solin's lab.
Bundeswehr University, Munich
Research Assistant, Earth Observation Lab · Feb 2023 – May 2024
Applied probabilistic ML and data fusion to remote sensing problems.
molab.ai, Munich
Data Scientist (Internship, then Part-time) · Aug 2022 – Feb 2023
Implemented GNNs for chemical property prediction in PyTorch.
Ablacon GmbH, Munich
Researcher (Working Student), then Bachelor's Thesis · Mar 2020 – May 2022
Extended Horn & Schunck optical flow for electrographic flow mapping.
Integrated the method into a prototype core product with a research engineering team.