Nikola Milosevic

Nikola Milosevic

Doktorand, MPI CBS, Leipzig

Ich bin Doktorand am MPI CBS und forsche an Reinforcement Learning, das sicher, theoretisch gut verstanden und praxistauglich ist.

Die meisten Fortschritte im Reinforcement Learning entstehen in Simulationen. Sie unter realen Bedingungen zum Laufen zu bringen (mit Sicherheitsanforderungen, begrenzten Daten und begrenzter Zeit, im Einsatz auf physischer Hardware) bleibt eine offene Herausforderung. Doch genau dort liegen auch die interessanten theoretischen Fragen, und dort lassen sich Erkenntnisse aus Biologie und Neurowissenschaft einbringen. Ich arbeite daran, für ebendiese Fragen ein tragfähiges Fundament für Reinforcement Learning über Anwendungsfelder hinweg zu schaffen.

Ausgewählte Publikationen

Active Inference as a Convex Markov Decision Process

Nikola Milosevic, Nicolás Hinrichs, Nico Scherf

Accepted — International Workshop on Active Inference (IWAI 2026)

Zusammenfassung

Active Inference (AIF) frames adaptive behavior as the minimization of expected free energy (EFE), combining epistemic and pragmatic objectives within a single variational principle. We frame AIF as policy optimization and show that, for closed-loop control policies, EFE minimization can be formulated as a convex Markov decision process (MDP). In this formulation, the pragmatic terms are linear in the predictive state marginals and therefore equivalent to reward maximization in a latent MDP, while the epistemic value introduces a nonlinear component that distinguishes EFE minimization from standard reinforcement learning. This perspective further reveals the epistemic drive of active inference as a policy-dependent (performative) reward. We analyze finite-horizon, discounted, and average-reward formulations of EFE and derive a mirror descent (MD) algorithm that locally linearizes the objective around the current state marginals, yielding a policy-dependent reward that is compatible with actor-critic methods and dynamic programming. Finally, we argue that coupling world-model learning with policy optimization gives active inference the structure of performative reinforcement learning, providing a route toward grounding active inference within modern reinforcement learning and optimization theory, including convergence analysis and principled policy improvement guarantees.

Physical embodiment enables information processing beyond explicit flow sensing in active matter

Diptabrata Paul, Nikola Milosevic, Nico Scherf, Frank Cichos

Science Advances, 12(11), eaec0783

Zusammenfassung

We show that physical embodiment in active matter systems enables information processing capabilities that exceed what is possible through explicit flow sensing alone. Using microswimmers as a model system, we demonstrate that the body itself acts as a computational resource, coupling sensory and motor degrees of freedom in ways that simplify the control problem.

Embedding Safety into RL: A New Take on Trust Region Methods

Nikola Milosevic, Johannes Müller, Nico Scherf

International Conference on Machine Learning (ICML), PMLR 267:44199-44224

Zusammenfassung

Reinforcement Learning (RL) agents can solve diverse tasks but often exhibit unsafe behavior. Constrained Markov Decision Processes (CMDPs) address this by enforcing safety constraints, yet existing methods either sacrifice reward maximization or allow unsafe training. We introduce Constrained Trust Region Policy Optimization (C-TRPO), which reshapes the policy space geometry to ensure trust regions contain only safe policies, guaranteeing constraint satisfaction throughout training. We analyze its theoretical properties and connections to TRPO, Natural Policy Gradient (NPG), and Constrained Policy Optimization (CPO). Experiments show that C-TRPO reduces constraint violations while maintaining competitive returns.

Ausbildung

Max-Planck-Institut für Kognitions- und Neurowissenschaften2022 — heuteNeural Data Science Lab · Doktorand
Hochschule für Technik, Wirtschaft und Kultur Leipzig2019 — 2021M.Sc. Elektrotechnik