Sample-Efficient Multi-Task and
Multi-Objective Reinforcement Learning
by Combining Multiple Behaviors
Advisor: Ana L. C. Bazzan
Co-advisor: Bruno C. da Silva


When solving sequential decision-making problems, humans
combine previously-learned behaviors to quickly solve new tasks.
Mastering the game of Go with deep neural networks and tree search. Silver, D., Huang, A., Maddison, C. et al.. Nature 529, (2016)

Autonomous navigation of stratospheric balloons using reinforcement learning. Bellemare, M.G., Candido, S., Castro, P.S. et al.. Nature 588, (2020)
Walk, Run, Crawl, RL Fun. Boston Dynamics (2025)
… and many other success stories (e.g., ChatGPT)
A Long-Term Goal of AI:
How to build generalist agents capable of learning multiple behaviors and solving multiple tasks?
Driving involves balancing speed, safety, comfort, travel time, …
Aggressive behavior
Safe behavior
Outracing champion Gran Turismo drivers with deep reinforcement learning. Wurman et al. Nature 602 (2022).

How can we design flexible RL agents that can sample-efficiently adapt their behavior to solve any task, where each task may involve multiple conflicting objectives?


Hypothesis:
It is possible to combine the foundations and ideas of
Multi-Task RL (MTRL) and Multi-Objective RL (MORL)
to address this research problem.
We introduce novel multi-policy methods that enable RL agents to:
\Pi =
\pi_1

\pi_2

…
\pi_n

\Pi
+
new task
zero-shot
transfer
new behavior

1. How to construct a set of policies such that combining them directly yields the optimal policy for any task?
Optimistic Linear Support and Successor Features as a Basis for Optimal Policy Transfer (ICML 2022)

2. How to learn a set of policies that solves MORL problems in a robust and sample-efficient way?
Sample-Efficient Multi-Objective Learning via Generalized Policy Improvement Prioritization (AAMAS 2023)
+ JAAMAS Extended Paper

3. How to use approximate models of the environment to improve zero-shot policy transfer?
Multi-Step Generalized Policy Improvement by Leveraging Approximate Models (NeurIPS 2023)

[Contrib. 1] Formal characterization of the connections between MTRL and MORL [ICML 22]

[Contrib. 2] Novel algorithms that use GPI to define prioritization schemes [AAMAS 23]
[Contrib. 3] Robust model-based method for zero-shot policy transfer [NeurIPS 23]

r_1 =
r_2 = …
r_3 = +

These problems have been investigated independently!
\boldsymbol{r} = [, , ]
Each reward component must be traded off!

