I work on probabilistic machine learning, vision-language models, and sample-efficient reinforcement learning.

Selected publications

Reinforcement Learning via Self-Distillation

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

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

Anton Baumann, Thomas Roßberg, Michael Schmitt

Oral ICCV 2023 Workshop on Uncertainty in Computer Vision

Experience

Aalto University, Helsinki

Researched probabilistic ML for vision-language models in Arno Solin's lab.

Bundeswehr University, Munich

Applied probabilistic ML and data fusion to remote sensing problems.

molab.ai, Munich

Implemented GNNs for chemical property prediction in PyTorch.

Ablacon GmbH, Munich

Extended Horn & Schunck optical flow for electrographic flow mapping.

Integrated the method into a prototype core product with a research engineering team.

Cliqz GmbH, Munich

Developed a logging/versioning system and JSON diff library in Go for privacy-first ad-tech infrastructure.

Education

ETH Zürich

Thesis supervision by Andreas Krause (ETH Zürich) and Zeynep Akata (TUM), with day-to-day supervision by Jonas Hübotter.

KTH Stockholm

Technical University of Munich