Reward function is a linear combination of features \;\phi(s,a,s') \in \mathbb{R}^d
r_{\textcolor{#ED7D31}{\boldsymbol{w}}}(s,a,s') = \phi(s,a,s') \cdot \textcolor{#ED7D31}{\boldsymbol{w}}
Successor Features (SFs)
\psi^{\textcolor{#0070C0}{\pi}}(s,a) \triangleq \mathbb{E}_{\textcolor{#0070C0}{\pi}}\left[\sum_{i=0}^{\infty} \gamma^i \phi_{t+i} \;\middle|\; S_t = s, A_t = a\right]
E.g.:
\psi^{\textcolor{#0070C0}{\pi}}(s,a) \triangleq \mathbb{E}_{\textcolor{#0070C0}{\pi}}\left[\begin{array}{rl} \textcolor{#548235}{\mathit{velocity}_t} & + \gamma\, \textcolor{#548235}{\mathit{velocity}_{t+1}} + \dots \\ \textcolor{#C00000}{\mathit{energy}_t} & + \gamma\, \textcolor{#C00000}{\mathit{energy}_{t+1}} + \dots \\ \textcolor{#203864}{\mathit{distance}_t} & + \gamma\, \textcolor{#203864}{\mathit{distance}_{t+1}} + \dots \end{array}\right]
Fast reinforcement learning with generalized policy updates. Barreto et al. Proceedings of the National Academy of Sciences, 2020.
Reward function is a linear combination of features \;\phi(s,a,s') \in \mathbb{R}^d
r_{\textcolor{#ED7D31}{\boldsymbol{w}}}(s,a,s') = \phi(s,a,s') \cdot \textcolor{#ED7D31}{\boldsymbol{w}}
Successor Features (SFs)
\psi^{\textcolor{#0070C0}{\pi}}(s,a) \triangleq \mathbb{E}_{\textcolor{#0070C0}{\pi}}\left[\sum_{i=0}^{\infty} \gamma^i \phi_{t+i} \;\middle|\; S_t = s, A_t = a\right]
Generalized Policy Evaluation (GPE)
directly evaluates how well a policy \pi would perform
under any reward function r_{\boldsymbol{w}} (without having to deploy it!):
q^{\textcolor{#0070C0}{\pi}}_{\textcolor{#ED7D31}{\boldsymbol{w}}}(s,a) = \psi^{\textcolor{#0070C0}{\pi}}(s,a) \cdot \textcolor{#ED7D31}{\boldsymbol{w}}
Fast reinforcement learning with generalized policy updates. Barreto et al. Proceedings of the National Academy of Sciences, 2020.
\textcolor{#00B050}{\pi_1}
w = [1,0,0]

\textcolor{#0070C0}{\pi_2}
w = [0,1,0]

zero-shot
\pi^{GPI}(s;\textcolor{#ED7D31}{w}) \in \operatorname*{arg\,max}_{\textcolor{#C00000}{a}\in\mathcal{A}}\ \max_{\pi\in\{\textcolor{#00B050}{\pi_1},\textcolor{#0070C0}{\pi_2}\}} q^{\pi}_{\textcolor{#ED7D31}{w}}(s,\textcolor{#C00000}{a})
\textcolor{#ED7D31}{w} = [\textcolor{#00B050}{1},\textcolor{#0070C0}{1},0]

GPI Theorem
q^{GPI}_{\textcolor{#ED7D31}{w}}(s,a) \geq \max_{\textcolor{#0070C0}{\pi}\in\Pi} q^{\textcolor{#0070C0}{\pi}}_{\textcolor{#ED7D31}{w}}(s,a) \qquad \textit{for any } \textcolor{#ED7D31}{w} \in \mathcal{W}
Agent
Action A_t
Next State S_{t+1}
Reward vector
\textcolor{#ED7D31}{\boldsymbol{R}_t = [R_1 \dots R_m]}
Environment

Idea: Learn a set of policies \Pi = \{\pi_1, \dots, \pi_n\} containing
an optimal policy for any preference over the objectives.
v^{\pi}_{\textcolor{#ED7D31}{\boldsymbol{w}}} = \boldsymbol{v}^{\pi} \cdot \textcolor{#ED7D31}{\boldsymbol{w}}
= v^{\pi}_{1} \textcolor{#ED7D31}{w_1} + \dots + v^{\pi}_{m} \textcolor{#ED7D31}{w_m}
↑ Value of the m-th objective
Convex Coverage Set (CCS)
\text{CCS} \triangleq \{\mathbf{v}^{\pi} \in \mathcal{F} \mid \exists\, \mathbf{w} \text{ s.t. } \forall \mathbf{v}^{\pi'} \in \mathcal{F},\ \mathbf{v}^{\pi} \cdot \mathbf{w} \geq \mathbf{v}^{\pi'} \cdot \mathbf{w}\}

The optimal policy for any
linear combination of rewards
is in the CCS!
[Contrib. 1]
We can combine ideas from SFs and multi-objective RL
to solve the following previously open problem:
Optimal Policy Transfer
How to construct a set of policies \Pi such that combining them via GPI yields the optimal policy for any new linear task \boldsymbol{w} \in \mathcal{W}?
\Pi =

new task
+\;\;\boldsymbol{w}
GPI
optimal policy
\pi^*_{\boldsymbol{w}}
[Contrib. 1] First formal characterization of the connections between multi-objective and multi-task RL
SFOLS: Solves the previously open Optimal Policy Transfer problem:
Reuses previous solutions to directly identify optimal solutions for any new task
\Pi =

New Task
+\;\;\boldsymbol{w}
GPI
Optimal Policy
\pi^*_{\boldsymbol{w}}
Optimistic Linear Support and Successor Features as a Basis for Optimal Policy Transfer (ICML 2022)
Qualis A1
[Contrib. 1]

optimistic maximal improvement
[Contrib. 1]

\phi(s,a,s') \in \mathbb{R}^4

[Contrib. 2]
GPI-based prioritization to improve sample efficiency in MORL:
GPI Linear Support (GPI-LS)
GPI Prioritized Dyna (GPI-PD)
[Contrib. 2] Novel algorithms for optimal multi-objective decision-making with theoretical guarantees

Sample-Efficient Multi-Objective Learning via Generalized Policy Improvement Prioritization (AAMAS 2023)
+ JAAMAS Extended Paper
Qualis A1
[Contrib. 2]

Time vs. Killing Enemies

[Contrib. 3] Novel method for zero-shot transfer of solutions

h steps of planning with an approximate model
zero-shot GPI on states reachable in h steps
Training Tasks

zero-shot
Test Task

Multi-Step Generalized Policy Improvement by Leveraging Approximate Models (NeurIPS 2023)
Qualis A1
[Contrib. 3]

[Contrib. 3]

h-GPI outperforms competitors for all values of h using a learned model
Requires a significantly shorter planning horizon
[Contrib. 1] Formal characterization of the connections between MTRL and MORL [ICML 22]

[Contrib. 2] Novel algorithms that use GPI to define prioritization schemes [AAMAS 23]
[Contrib. 3] Robust model-based method for zero-shot policy transfer [NeurIPS 23]

Widely used by the community and industry:
A Toolkit for Reliable Benchmarking and Research in Multi-Objective Reinforcement Learning (NeurIPS 2023)
Qualis A1





In this thesis, we introduced:

Sample-Efficient Multi-Task and
Multi-Objective Reinforcement Learning
by Combining Multiple Behaviors
Lucas N. Alegre
Thank you!

39th CTD – CSBC 2026
July 22, 2026