Constrained Multi-Objective Reinforcement Learning with Max-Min Criterion


Abstract

Multi-Objective Reinforcement Learning (MORL) extends standard RL by optimizing policies with respect to multiple, often conflicting, objectives. While max-min MORL has emerged as an effective approach for promoting fairness, its applicability remains limited, particularly when constraints must be incorporated. In this paper, we propose a MORL framework that integrates the max-min criterion with explicit constraint satisfaction. We establish a theoretical foundation for the proposed framework and validate the resulting algorithm through convergence analysis and experiments in tabular settings. We further demonstrate the practical relevance of our approach in simulated building thermal control, multi-objective locomotion control, and greenhouse-gas-emission-aware traffic management. Across these domains, our method effectively balances fairness and constraint satisfaction in multi-objective decision-making.

1 Introduction↩︎

Reinforcement Learning (RL) is a powerful machine learning framework that enables an agent to derive effective decision-making policies by actively engaging with a given environment. In recent years, Multi-Objective Reinforcement Learning (MORL) has gained significant interest because many real-world control problems inherently involve multiple, often conflicting objectives [1][9]. MORL extends standard RL to handle simultaneous optimization of multiple objectives.

The central objective of MORL is to identify a policy that yields an expected cumulative return vector lying on the Pareto frontier of the set of achievable return vectors. A common strategy involves optimizing a scalarized function defined over multiple objective returns [1], [3]. This framework seeks to identify a policy \(\pi\) that maximizes a scalarized value \(f(J_1(\pi),\ldots,J_K(\pi))\), where each \(J_k(\pi)\) represents the expected discounted return for the \(k\)-th objective among \(K (\geq 2)\) objectives, and \(f: \mathbb{R}^K \to \mathbb{R}\) is a non-decreasing scalarization function such that \(J_k(\pi) \geq J_k(\pi'), 1 \leq k \leq K \Rightarrow f(J(\pi)) \geq f(J(\pi'))\). Thus, \(f\) plays a key role in imposing the designer’s preference among multiple objectives [10], [11].

Although much of the MORL literature employs a linear \(f\) (that is, the weighted sum: \(\max_\pi \sum_{k=1}^K w_k J_k(\pi)\) with fixed \(\{ w_k \}_{k=1}^K\)) due to its simplicity, the weighted sum does not always accurately represent the preference of a designer, especially regarding fairness among objectives [3], [12]. For instance, imagine a traffic light system managing an intersection where several roads converge with asymmetric arrival rates. Instead of simply aiming to reduce the total sum waiting time for all vehicles across the roads, the designer could prioritize fairness by minimizing the longest individual waiting time among the roads. This helps reduce localized congestion [13] and avoid severe delays for individual drivers.

As such, fairness-driven objectives frequently arise in real-world scenarios and are addressed beyond the standard weighted sum, such as max-min optimization [14] or proportionally fair optimization [15] in MORL. Specifically, if the Pareto frontier contains an equalization point at which \(J_1(\pi) = J_2(\pi) = \cdots = J_K(\pi)\) is satisfied, the max-min solution attains this point [14]. This notion of max-min fairness is valuable in many applications [14], [16][20], such as mitigating bottlenecks in cloud and edge resource management systems [17], [18] and analyzing societal fairness across diverse demographic groups [20].

However, because the standard max-min MORL framework is formulated as an unconstrained optimization problem, it cannot address many real-world problems that involve constraints which must be satisfied. For example, in resource allocation, a MORL-based scheduler may seek to maximize throughput and fairness across task queues while operating under a strict power consumption constraint [21], [22]. Similarly, in traffic management systems, a controller must minimize the maximum total waiting time across lanes while maintaining low greenhouse-gas emissions to ensure regional sustainability. Incorporating constraints into the max-min MORL framework therefore substantially broadens its practical applicability.

Despite this motivation, incorporating constraints into unconstrained max-min MORL is not straightforward. On the one hand, existing max-min MORL algorithms may lack solution exactness, for example by optimizing a lower bound of the original max-min objective [23], [24] or by being susceptible to inexact gradient estimates [12]. On the other hand, existing constrained single-objective RL methods [25][29] do not account for multi-objective reward settings with \(K \geq 2\) and fairness across those objectives. Moreover, their analyses are not directly applicable to our setting due to the non-differentiability of the max-min objective.

In this paper, we propose a MORL framework that maximizes max-min fairness among homogeneous objectives while simultaneously incorporating additional quantities as constraints. To address the non-differentiability and nonlinearity of the max-min objective, we reformulate the problem as a convex optimization over occupancy measures [30] and derive a convex program admitted by the dual problem. This reformulation enables us to handle the max-min criterion and the constraints in a unified manner. We further provide a detailed theoretical foundation, including convergence guarantees for the proposed algorithm.

Our main contributions are as follows:

\(\bullet\) We introduce a MORL framework that integrates the max-min criterion into constrained optimization and show that fairness and constraint satisfaction can be addressed in a unified manner.

\(\bullet\) We propose an iterative algorithm for constrained max-min MORL and establish its theoretical foundations, accompanied by a formal convergence analysis. We also empirically evaluate its behavior in tabular settings.

\(\bullet\) We demonstrate the practical relevance of our method through applications to simulated building thermal control, multi-objective locomotion control, and greenhouse-gas-emission-aware traffic management. Our approach consistently achieves a better balance between max-min fairness and constraint satisfaction than the considered baselines.

2 Background↩︎

A multi-objective Markov decision process (MOMDP) is represented as \(\langle \mathcal{S}, \mathcal{A}, T, \mu_0, r, \gamma \rangle\), where \(\mathcal{S}\) and \(\mathcal{A}\) are the sets of states and actions, respectively, \(T\) represents the transition probability distribution, \(\mu_0\) specifies the initial state distribution, and \(\gamma \in [0,1)\) is the discount factor. The reward function \(r: \mathcal{S} \times \mathcal{A} \rightarrow \mathbb{R}^{K+L}, ~K \geq 1, L \geq 0\) is vector-valued with its \(k\)-th element denoted by \(r^{(k)} ~ (1 \leq k \leq K+L)\) such that \(|r^{(k)}| \leq r^{(k)}_{\text{max}}\), where \(K+L\) is the total number of objectives. For simplicity, we abbreviate \(\forall s \in \mathcal{S}, ~ \forall a \in \mathcal{A}\) and \(\forall (s,a) \in \mathcal{S} \times \mathcal{A}\) as \(\forall s, ~ \forall a\) and \(\forall (s,a)\), respectively.

At each timestep, the agent selects an action \(a\) in the current state \(s\) according to its (stationary) policy \(\pi: \mathcal{S} \to \mathcal{P}(\mathcal{A})\), where \(\mathcal{P}(\mathcal{A})\) represents the set of probability distributions in the action space \(\mathcal{A}\). The occupancy measure is defined as \(\rho(s,a) := \sum_{s'} \mu_0(s') \sum_{t=0}^\infty \gamma^t \text{Pr}( s_t = s, a_t = a |s_0 = s', \pi^\rho)\) where \(\pi^\rho\) is the corresponding stationary policy induced by \(\rho\), expressed as \(\pi^\rho(a|s) = \frac{\rho(s,a)}{\sum_{a'}\rho(s,a')}\) [30]. Then, the vector return evaluated by \(\pi^\rho\) is given by \[\begin{align} J(\pi^\rho) &:=& [J_1(\pi^\rho),\cdots,J_{K+L}(\pi^\rho)]^\top \in \mathbb{R}^{K+L} \nonumber \\ &=&\mathbb{E}_{\pi^\rho} \left[ \sum_{t=0}^\infty \gamma^t r_t \right] = \sum_{(s,a)} r(s,a) \rho(s,a). \end{align}\]

In this paper, we define \(\mathbb{R}_{+}^L := \{ u \in \mathbb{R}^L | u_l \geq 0, ~ 1 \leq l \leq L \}\) and \(\Delta^K := \{ w \in \mathbb{R}^K | \sum_{k=1}^K w_k = 1; ~ w_k \geq 0, ~ 1 \leq k \leq K \}\), i.e., the \((K-1)\)-dimensional simplex.

3 Constrained Max-Min MORL Framework↩︎

3.1 Theoretical Foundation↩︎

We consider the MORL setting where the last \(L\) of the total \(K+L\) objectives should satisfy certain constraints. For theoretical development in this section, we assume that \(\mathcal{S}\) and \(\mathcal{A}\) are finite. The problem is formulated as follows: \[\begin{align} &&\max_{\pi^\rho} ~f( J_1(\pi^\rho),\cdots,J_{K}(\pi^\rho)) + \beta \sum_{s} \mathcal{H}_\rho(s) \rho(s) \tag{1} \\ && ~~s.t.~~ J_{K+l}(\pi^\rho) \geq C^{(l)}, ~~ l =1, \cdots, L \tag{2} \end{align}\] where \(\mathcal{H}_\rho(s) := - \sum_a \pi^\rho(a|s) \log \pi^\rho(a|s)\) is the entropy of \(\pi^\rho(\cdot|s)\), \(\rho(s) := \sum_a \rho(s,a)\) is the stationary state distribution in \(\mathcal{S}\), \(\beta > 0\) is a balancing coefficient, and \(\{ C^{(l)} \}_{l=1}^L\) is a set of threshold values. We assume a mild condition that the set \(\{ C^{(l)} \}_{l=1}^L\) is chosen by the designer such that the optimization in 1 and 2 is feasible, an assumption commonly made in the constrained MDP literature [26], [31].

In this paper, we set \(f\) the minimum function, i.e., \(f( J_1(\pi^\rho),\cdots,J_{K}(\pi^\rho))=\min_{1 \leq k \leq K} J_k(\pi^\rho)\). We note that the entropy term is required to eliminate the indeterminacy of the max-min solution without any regularization [12]. The problem reduces to the unregularized formulation as \(\beta \to 0\), with the optimality gap decreasing linearly:

Proposition 1. The gap between the optimal max-min value of the unregularized problem and that of the regularized problem in 1 and 2 with \(f=\min\) is upper bounded by \(\tfrac{\beta \log|\mathcal{A}|}{1-\gamma}\). (Proof: See Appendix 7.)

Proposition 1 shows that the regularized problem is a valid approximation of the unregularized criterion.

Since directly optimizing 1 and 2 with \(f=\min\) and \(J_k(\pi^\rho) = \mathbb{E}_{\pi^\rho}[\sum_{t=0}^\infty \gamma^t r^{(k)}_t]\) is non-trivial due to its non-differentiable and nonlinear structure, we address this challenge using the occupancy measure (i.e., stationary distribution [30]) formulation. The above optimization problem with \(f=\min\) can be rewritten as \[\label{eq:maxmin95obj} \max_{\rho \geq 0} \min_{1 \leq k \leq K} \bigg( \sum_{(s,a)} r^{(k)}(s,a) \rho(s,a) \bigg) + \beta \sum_{s} \mathcal{H}_\rho(s) \rho(s)\tag{3}\] \[\label{eq:maxmin95flow} \sum_{a'} \rho(s',a') = \mu_0(s') + \gamma \sum_{(s,a)} T(s' | s,a) \rho(s,a) ,~\forall s'\tag{4}\] \[\label{eq:maxmin95const95c} \sum_{(s,a)} c^{(l)}(s,a) \rho(s,a) \geq C^{(l)}, ~~ l =1, \cdots, L\tag{5}\] where 4 is the Bellman flow equation for the occupancy measure [30]. Here, we use the notation \(c^{(l)}(s,a) := r^{(K+l)}(s,a), ~l=1,\cdots,L\) to explicitly represent the dimensions associated with the constraint. For example, these quantities can be negative of some costs. Then the formulation in 3 , 4 , and 5 constitutes a convex optimization problem.

Now we derive a convex optimization problem admitted by the dual of 3 , 4 , and 5 , which forms the foundation of our algorithm. Given \((u,w)\), let \(v^{*}_{u, w}\) be the fixed point of the operator \(\mathcal{T}_{u,w}\): \[\begin{align} \label{eq:equivalent95fixed95point} [\mathcal{T}_{u,w} v](s) &= \beta \log \sum_a \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a) \nonumber \\ &+ \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v(s') \} ], \forall s. \end{align}\tag{6}\]

Proposition 2. The dual problem of 3 , 4 , and 5 admits the following convex optimization formulation: \[\label{eq:equivalent95objective} \min_{u \in \mathbb{R}_{+}^L, w \in \Delta^K} \mathcal{L}(u,w) = \sum_s \mu_0(s) v^{*}_{u,w}(s) - \sum_{l=1}^L u_l C^{(l)}.\qquad{(1)}\] (Proof: See Appendix 8.)

We emphasize that the optimization problem in ?? is not the dual per se of 3 , 4 , and 5 , since the dual problem itself does not involve the fixed point \(v^{*}_{u, w}\) in the objective. Consequently, establishing the convexity of ?? requires proving the convexity of \(v^{*}_{u, w}\) with respect to \((u,w)\), as explained in Appendix 8. Throughout our analysis, we assume that Slater condition holds [32], [33], which encompasses the standard constraint feasibility assumption [26], [31].

Proposition 2 hints that \(v^{*}_{u, w}\) can be obtained via 6 and the weights \((u,w)\) can be obtained by minimizing the loss \(\mathcal{L}(u,w)\) in ?? by some method. However, solving the optimization problem ?? is non-trivial because the fixed point \(v^*_{u,w}\) does not have a closed-form expression in terms of \((u,w)\). To address this issue, we derive the key properties of \(v^*_{u,w}\). We define a policy \(\pi^*_{u,w}\) as \[\label{eq:optimal95policy} \pi^*_{u,w}(a | s) := \frac{ \exp ( \frac{1}{\beta} Q^*_{u,w}(s,a) ) }{\sum_{a'} \exp ( \frac{1}{\beta} Q^*_{u,w}(s,a') ) }\tag{7}\] where \[\begin{align} \label{eq:eq10} Q^*_{u,w}(s,a) &:= \sum_{l=1}^L u_l c^{(l)}(s,a) + \sum_{k=1}^K w_k r^{(k)}(s,a) \nonumber \\ &+ \gamma \sum_{s'}T(s'|s,a) v^{*}_{u,w}(s'). \end{align}\tag{8}\]

Then, \(\pi^*_{u,w}\) is an optimal policy for the entropy-regularized RL [34] with a scalar reward \(\sum_{l=1}^L u_l c^{(l)}(s,a) + \sum_{k=1}^K w_k r^{(k)}(s,a), \forall (s,a)\). We derive the following theorem characterizing the relationship between \(\pi^*_{u,w}\) and the gradient of \(v^{*}_{u, w}\):

Theorem 3. For each \(s\), \(v^*_{u,w}(s)\) is differentiable in \((u,w) \in \mathbb{R}^{L+K}\), and its gradient \(\nabla v^*_{u,w}(s) = [\nabla_{u} v^*_{u,w}(s)^\top, \nabla_{w} v^*_{u,w}(s)^\top]^\top\) has the form of \[\label{eq:eq20} \nabla_{u} v^*_{u, w}(s) = v_c^{\pi^*_{u,w}}(s)~~and ~~ \nabla_{w} v^*_{u, w}(s) = v_r^{\pi^*_{u,w}}(s),\tag{9}\] where \(v_c^{\pi^*_{u,w}}(s) \in \mathbb{R}^L\) and \(v_r^{\pi^*_{u,w}}(s) \in \mathbb{R}^K\) are the value functions evaluated with the policy \(\pi^*_{u,w}\) for the constrained reward \(\{c^{(l)}\}_{l=1}^L\) and the unconstrained reward \(\{r^{(k)}\}_{k=1}^K\), respectively. (Proof: See Appendix 9.)

The main advantage of Theorem 3 lies not only in the simplicity of its gradient expression, but also in the fact that the gradient components corresponding to the unconstrained (\(w\)) and constrained (\(u\)) parts are obtained in a unified manner and are directly linked to the value function evaluated under \(\pi^*_{u,w}\). This unified form enables us to apply gradient descent to the optimization problem using the gradient \((\nabla_{v} v^*_{u,w}(s),\nabla_{w} v^*_{u,w}(s))\) combined with value iteration.

First, consider the unconstrained part. From ?? , \(\nabla_w \mathcal{L}(u,w)\) is given by \(\mathbb{E}_{s \sim \mu_0}[ v_r^{\pi^*_{u,w}}(s) ] \in \mathbb{R}^K\). Consequently, dimensions with smaller values of \(\mathbb{E}_{s \sim \mu_0}[ v_r^{\pi^*_{u,w}}(s) ]\) undergo smaller decreases in \(w_k\) under gradient descent, resulting in larger \(w_k\) relative to other dimensions after update with projection onto \(\Delta^K\). As a result, dimensions with smaller values receive larger updated weights, thereby realizing the max-min principle. The constrained component follows a similar logic to enforce feasibility, enabling the simultaneous achievement of max-min fairness and constraint satisfaction. In addition, our framework encompasses unconstrained max-min MORL (\(L=0\) with learned \(w\)) as well as both constrained and unconstrained RL with fixed \(w\) such as max-average RL (\(w=[1/K, \cdots, 1/K] \in \Delta^K\)).

A natural approach to solving the convex optimization problem in ?? , using Theorem 3, is projected gradient descent, since the variables \((u,w)\) lie in the convex set \(\mathbb{R}_+^L \times \Delta^K\). The convergence of projected gradient descent depends on the smoothness of the objective function [32], [35]. In our setting, \(\mathcal{L}(u,w)\) satisfies the following smoothness property, which is essential for establishing the convergence guarantee in Section 3.2:

Theorem 4. For each \(s, v^*_{u,w}(s)\) is smooth with respect to \(\| \cdot \|_2\) in \((u,w) \in \mathbb{R}^{L+K}\). Furthermore, \(\mathcal{L}(u,w)\) is \(\alpha\)-smooth with respect to \(\| \cdot \|_2\) in \((u,w) \in \mathbb{R}^{L+K}\) with \(\alpha := \frac{1}{\beta (1-\gamma)} \sum_{m=1}^{L+K} \left( \frac{r^{(m)}_{\text{max}}}{1-\gamma} \right)^2\). (Proof: See Appendix 10.)

In other words, \(\nabla_{(u,w)} \mathcal{L}(u,w)\) is \(\alpha\)-Lipschitz continuous in \(\| \cdot \|_2\). For the proof of Theorem 4, we first show that \(\nabla v^*_{u,w}\) is differentiable in \((u,w) \in \mathbb{R}^{L+K}\) (equivalently, that \(v^*_{u,w}\) is twice-differentiable) in Proposition 5. Then incorporating the closed-form expression of the Hessian of \(v^*_{u,w}(s)\) for each \(s\) into the generalized mean value inequality constitutes a key ingredient in the proof.

Let \(Q^{\pi^*_{u,w}}(s,a) \in \mathbb{R}^{L+K}\) be the action-value function evaluated with the policy \(\pi^*_{u,w}\), and define \(B^{\pi^*_{u,w}}(s) := \mathbb{E}_{a \sim \pi^*_{u,w}(\cdot|s)} \bigg[ ( Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot|s)} [Q^{\pi^*_{u,w}}(s,a')] ) ( Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot|s)} [Q^{\pi^*_{u,w}}(s,a')] )^\top \bigg] \in \mathbb{R}^{(L+K) \times (L+K)}\).

Proposition 5. For each \(s, v^*_{u,w}(s)\) is twice-differentiable in \((u,w) \in \mathbb{R}^{L+K}\). Let \(|\mathcal{S}|=p\), and suppose the states are enumerated as \(\{s_1, \cdots, s_p\}\). Then, the \((L+K)\times (L+K)\) Hessian matrix \(H[v^*_{u,w}(s_k)], ~ 1 \leq k \leq p\), has the form of \[\label{eq:hessian} H[v^*_{u,w}(s_k)] = \frac{1}{\beta} \sum_{l=1}^p [(I_p - \gamma T^{\pi^*_{u,w}})^{-1}]_{kl} B^{\pi^*_{u,w}}(s_l).\qquad{(2)}\] Here, \(I_p\) is the \(p\times p\) identity matrix; \(T^{\pi^*_{u,w}}\) is a \(p\times p\) matrix of which \(i\)-th row and \(j\)-th column element is given by \([T^{\pi^*_{u,w}}]_{ij} = \mathbb{E}_{a \sim \pi^*_{u,w}(\cdot | s_i)}[T(s_j | s_i, a)] ~ (1 \leq i,j \leq p)\); and \([(I_p - \gamma T^{\pi^*_{u,w}})^{-1}]_{kl}\) denotes the \(k\)-th row and \(l\)-th column element of \((I_p - \gamma T^{\pi^*_{u,w}})^{-1}\). (Proof: See Appendix 11.)

This result reflects the fact that the optimization problem in ?? is a convex program, as established in Proposition 2, owing to the positive semidefiniteness of ?? from \((I_p - \gamma T^{\pi^*_{u,w}})^{-1} = \sum_{i=0}^\infty (\gamma T^{\pi^*_{u,w}})^i\) and the relation \(H[\mathcal{L}(u,w)] = \mathbb{E}_{s \sim \mu_0} [H[v^*_{u,w}(s)]]\).

Note that \(Q^{\pi^*_{u,w}}(s,a)\) in Proposition 5 is different from \(Q^*_{u,w}(s,a)\) in 8 . By definition in the entropy-regularized RL, \(Q^*_{u,w}(s,a) \in \mathbb{R}\) is the cumulative scalarized return plus the cumulative entropy sum from \(\pi^*_{u,w}\). On the other hand, \(Q^{\pi^*_{u,w}}(s,a) \in \mathbb{R}^{L+K}\) is a cumulative sum of unconstrained rewards and constrained rewards from \(\pi^*_{u,w}\) without the entropy sum. Therefore, \([u;w]^\top Q^{\pi^*_{u,w}}(s,a)\) equals to \(Q^*_{u,w}(s,a)\) minus the cumulative entropy sum of \(\pi^*_{u,w}\).

3.2 Algorithm and Convergence Analysis↩︎

Based on the foundation built in the previous section, we propose an algorithm for constrained MORL with max-min fairness. Note that we need to jointly update the weights \((u,w)\) and the value function, which approximates \(v^{*}_{u, w}\). We adopt the following update method alternating between update of the value function and the weights \((u,w)\).

First, given a weight \((u,w)\), we update the value function to realize 6 . For this, we use an action value function \(Q\), which approximates \(Q^*_{u,w}\). If we plug 8 into the right-hand side of 6 , we have \(v^{*}_{u, w}(s) = [\mathcal{T}_{u,w} v^{*}_{u, w}](s) = \beta \log \sum_a \exp \left( \frac{Q^*_{u,w}(s,a)}{\beta} \right)\) for each \(s\). Using this form of \(v_{u,w}^*(s)\), we implement applying \(\mathcal{T}_{u,w}\) as updating the Q-function with the following: \[\label{eq:sql} \begin{align} Q(s,a) \leftarrow {} & [u; w]^\top [c; r] \\ & + \gamma \sum_{s'} T(s'|s,a)\, \beta \log \sum_{a'} \exp\!\left(\frac{Q(s',a')}{\beta}\right), \end{align}\tag{10}\]

for all \((s,a)\). In practice, 10 is iterated until the maximum change in the \(Q\)-function across successive iterations is below a given threshold.

Next, we compute an estimated gradient of \(\nabla_{(u,w)}\mathcal{L}(u,w)\) at the current weight \((u,w) = (u^m,w^m)\) where \(m=1,2,\cdots\) is the iteration index. We have shown that \(\nabla_{u} v^*_{u, w}(s) = v_c^{\pi^*_{u,w}}(s), ~ \nabla_{w} v^*_{u, w}(s) = v_r^{\pi^*_{u,w}}(s)\) for each \(s,\) where we denote \(v_c^{\pi^*_{u,w}}(s) \in \mathbb{R}^L\), \(v_r^{\pi^*_{u,w}}(s) \in \mathbb{R}^K\) as the value functions evaluated with the policy \(\pi^*_{u,w}\) for constrained reward \(c\) and unconstrained reward \(r\), respectively. Note that the policy is extracted from the updated Q-function in 10 based on the form of 7 . We then update \((u,w)\) using projected gradient descent: \[\label{eq:proj95grad} (u^{m+1},w^{m+1}) = \mathcal{P}_{K,L} [ (u^m,w^m) - l_{\text{w}} \nabla_{(u,w)}\mathcal{L}(u^m,w^m) ]\tag{11}\] where \(l_{\text{w}}\) is a learning rate for \((u,w)\) and \(\mathcal{P}_{K,L}[\cdot]\) is the projection onto the \(\mathbb{R}_+^L \times \Delta^K\). We use the convex optimization method from [36] to project onto the simplex \(\Delta^K\), and apply non-negativity clipping for projection onto \(\mathbb{R}_+^L\). Note that the projection onto \(\Delta^K\) is numerically stable as it is fully deterministic and avoids randomized procedures. In addition, its complexity is \(O(K \log K)\) [36] which is lightweight due to the logarithmic term.

We iterate this process for each \(m\), and Algorithm 1 presents the pseudocode of our algorithm.

Figure 1: Constrained Max-Min MORL Algorithm

We now provide our convergence analysis under the following assumption. Let \(\rho^{\pi^*_{u,w}}(s) = \sum_{t=0}^\infty \gamma^t \text{Pr}(s_t=s | \pi^*_{u,w}, T, \mu_0)\) be the discounted state visitation frequency of policy \(\pi^*_{u,w}\) at state \(s\) [30].

Assumption There exists at least one state \(s \in \mathcal{S}\) such that \(\rho^{\pi^*_{u,w}}(s) > 0\) and the centered action-value vectors \(\bigl\{\,Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot\mid s)}[\,Q^{\pi^*_{u,w}}(s,a')\,]\;:\;a\in\mathcal{A}\bigr\}\) span \(\mathbb{R}^{K+L}\).

This condition fails only in degenerate multi-objective settings when for every state \(s\in\mathcal{S}\) with \(\rho^{\pi^*_{u,w}}(s) > 0\), the set \(S_{\text{center}}(s)\) lies entirely within an affine subspace of dimension less than \(K+L\). Under Assumption, the Hessian \(H[\mathcal{L}(u,w)]\) is positive definite. We note that the property \(\pi^*_{u,w}(a|s) > 0\) for all \((s,a)\), implied by 7 , plays an essential role in ensuring the positive definiteness of \(H[\mathcal{L}(u,w)]\), which constitutes an additional technical benefit of the entropy regularization in 3 . (See Appendix 12.1 for more details.)

Moreover, under Slater condition with strict feasibility of the \(L\) constraints, we can restrict \(u \in \mathbb{R}^L_+\) to a compact set \(\mathcal{C}\) [29], ensuring the existence of \(\lambda := \min_{(u,w) \in \mathcal{C} \times \Delta^K} \lambda_{\text{min}}(H[\mathcal{L}(u,w)])\), where \(\lambda_{\min}(\cdot)\) denotes the minimum eigenvalue of a real symmetric matrix. (See Appendix 12.2 for more details.) Furthermore, \(\lambda\) satisfies \(0 < \lambda \leq \alpha\) [35].

Theorem 6 provides a formal guarantee of convergence for Algorithm 1 under approximate Q-updates.

Theorem 6. Let \((u^*,w^*)\) denote an optimal solution to ?? and \(l_{\text{w}} = 1/\alpha\). For each outer-loop index \(m \geq 1\) in Algorithm 1, let \(Q^*_{u^m, w^m}\) denote the fixed point of 10 with \([u;w] = [u^m;w^m]\), and let \(Q^m\) denote the Q-function after completing the \(m\)-th inner-loop update. For each \(m\), assume \(\| Q^m - Q^*_{u^m, w^m} \|_\infty < \epsilon\) for some \(\epsilon > 0\). Then for \(m \geq 1\), \[\begin{align} \label{eq:final95convergence95result} \| [u^m;w^m] - [u^*;w^*] \|_{2} &\leq (1-\frac{\lambda}{\alpha})^m \| [u^0;w^0] - [u^*;w^*] \|_{2} \nonumber \\ &+ \frac{\sqrt{|\mathcal{S}|}}{\lambda} \sqrt{\sum_{i=1}^{K+L} \{ r^{(i)}_{\text{max}} \}^2} \frac{1+\gamma }{(1-\gamma)^2} \epsilon. \end{align}\tag{12}\] (Proof: See Appendix 12.3.)

Theorem 6 establishes that the error decreases geometrically at rate \(O\left((1-\tfrac{\lambda}{\alpha})^m\right)\), up to \(O(\epsilon)\). For completeness, Appendix 12.4 provides an analysis of the case without Assumption, supporting the practical implementation of our algorithm beyond Assumption in Section 5.2.

4 Related Work↩︎

MORL   The dominant approach in MORL is utility-based [1], [3], where the goal is to find an optimal policy \(\pi^* = \arg \max_{\pi} f(J(\pi))\) given a non-decreasing scalarization function \(f: \mathbb{R}^K \rightarrow \mathbb{R}\). When \(f\) is linear, each non-negative weight vector defines a scalarized MDP [37], motivating work on learning a single model capable of generating policies across a continuum of preferences

[2], [9], [38][42]. This family of approaches is known as multi-policy MORL [1], [3].

For nonlinear scalarization functions, however, Bellman optimality no longer holds in its standard form due to the loss of linearity, making optimization substantially more difficult [1], [3]. Algorithms that directly optimize a given nonlinear scalarized objective fall under the single-policy MORL category [1], [3]. Most work in this category considers a welfare function [8], [10], [11] as the nonlinear scalarization \(f\). Note that single-policy and multi-policy methods are complementary rather than interchangeable [1], [3].

Unconstrained Max-min MORL   Max-min MORL studies the case where \(f=\min\), aiming to enforce max-min fairness. This formulation is useful in many applications such as mitigating bottlenecks in cloud and edge resource management systems [17], [18]. Several studies optimize proxy objectives related to the unconstrained max-min formulation, for example, maximizing a conservative lower bound of \(\mathbb{E}_\pi \left[ \min_{1 \leq k \leq K}( \sum_{t=0}^\infty \gamma^t {r_t^{(k)}} ) \right]\) [23], [24], or maximizing the total return while enforcing per-group performance constraints [20].

[12] proposed a tractable approach for exact unconstrained max-min MORL using Gaussian smoothing to estimate gradients. However, this approach requires maintaining multiple network copies, increasing computational overhead. Furthermore, the gradient estimates are inherently inexact, as Gaussian smoothing of a convex function yields a convex upper bound rather than the true function [43]. In contrast, our method produces direct, theoretically grounded gradient estimates and extends naturally to constrained MORL. Recently, [44] introduced an alternative unconstrained max-min MORL formulation based on a game-theoretic framework [45], [46]. However, their method does not address the incorporation of constraints.

Constrained RL   Many approaches to constrained MDPs reformulate the problem with a scalar reward (i.e., a special case of 1 and 2 with \(K=1\) and without \(f\)) into an unconstrained one by augmenting the objective with a weighted sum of constraint violations, typically via a Lagrangian formulation [26][29], [31], [47], [48]. The motivation for this line of work is that the Lagrangian relaxation exhibits no duality gap, even when the original problem is non-convex with respect to the policy [48]. However, these approaches do not consider the multi-objective reward setting in 1 and 2 with \(K \geq 2\). Moreover, applying them directly to our setting is non-trivial, since \(f = \min\) introduces non-differentiability in 1 .

To resolve this, we reformulate our problem as a convex program using occupancy measures and then derive another convex program admitted by the dual problem. In particular, we show that both the max-min criterion and the constraints can be satisfied by jointly updating the weights \(u\) and \(w\), a simple yet effective approach that to our knowledge has not been explored in the constrained MDP literature. Although [49] also leverages convex analysis with occupancy measures, its focus is on constrained single-objective RL with a scalar reward (i.e., \(K=1\)) in an offline setting. Unlike our work, it neither addresses max-min fairness across multiple objectives nor provides a convergence analysis.

Several recent works have incorporated constraints into MORL, but under settings that differ from our framework which explicitly integrates max-min optimization. Some approaches learn preference-vector-conditioned policies to explore trade-offs among objectives [7], [50]. While complementary to our work, it remains unclear how to select a preference vector such that the resulting policy exactly corresponds to the optimal solution under nonlinear scalarizations, such as the max-min criterion. In contrast, our approach directly solves a single-policy constrained max-min optimization problem, mirroring the distinction between single- and multi-policy methods in unconstrained MORL [1], [3]. [51] study offline constrained MORL, where policies are trained on offline data and adapted to target preferences using additional demonstrations. In contrast, our work focuses on online learning and does not rely on offline data or demonstrations. [6] improve Pareto frontier coverage in unconstrained MORL by training multiple policies via constrained optimization, rather than directly addressing constrained MORL.

5 Experiments↩︎

In this section, we present experimental validations of our theoretical analysis and algorithm. Section 5.1 examines the convergence behavior of our method in tabular settings. In Section 5.2, we further demonstrate the practical relevance of our approach through applications beyond tabular settings including building thermal control, multi-objective locomotion control, and greenhouse-gas-emission-aware traffic management.

5.1 Tabular Settings↩︎

We conducted experiments in tabular settings to evaluate the convergence of our algorithm. Constrained MOMDPs were randomly generated, after a feasibility check, within two widely used classes of structured MDPs. (See Appendix 13 for details on the experimental setup, feasibility check, and baselines.) First, bipartite state graphs partition the state space into two disjoint subsets, enforcing transitions between them at alternating time steps. This structure captures temporal dynamics in systems with role alternation or interleaving phases [52]. Second, hierarchical MDPs organize the state space into multiple levels or stages, where transitions flow sequentially from one level to the next. This reflects tasks with subgoals or temporal abstraction [53].

The optimal value for each constrained MOMDP was computed by solving 3 , 4 , and 5 with \(\beta=0\) via linear programming (LP), and performance was evaluated as the error relative to these LP-optimal values. The goal of this experiment is to verify that our algorithm, which simultaneously accounts for both max-min fairness and constraint satisfaction, achieves near-optimal performance compared to methods that lack either max-min fairness or constraint enforcement. We compare our constrained max-min algorithm against unconstrained max-min, constrained max-average which does not enforce max-min fairness, and unconstrained max-average baselines.

5pt

Table 1: Optimal value error in tabular constrained MOMDPs
Algorithm Optimal value error \((\downarrow)\)
Constrained max-min \(0.004\)
Unconstrained max-min \(0.325\)
Constrained max-average \(0.657\)
Unconstrained max-average \(1.008\)
Table 2: Optimal value errors of the constrained max-min MORL algorithm for different values of \(\beta\)
\(\beta\) 0.1 0.03 0.01 0.003 0.001
Error \((\downarrow)\) \(0.061\) \(0.004\) \(0.009\) \(0.020\) \(0.021\)

Table 1 shows that the constrained max-min approach achieves near-optimal values compared to the other methods. Baselines that incorporate neither max-min fairness nor constraint satisfaction exhibit the largest optimal value errors. We further analyze the effect of \(\beta\) on convergence. As shown in Table 2, \(\beta=0.03\) and \(0.01\) yield the smallest errors. The slight increase in optimal value error when \(\beta < 0.01\) may be attributed to the first term in 12 to converge more slowly as the iteration count \(m \to \text{ITER}\), the maximum number of iterations used in Algorithm 1.

5.2 Extension to Applications↩︎

In this section, we extend our algorithm to practical applications, demonstrating that our method is not limited to tabular settings. The Building (Section 5.2.1) and MoAnt (Section 5.2.2) environments have continuous state and action spaces, while the Traffic environment (Section 5.2.3) has a continuous state space and a discrete action space. To ensure stable gradient estimation of our algorithm in continuous state spaces, we parameterize a gradient network \(g_\theta(s) \in \mathbb{R}^{L+K}\) to estimate \(\nabla_u v^{*}_{u, w}(s)\) and \(\nabla_w v^{*}_{u, w}(s)\), following Theorem 3. Implementation details for deep neural network policies, including gradient estimation and our constrained max-min algorithm for practical applications, are provided in Appendix 14.1. Our sourcecode can be found in https://github.com/Giseung-Park/Constrained-Maxmin-MORL.

5.2.1 Building Thermal Control↩︎

We consider a thermal control environment for a building from SustainGym [54]. The environment models a large commercial building with \(N_{\text{zone}}=23\) thermal zones, where the system controls the zone temperatures under changing outside conditions and occupancy. At each timestep, the system observes the current temperatures of the \(N_{\text{zone}}\) zones together with additional indicators (e.g., outdoor temperature, global horizontal irradiance, ground temperature, occupancy power, carbon intensity, and electricity price). Then the system outputs a continuous action that controls the heating/cooling power of each zone.

The zones are grouped into \(K=3\) categories corresponding to the bottom, middle, and top floors of the building. The unconstrained reward is a \(K\)-dimensional vector, where each component quantifies the comfort of the corresponding group based on the proximity of the zone temperatures to their target values. The cost is defined as the total power consumption of the system, normalized by the environment. The system’s objective is to control the building so as to maximize the minimum cumulative comfort across all groups, while satisfying the energy constraint with its designed threshold value \(C_{th}=180\).

We consider five baselines: (i) a random policy that samples a continuous action at each timestep (Random), (ii) unconstrained max-average SAC (MA-SAC) [55], (iii) max-average SAC with a Lagrangian relaxation (MA-SAC-L) [26], [56], (iv) unconstrained max-min MORL algorithm with Gaussian smoothing (Max-min GS) [12], and (v) unconstrained max-min MORL algorithm based on a two-player zero-sum game (ARAM) [44]. Each of the baselines lacks either max-min fairness ((iii)), constraint handling ((iv),(v)), or both ((i), (ii)). We report the mean performance computed across five random seeds. (See Appendices 14.2 and 14.3 for the implementation of the max-min GS baseline and for details on the environment and the hyperparameters, respectively.)

Table 3: Cumulative cost sum and worst-group comfort return in the building thermal control environment, with the two constraint-satisfying algorithms highlighted in bold
Algorithm
\((C_{th}=180)\)
return \((\uparrow)\)
Random \(227.8\) \(433.9\)
MA-SAC \(203.7\) \(641.5\)
MA-SAC-L \(\boldsymbol{171.4}\) \(620.9\)
Ours \(\boldsymbol{178.7}\) \(639.8\)
Max-min GS \(202.1\) \(653.6\)
ARAM \(276.9\) \(664.3\)
Table 4: Ablation study on the impact of weight learning
Algorithm
\((C_{th}=180)\)
return \((\uparrow)\)
Ours \(178.7\) \(639.8\)
w/o \(w\) update \(178.1\) \(626.7\)
w/o \(u\) update \(200.7\) \(653.5\)
w/o \((u,w)\) upd. \(222.0\) \(646.9\)

Table 3 presents the cumulative cost sum and the worst-group comfort return. Compared to the Random baseline, MA-SAC increases the worst-group comfort return but still fails to satisfy the power consumption constraint, with its cost sum exceeding the threshold \(C_{th}=180\). While MA-SAC-L satisfies the power constraint, it does so at the cost of a lower worst-group comfort return compared to MA-SAC. However, our method achieves a higher worst-group comfort return relative to MA-SAC-L, while still adhering to the power constraint. We note that both Max-min GS and ARAM violate the power constraint.

Table 4 shows that removing the max-min-related \(w\) update reduces the worst-group comfort return, while ablating the constraint-related \(u\) update causes constraint violations. These results confirm that our method effectively balances max-min performance with constraint satisfaction. (For further analysis of the effect of \(\beta\) and the computational complexity, see Appendix 14.4.)

5.2.2 Multi-objective Locomotion Control↩︎

Table 5: MoAnt-v5 results over five seeds, with the two constraint-satisfying algorithms highlighted in bold
Algorithm
\((C_{th}=50)\)
return \((\uparrow)\)
Random 146.5 48.2
MA-SAC 275.3 98.8
MA-SAC-L 47.8 83.0
Ours 28.3 92.2
Max-min GS 111.7 92.7
ARAM 620.7 101.3

We include MoAnt-v5 environment [57], where the agent learns locomotion to maximize \(x\) and \(y\) velocities while keeping energy consumption under a threshold. We consider an asymmetric case where movement in the \(x\) direction is attenuated by friction at rate 0.3. The velocities \((v_x, v_y)\), combined with bonus terms, constitute a two-dimensional reward, while the control cost is treated as a constraint. (See Appendix 14.5 for details on hyperparameters.)

Table 5 shows that both our method and MA-SAC-L satisfy the constraints, but our method achieves superior max-min performance. In contrast, the other four algorithms severely violate the constraints, as they do not explicitly account for constraint satisfaction. Overall, our algorithm balances constraint satisfaction and max-min fairness. (See Appendix 14.6 for the result with \(C_{th}=40\).)

5.2.3 Greenhouse-gas-emission-aware Traffic Management↩︎

We extend our method to an environment with a larger objective space. We note that MORL benchmark environments, particularly those with more than four objectives, remain limited [3]. Recently, [42] proposed a traffic signal control environment with a 16-dimensional objective vector to evaluate scalability in the MORL setting. We modify this environment to incorporate a setting aimed at reducing greenhouse-gas emissions while maintaining max-min fairness.

In a simulated 16-lane four-way intersection, the agent receives a 37-dimensional continuous state encoding traffic information at each timestep. The agent manages the traffic lights by choosing the traffic signal phase as its action. The feedback signal consists of a 16-dimensional reward vector, where each component represents the negative total waiting time for a corresponding lane. Following prior work [42], we evaluate performance in an asymmetric traffic flow scenario. Vehicles enter from each direction according to predefined inflow proportions and may proceed straight or make left or right turns.

Table 6: Traffic signal control results over five seeds, with the two constraint-satisfying algorithms highlighted in bold
Algorithm
\((C_{th}=70{,}000)\)
return \((\uparrow)\)
Random 82,698 \(-27{,}523\)
MA-PGO 77,757 \(-20{,}434\)
MA-CPGO 67,887 \(-27{,}830\)
Ours 69,147 \(-25{,}229\)
Max-min GS 73,162 \(-21{,}527\)
ARAM 88,748 \(-19{,}700\)

Our goal is to achieve fair traffic flow across all lanes while enforcing a constraint on total \(\isotope{CO}_2\) emissions, contributing to a more sustainable traffic control system. Since this environment operates with a discrete action space, we replaced the MA-SAC and MA-SAC-L baselines with policy-gradient-optimization (PGO) variants, namely MA-PGO and MA-CPGO, respectively. (See Appendix 14.7 for implementation details.)

As shown in Table 6, both our method and MA-CPGO satisfy the constraints; however, our approach achieves better max-min performance. In contrast, the remaining four algorithms exhibit constraint violations. Overall, our method consistently balances max-min fairness with constraint satisfaction.

6 Conclusion↩︎

We have proposed a MORL framework that integrates max-min fairness with constraint satisfaction. Our approach offers flexibility in modeling problems that satisfy fairness and operational constraints. We establish a theoretical foundation and develop an algorithm that achieves a balance between max-min fairness and constraint satisfaction, as demonstrated in our experiments. A discussion of limitations and future work is provided in Appendix 15.

Acknowledgment↩︎

We sincerely thank anonymous reviewers for their valuable feedback during the review process. This work was supported by the Institute of Information & Communications Technology Planning & Evaluation (IITP) under Grant No. RS-2024-00457882 (AI Research Hub Project) and the National Research Foundation of Korea (NRF) grant (2022K1A3A1A31093462) funded by the Korea government (MSIT), and by the Israel Ministry of Innovation, Science & Technology, under Grant No. 1001556423 and the ISF grant 2197/22.

Impact Statement↩︎

In this work, we propose a MORL algorithm that integrates max-min fairness with constraint satisfaction. First, max-min MORL plays a critical role in promoting fairness across objectives in domains such as traffic management and resource allocation. Unfair results can lead to user dissatisfaction and, in turn, degrade overall system performance, for example, by contributing to traffic congestion. Second, incorporating constraints into RL systems is essential for the responsible development of AI systems, especially given real-world limitations on resources such as electricity, power consumption, and fossil fuels.

Our work advances the goal of sustainable AI by simultaneously incorporating fairness and resource constraints into decision-making. This contrasts to traditional methods that prioritize performance alone, often overlooking concerns of equity and efficient resource use. We believe our framework has the potential to make a meaningful and positive impact on the broader AI community, not only in resource allocation but also in emerging areas such as fair and safe alignment of large language models.

7 Proof on Optimality Gap↩︎

Proof. With a slight abuse of notation, let \(J(\pi):= [J_1(\pi),\cdots,J_{K}(\pi)]^\top \in \mathbb{R}^{K}\) and let \(\mathcal{H}(\pi)\) denote the expected cumulative entropy of \(\pi\). We express the optimization of 1 and 2 with \(f=\min\) as follows: \[\label{eq:append95reg95prob95pi} \max_{ \pi \in \Pi_{\text{feas}} } \min_{1 \leq k \leq K} J_k(\pi) + \beta \mathcal{H}(\pi)\tag{13}\] where \(\Pi_{\text{feas}} := \left\{ \pi \,\middle|\, \mathbb{E}_{\mu_0, \pi} \left[ \sum_{t=0}^\infty \gamma^t c_t^{(l)} \right] \ge C^{(l)}, \quad \forall l = 1, \dots, L \right\}\) and it is assumed to be non-empty under the typical assumption in constrained RL [26], [31].

Let the optimal solution to the regularized problem in 13 be \(\pi_r^*:= \arg \max_{ \pi \in \Pi_{\text{feas}} } \min_{1 \leq k \leq K} J_k(\pi) + \beta \mathcal{H}(\pi) = \arg \max_{ \pi \in \Pi_{\text{feas}} } \min_{w} \langle w, J(\pi) \rangle + \beta \mathcal{H}(\pi)\) where \(\min_{w \in \Delta^K}\) is abbreviated as \(\min_w\) for brevity. Let \(w^*(\pi) := \arg \min_w \langle w,J(\pi)\rangle\) and \(w_r^* := w^*(\pi_r^*)\).

Let the optimal solution to the unregularized problem be \(\pi^*:=\arg \max_{ \pi \in \Pi_{\text{feas}} } \min_{w} \langle w, J(\pi) \rangle\) and \(w^* = w^*(\pi^*)\). Let the optimal max-min value of the unregularized problem be \(V_{w^*}^{\pi^*} := \langle w^*,J(\pi^*)\rangle\). Similarly, let the optimal value of the regularized problem be \(V_{w_r^*}^{\pi_r^*} := \langle w_r^*,J(\pi_r^*)\rangle\). For simplicity, we abbreviate \(\max_{ \pi \in \Pi_{\text{feas}} }\) as \(\max_\pi\) below.

First, we have the following: \[V_{w_r^*}^{\pi_r^*} + \beta {\mathcal{H}}(\pi_r^*) = \max_\pi\min_w \langle w,J(\pi)\rangle+\beta \mathcal{H}(\pi) \ge \min_w \langle w,J(\pi^*)\rangle+\beta \mathcal{H}(\pi^*) =\langle w^*,J(\pi^*)\rangle+\beta \mathcal{H}(\pi^*) = V_{w^*}^{\pi^*} +\beta \mathcal{H}(\pi^*).\]

Since the stepwise entropy of \(\pi\) is upper-bounded by \(\log |\mathcal{A}|\), we have \(0\le \mathcal{H}(\pi)\le \frac{\log|\mathcal{A}|}{1-\gamma}\) and obtain

\[\label{eq:append195upper95bound} V_{w^*}^{\pi^*} -V_{w_r^*}^{\pi_r^*} \le \frac{\beta\log|\mathcal{A}|}{1-\gamma}.\tag{14}\]

Similarly, we have the following: \[\label{eq:append195lower95bound} V_{w^*}^{\pi^*} = \max_\pi\min_w \langle w,J(\pi)\rangle \ge \min_w \langle w,J(\pi_r^*)\rangle =\langle w^*_r,J(\pi_r^*)\rangle = V_{w_r^*}^{\pi_r^*}.\tag{15}\]

Combining 14 and 15 , we obtain that the optimality value gap ranges as \(0 \le V_{w^*}^{\pi^*} - V_{w_r^*}^{\pi_r^*} \le \frac{\beta\log|\mathcal{A}|}{1-\gamma}\). ◻

8 Proof on Equivalent Optimization↩︎

Proof. The dual problem of 3 , 4 , and 5 is rewritten as follows: \[\begin{align} \label{eq:dual95of95maxent95maxmin95const} & {\min_{u \geq 0}} \min_{w\geq 0, v} \min_{\xi \geq 0} \max_{\rho,b} \Bigg[ b(1 - \sum_{k=1}^K w_k) - \beta \sum_{s,a} \rho(s,a) \log \frac{\rho(s,a)}{\sum_{a'} \rho(s,a')} \nonumber \\ & + \sum_{s}\mu_0(s) v(s) + \sum_{s,a} \xi(s,a)\rho(s,a) {- \sum_{l=1}^L u_l C^{(l)}} \nonumber \\ & + \sum_{s,a} \rho(s,a) [\sum_{k=1}^K w_k r^{(k)}(s,a){+\sum_{l=1}^L u_l c^{(l)}(s,a)} + \gamma \sum_{s'} T(s' | s,a) v(s') - v(s) ] \Bigg]. \end{align}\tag{16}\]

Here \(b\) is an auxiliary variable satisfying \(\sum_{s,a} r^{(k)}(s,a) \rho(s,a) \geq b, ~ 1 \leq k \leq K\). Let \(\eta_{u, v,w}(s,a) := \sum_{k=1}^K w_k r^{(k)}(s,a) +\sum_{l=1}^L u_l c^{(l)}(s,a) + \gamma \sum_{s'} T(s' | s,a) v(s') - v(s)\). We apply KKT conditions.

1. Stationarity condition gives \[\label{eq:stationarity95rho} \forall (s,a), ~~ - \beta \log \frac{\rho(s,a)}{\sum_{a'} \rho(s,a')} + \xi(s,a) + \eta_{u,v,w}(s,a) = 0\tag{17}\] and \[\label{eq:w95sum95to951} 1 - \sum_{k=1}^K w_k = 0.\tag{18}\]

2. Complementary slackness condition gives \[\label{eq:slackness} \forall (s,a), ~~ \xi(s,a) \rho(s,a) = 0.\tag{19}\]

From 17 , we derive \[\label{eq:kkt95intermediate95const} \forall (s,a), ~~ \frac{\rho(s,a)}{\sum_{a'} \rho(s,a')} = \exp \left( \frac{\xi(s,a) + \eta_{u,v,w}(s,a)}{\beta} \right)\tag{20}\] so \(\rho(s,a) > 0\) and \(\xi(s,a) = 0\) from 19 . Therefore, \[\label{eq:optimal95policy95append} \forall (s,a), ~~ \frac{\rho(s,a)}{\sum_{a'} \rho(s,a')} = \exp \left( \frac{\eta_{u,v,w}(s,a)}{\beta} \right).\tag{21}\]

Inserting 18 and 21 , we obtain: \[\label{eq:joint95dual95maxent95maxmin95obj95const} \min_{u \in \mathbb{R}_{+}^L} \min_{v,w} \sum_s \mu_0(s) v(s) - \sum_{l=1}^L u_l C^{(l)}\tag{22}\] \[\label{eq:joint95dual95maxent95maxmin} \forall s, ~~ v(s)= \beta \log \sum_a \exp [ \frac{1}{\beta} \{ \sum_{k=1}^K w_k r^{(k)}(s,a) +\sum_{l=1}^L u_l c^{(l)}(s,a) + \gamma \sum_{s'}T(s'|s,a)v(s') \} ] := [\mathcal{T}_{u,w} v](s)\tag{23}\] \[\sum_{k=1}^K w_k = 1; ~ w_k \geq 0 ~~ \forall 1 \leq k \leq K.\] where 23 is derived from \(\sum_a \exp \left( \frac{\eta_{u,v,w}(s,a)}{\beta} \right) = 1, ~~ \forall s\), and strong duality holds under Slater condition [32]. Since \(\mathcal{T}_{u,w}\) is a contraction mapping [34], [58], it has the unique fixed point \(v^*_{u,w}\). Therefore, \(v = v^*_{u,w}\) is the only feasible solution that satisfies 23 and we have the following: \[\min_{u \in \mathbb{R}_{+}^L, w \in \Delta^K} \mathcal{L}(u,w) = \sum_s \mu_0(s) v^{*}_{u,w}(s) - \sum_{l=1}^L u_l C^{(l)}.\] Under Slater condition, this optimization attains the same optimal value as in the original convex optimization. The convexity of \(v^{*}_{u,w}(s)\) follows from the proof of Theorem 4.1 in [12]. ◻

9 Proof of Differentiability↩︎

Proof. For notational simplicity, we first present the theorem for the case \(L=0\) (i.e., without constraints) and then extend the result to the case \(L \geq 1\). This holds because, given \((u,w) \in \mathbb{R}^{L+K}\), the mapping \(\mathcal{T}_{u,w}\) is defined by \([\mathcal{T}_{u,w} v](s) = \beta \log \sum_a \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a) + \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v(s') \} ], \forall s\), and we can regard the concatenation of \(c(s,a)\) and \(r(s,a)\) as a new vector reward of size \(L+K\) with its weight \((u,w)\). Therefore, we use the notation of the following mapping \([\mathcal{T}_{w} v](s) = \beta \log \sum_a \exp [ \frac{1}{\beta} \{ \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v(s') \} ], \forall s\).

Let \(|\mathcal{S}|=p\). We define \(F(w,v):=v-\mathcal{T}_w v\), \(F: \mathbb{R}^{K} \times \mathbb{R}^{p} \to \mathbb{R}^{p}\). Let \(v^*_w\) be the unique fixed point of \(\mathcal{T}_w\). Then \(F(w,v^*_w)=v^*_w-\mathcal{T}_w v^*_w = 0\). Here \(v^*_w\) is implicitly expressed with respect to (w.r.t.) \(w\), and we aim to analyze \(v^*_w\) using implicit function theorem.

First of all, \(F: \mathbb{R}^{K} \times \mathbb{R}^{p} \to \mathbb{R}^{p}\) is a continuously differentiable function. For each \(s\), \(F(w,v)(s)=v(s)- [\mathcal{T}_w v](s) = v(s) - \beta \log \sum_a \exp [ \frac{1}{\beta} \{ \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v(s') \} ]\) which is a composition of linear, logarithm, summation, exponential, and linear functions.

Now we fix \(w\) and check whether the Jacobian matrix \(\partial_v F(w,v)|_{v = v^{*}_w} \in \mathbb{R}^{p \times p}\) is invertible where \([\partial_v F(w,v)|_{v = v^{*}_w}]_{ij} = \frac{\partial F(w,v)(s_i)}{\partial v(s_j)}|_{v = v^{*}_w}\). We have \(\partial_v F(w,v) = I_p -\partial_v [\mathcal{T}_w v]\) where \(I_p\) is the \(p \times p\) identity matrix. Then \[\frac{\partial [\mathcal{T}_w v](s_i)}{\partial v(s_j)}|_{v = v^{*}_w} = \gamma \mathbb{E}_{a \sim \pi^*_w(\cdot | s_i)}[T(s_j | s_i, a)]\] where \[\label{eq:pi95star95w95only} \pi^*_w(a | s) = \frac{ \exp [ \frac{1}{\beta} \{ \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v^{*}_w(s') \} ]}{\sum_{a'} \exp [ \frac{1}{\beta} \{ \sum_{k=1}^K w_k r^{(k)}(s,a') + \gamma \sum_{s'}T(s'|s,a') v^{*}_w(s') \} ]}.\tag{24}\]

If we denote \(T(\cdot |s,a) := [T(s_1|s,a)\cdots T(s_p|s,a)]\), we have \[\partial_v F(w,v)|_{v = v^{*}_w} =I_p -\gamma\begin{bmatrix} \mathbb{E}_{a \sim \pi^*_w(\cdot | s_1)}[T(\cdot | s_1, a)] \\ \vdots\\ \mathbb{E}_{a \sim \pi^*_w(\cdot | s_p)}[T(\cdot | s_p, a)] \end{bmatrix} =: I_p -\gamma\begin{bmatrix} T^{\pi^*_w}(\cdot | s_1) \\ \vdots\\ T^{\pi^*_w}(\cdot | s_p) \end{bmatrix}\] where \(T^{\pi^*_w}(s_j | s_i) = \mathbb{E}_{a \sim \pi^*_w(\cdot | s_i)}[T(s_j | s_i, a)] =: [T^{\pi^*_{w}}]_{ij}\). Then \(I_p - \gamma T^{\pi^*_w}\) is invertible since \(T^{\pi^*_w}\) is a row-stochastic square matrix [59].

Therefore, \(\partial_v F(w,v)|_{v = v^{*}_w}\) is invertible. By implicit function theorem, there exists an open set \(U\subset\mathbb{R}^K\) containing \(w\) such that there exists a unique continuously differentiable function \(h:U\to\mathbb{R}^{p}\) such that \(h(w)=v^{*}_w\) and \(F(w',h(w'))=0\), i.e., \(h(w')=\mathcal{T}_{w'} h(w')\) for all \(w'\in U\). Since \(h(w')\) is the unique fixed point of \(\mathcal{T}_{w'}\), \(h(w') = v^{*}_{w'}, \forall w'\in U\). If we use the implicit function theorem for all \(w \in \mathbb{R}^K\), we can conclude that \(v = v^*_w\) is a unique continuously differentiable function in \(w \in \mathbb{R}^K\) that satisfies \(v = \mathcal{T}_w v\).

Moreover, for \(1 \leq k \leq K\), \[\frac{\partial [\mathcal{T}_w v](s_i)}{\partial w_k}|_{v = v^{*}_w} = \mathbb{E}_{a \sim \pi^*_w(\cdot | s_i)}[r^{(k)}(s_i, a)].\] With a slight abuse of notation, if we denote \(r(s,a) := [r^{(1)}(s,a)\cdots r^{(K)}(s,a)]\), we have \[\partial_w F(w,v)|_{v = v^{*}_w} = -\begin{bmatrix} \mathbb{E}_{a \sim \pi^*_w(\cdot | s_1)}[ r(s_1, a)] \\ \vdots\\ \mathbb{E}_{a \sim \pi^*_w(\cdot | s_p)}[r(s_p, a)] \end{bmatrix} =: -\begin{bmatrix} r^{\pi^*_w}(s_1) \\ \vdots\\ r^{\pi^*_w}(s_p) \end{bmatrix}\] where \(r^{\pi^*_w}(s) = \mathbb{E}_{a \sim \pi^*_w(\cdot | s)}[ r(s, a)] \in \mathbb{R}^{1 \times K}\). By implicit function theorem, we have \[\label{eq:grad95all} \begin{bmatrix} \nabla_{w} v^*_{w}(s_1)^\top \\ \vdots\\ \nabla_{w} v^*_{w}(s_p)^\top \end{bmatrix} = - [\partial_v F(w,v)|_{v = v^{*}_w}]^{-1} \partial_w F(w,v)|_{v = v^{*}_w} = (I_p - \gamma T^{\pi^*_w})^{-1} r^{\pi^*_w}.\tag{25}\] Note that the \(k\)-th (\(1 \leq k \leq K\)) column of 25 is equivalent to the policy evaluation of \(\pi^*_w\) considering a scalar reward function \(r^{(k)}\) [60], [61]. We denote the value function as \(v^{\pi^*_w}_k \in \mathbb{R}^p\). Then \[\frac{\partial v^*_w (s)}{\partial w_k} = v^{\pi^*_w}_k(s), ~ \forall s.\] If we denote \(v^{\pi^*_w}(s) = [v^{\pi^*_w}_1(s), \cdots, v^{\pi^*_w}_K(s)]^\top \in \mathbb{R}^K\) for all \(s\), then \(v^{\pi^*_w}(s)\) is the value function evaluated with the policy \(\pi^*_w\) in a given MOMDP. We have \[\nabla_{w} v^*_{w}(s) = v^{\pi^*_w}(s), ~ \forall s.\] For the case of \(L \geq 1\), the only difference is that \(\pi^*_{w}\) is changed to \[\pi^*_{u,w}(a | s) = \frac{ \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a) + \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v^{*}_{u,w}(s') \} ]}{\sum_{a'} \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a') + \sum_{k=1}^K w_k r^{(k)}(s,a') + \gamma \sum_{s'}T(s'|s,a') v^{*}_{u,w}(s') \} ]}\] where \(v^{*}_{u, w}\) is the fixed point of the operator \(\mathcal{T}_{u,w}\): \[\forall s, ~ [\mathcal{T}_{u,w} v](s) = \beta \log \sum_a \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a) + \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v(s') \} ]\] and the column size of \(r^{\pi^*_{u,w}}\) is \(L+K\), not \(K\). We denote \(v_c^{\pi^*_{u,w}}(s) \in \mathbb{R}^L\), \(v_r^{\pi^*_{u,w}}(s) \in \mathbb{R}^K\) as the value functions evaluated with the policy \(\pi^*_{u,w}\) for constrained reward \(c\) and unconstrained reward \(r\), respectively. Finally, we have \[\nabla_{u} v^*_{u, w}(s) = v_c^{\pi^*_{u,w}}(s), ~ \nabla_{w} v^*_{u, w}(s) = v_r^{\pi^*_{u,w}}(s), ~ \forall s.\] ◻

10 Proof of Smoothness↩︎

Proof. Let \(a = (u',w')\) and \(b = (u'', w'')\) in \(\mathbb{R}^{L+K}\) and \(a \neq b\). By the differentiability of \(\nabla v^*_{u,w}(s)\) for each \(s\) in Proposition 5, \(\nabla v^*_{u,w}(s)\) is continuous on \(\mathbb{R}^{L+K}\). Specifically, \(\nabla v^*_{u,w}(s)\) is continuous on \(\{a + t(b-a) | 0 \leq t \leq 1 \}\) and differentiable on \(\{a + t(b-a) | 0 < t < 1 \}\). Therefore, by the generalized mean value inequality, we have \[\label{eq:general95mean95ineq} \| \nabla v^*_{u,w}(s)|_{(u,w) = b} - \nabla v^*_{u,w}(s)|_{(u,w) = a} \|_2 \leq \sup_{t \in [0,1] }\| H[v^*_{u,w}(s)]|_{(u,w) = a + t(b-a)} \|_2 \| b - a \|_2.\tag{26}\] where \(H[v^*_{u,w}(s)]\) is the Hessian of \(v^*_{u,w}(s)\). Let \(\lambda_{\text{max}}( A )\) be the maximum eigenvalue of a real symmetric matrix \(A\). For each \(s_k ~ (1 \leq k \leq p)\), the eigenvalues of \(H[v^*_{u,w}(s_k)]\) are nonnegative because of its positive semidefiniteness in ?? in Proposition 5. Since trace operator is additive, we have \[\| H[v^*_{u,w}(s_k)] \|_2 = \lambda_{\text{max}}( H[v^*_{u,w}(s_k)] ) \leq \text{Tr}( H[v^*_{u,w}(s_k)] ) = \frac{1}{\beta} \sum_{l=1}^p [(I_p - \gamma T^{\pi^*_{u,w}})^{-1}]_{kl} \text{Tr}(B^{\pi^*_{u,w}}(s_l)).\] For each \(s\), we also have \[\text{Tr}( B^{\pi^*_{u,w}}(s)) = \sum_{k=1}^{L+K} \text{Var}(Q_k^{\pi^*_{u,w}}(s,a)) \leq \sum_{k=1}^{L+K} \mathbb{E}[ | Q_k^{\pi^*_{u,w}}(s,\cdot) |^2 ] \leq \sum_{k=1}^{L+K} \left( \frac{r^{(k)}_{\text{max}}}{1-\gamma} \right)^2\] where \(Q_k^{\pi^*_{u,w}}(s,a)\) is the \(k\)-th dimension of \(Q^{\pi^*_{u,w}}(s,a) \in \mathbb{R}^{K+L}\).

Since \((I_p - \gamma T^{\pi^*_{u,w}})^{-1} = \sum_{i=0}^\infty (\gamma T^{\pi^*_{u,w}})^i\) and each \((T^{\pi^*_{u,w}})^i\) is a probability transition matrix, \[\| H[v^*_{u,w}(s_k)] \|_2 \leq \frac{1}{\beta} \sum_{m=1}^{L+K} \left( \frac{r^{(m)}_{\text{max}}}{1-\gamma} \right)^2 \left( \sum_{i=0}^\infty \gamma^i \sum_{l=1}^p (T^{\pi^*_{u,w}})^i_{kl} \right) = \frac{1}{\beta (1-\gamma)} \sum_{m=1}^{L+K} \left( \frac{r^{(m)}_{\text{max}}}{1-\gamma} \right)^2.\] Here, the summations of \(\sum_{l=1}^p\) and \(\sum_{i=0}^\infty\) are interchangable due to the absolute summability. Finally, we have \(\| H[\mathcal{L}(u,w)]\|_2 \leq \sum_s \mu_0(s) \| H[v^*_{u,w}(s)] \|_2 \leq \frac{1}{\beta (1-\gamma)} \sum_{m=1}^{L+K} \left( \frac{r^{(m)}_{\text{max}}}{1-\gamma} \right)^2 =: \alpha\).

It should be noted that \(\| H[v^*_{u,w}(s_k)] \|_2\) is uniformly bounded regardless of \(s_k\) and \((u,w)\). Therefore, from 26 , \(\nabla \mathcal{L}(u,w)\) is \(\alpha\)-Lipschitz continuous in \(\| \cdot \|_2\). ◻

11 Proof of Twice-Differentiability↩︎

Proof. Here we also use the implicit function theorem and follow a similar logic in the proof of differentiability in Appendix 9. Let \(|\mathcal{S}| = p\). We show the theorem for the case of \(L=0\) to guarantee notational simplicity. For each \(1 \leq i \leq K\), we want to show that \(\frac{\partial v^*_w}{\partial w_i} := [\frac{\partial v^*_w(s_1)}{\partial w_i}, \cdots, \frac{\partial v^*_w(s_p)}{\partial w_i}]^\top \in \mathbb{R}^{p}\) is differentiable in \(w \in \mathbb{R}^K\). From the result in Appendix 9, we have \[\frac{\partial v^*_w}{\partial w_i} = v^{\pi^*_w}_i\] where \(v^{\pi^*_w}_i \in \mathbb{R}^p\) is the value function evaluated with the policy \(\pi^*_w\) in 24 with the \(i\)-th reward \(r^{(i)}\). Let \(r_i^{\pi^*_w}(s) = \mathbb{E}_{a \sim \pi^*_w(\cdot | s)}[ r^{(i)}(s, a)] \in \mathbb{R}\). From 25 , we have \[v^{\pi^*_w}_i = (I_p - \gamma T^{\pi^*_w})^{-1} r_i^{\pi^*_w}\] or equivalently, \[v^{\pi^*_w}_i = r_i^{\pi^*_w} + \gamma T^{\pi^*_w} v^{\pi^*_w}_i =: \mathcal{T}_w^* v^{\pi^*_w}_i.\]

We define \(F(w,v):=v-\mathcal{T}_w^* v\), \(F: \mathbb{R}^{K} \times \mathbb{R}^{p} \to \mathbb{R}^{p}\). Then \(F(w,v^{\pi^*_w}_i)=v^{\pi^*_w}_i-\mathcal{T}_w v^{\pi^*_w}_i = 0\). Here \(v^{\pi^*_w}_i\) is the unique fixed point of \(\mathcal{T}_w^*\) and is implicitly expressed w.r.t. \(w\), and we aim to analyze \(v^{\pi^*_w}_i\) using implicit function theorem.

First of all, \(F: \mathbb{R}^{K} \times \mathbb{R}^{p} \to \mathbb{R}^{p}\) is a continuously differentiable function. For each \(s\), \(F(w,v)(s)=v(s)- [\mathcal{T}_w^* v](s) = v(s) - [ r_i^{\pi^*_w}(s) + \gamma \sum_{s'} T^{\pi^*_w}(s'|s) v(s') ] = v(s) - \sum_a \pi^*_w(a|s)[ r^{(i)}(s,a) + \gamma \sum_{s'} T(s'|s,a) v(s') ]\). As seen in 24 , \(\pi^*_w\) contains \(v^*_w\) which is continuously differentiable in \(w\) (as a result of the proof in Appendix 9), and \(\pi^*_w\) is a composition of quotient, exponential, summation and linear functions of \(w\) and \(v^*_w\).

Now we fix \(w\) and check whether the Jacobian matrix \(\partial_v F(w,v)|_{v = v^{\pi^*_w}_i} \in \mathbb{R}^{p \times p}\) is invertible where \([\partial_v F(w,v)|_{v = v^{\pi^*_w}_i}]_{ij} = \frac{\partial F(w,v)(s_i)}{\partial v(s_j)}|_{v = v^{\pi^*_w}_i}\). We have \(\partial_v F(w,v) = I_p -\partial_v [\mathcal{T}_w^* v]\) where \(I_p\) is the \(p \times p\) identity matrix. Then \[\frac{\partial [\mathcal{T}_w^* v](s_i)}{\partial v(s_j)}|_{v = v^{\pi^*_w}_i} = \gamma \mathbb{E}_{a \sim \pi^*_w(\cdot | s_i)}[T(s_j | s_i, a)].\]

If we denote \(T(\cdot |s,a) := [T(s_1|s,a)\cdots T(s_p|s,a)]\), we have \[\partial_v F(w,v)|_{v = v^{\pi^*_w}_i} =I_p -\gamma\begin{bmatrix} \mathbb{E}_{a \sim \pi^*_w(\cdot | s_1)}[T(\cdot | s_1, a)] \\ \vdots\\ \mathbb{E}_{a \sim \pi^*_w(\cdot | s_p)}[T(\cdot | s_p, a)] \end{bmatrix} =: I_p -\gamma\begin{bmatrix} T^{\pi^*_w}(\cdot | s_1) \\ \vdots\\ T^{\pi^*_w}(\cdot | s_p) \end{bmatrix}\] where \(T^{\pi^*_w}(s_j | s_i) = \mathbb{E}_{a \sim \pi^*_w(\cdot | s_i)}[T(s_j | s_i, a)] =: [T^{\pi^*_{w}}]_{ij}\). Then \(I_p - \gamma T^{\pi^*_w}\) is invertible since \(T^{\pi^*_w}\) is a row-stochastic square matrix [59].

Therefore, \(\partial_v F(w,v)|_{v = v^{\pi^*_w}_i}\) is invertible. By implicit function theorem, there exists an open set \(U\subset\mathbb{R}^K\) containing \(w\) such that there exists a unique continuously differentiable function \(h:U\to\mathbb{R}^{p}\) such that \(h(w)=v^{\pi^*_w}_i\) and \(F(w',h(w'))=0\), i.e., \(h(w')=\mathcal{T}_{w'}^* h(w')\) for all \(w'\in U\). Since \(h(w')\) is the unique fixed point of \(\mathcal{T}_{w'}^*\), \(h(w') = v^{\pi^*_{w'}}_i, \forall w'\in U\). If we use the implicit function theorem for all \(w \in \mathbb{R}^K\), we can conclude that \(v = v^{\pi^*_w}_i\) is a unique continuously differentiable function in \(w \in \mathbb{R}^K\) that satisfies \(v = \mathcal{T}_w^* v\).

Now, for \(1 \leq j \leq K\), we aim to calculate \(\frac{\partial [\mathcal{T}_w^* v](s)}{\partial w_j}|_{v = v^{\pi^*_w}_i}\). For notational simplicity, let \(Q^*_w(s,a) := \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v^{*}_w(s')\). Then we express \(\pi^*_w\) as follows: \[\pi^*_w(a | s) = \frac{ \exp [ \frac{1}{\beta} \{ Q^*_w(s,a) \} ]}{\sum_{a'} \exp [ \frac{1}{\beta} \{ Q^*_w(s,a') \} ]}.\] We also have \[\frac{\partial Q^*_w(s,a)}{\partial w_j} = r^{(j)}(s,a) + \gamma \sum_{s'}T(s'|s,a) \frac{\partial v^{*}_w(s')}{\partial w_j} = r^{(j)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v^{\pi^*_w}_j(s') := Q_j^{\pi^*_w}(s,a).\]

In other words, we denote \(Q_j^{\pi^*_w}\) as the action-value function evaluated with \(\pi^*_w\) for a scalar reward function \(r^{(j)}\). Then \[\frac{\partial [\mathcal{T}_w^* v](s)}{\partial w_j}|_{v = v^{\pi^*_w}_i} = \sum_a Q_i^{\pi^*_w}(s,a) \frac{\partial \pi^*_w(a | s) }{\partial w_j}\] which is equivalent to \[\frac{\partial [\mathcal{T}_w^* v](s)}{\partial w_j}|_{v = v^{\pi^*_w}_i} = \frac{1}{\beta} \sum_a Q_i^{\pi^*_w}(s,a) \bigg[ \pi^*_w(a | s) Q_j^{\pi^*_w}(s,a) - \pi^*_w(a | s) \sum_{a'} \{ \pi^*_w(a' | s) Q_j^{\pi^*_w}(s,a') \} \bigg]\] and we have \[\frac{\partial [\mathcal{T}_w^* v](s)}{\partial w_j}|_{v = v^{\pi^*_w}_i} = \frac{1}{\beta} \bigg[ \mathbb{E}_{a \sim \pi^*_w(\cdot | s)} [ Q_i^{\pi^*_w}(s,a) Q_j^{\pi^*_w}(s,a) ] - \mathbb{E}_{a \sim \pi^*_w(\cdot | s)} [ Q_i^{\pi^*_w}(s,a) ] \mathbb{E}_{a \sim \pi^*_w(\cdot | s)} [ Q_j^{\pi^*_w}(s,a) ] \} \bigg].\]

By implicit function theorem, we have \[\begin{bmatrix} \nabla_{w} \frac{\partial v^*_{w}(s_1)}{\partial w_i}^\top \\ \vdots\\ \nabla_{w} \frac{\partial v^*_{w}(s_p)}{\partial w_i}^\top \end{bmatrix} = - [\partial_v F(w,v)|_{v = v^{\pi^*_w}_i}]^{-1} \partial_w F(w,v)|_{v = v^{\pi^*_w}_i} = \frac{1}{\beta}(I_p - \gamma T^{\pi^*_w})^{-1} E_i^{\pi^*_w}\] where \(E_i^{\pi^*_w}\) is a \(p \times K\) matrix where for each row corresponding to \(s\), the \(j\)-th element is \(\mathbb{E}_{a \sim \pi^*_w(\cdot | s)} [ Q_i^{\pi^*_w}(s,a) Q_j^{\pi^*_w}(s,a) ] - \mathbb{E}_{a \sim \pi^*_w(\cdot | s)} [ Q_i^{\pi^*_w}(s,a) ] \mathbb{E}_{a \sim \pi^*_w(\cdot | s)} [ Q_j^{\pi^*_w}(s,a) ] \}\). This formulation holds for each \(1 \leq i \leq K\).

Therefore, we construct a \(p \times K \times K\) tensor, say \(B^{\pi^*_w}\), by stacking \(\{E_i^{\pi^*_w}\}_{i}\) along the new (third) dimension. Then along the first dimension of size \(p\), for each \(s\), let \(B^{\pi^*_w}(s) \in \mathbb{R}^{K \times K}\) be the corresponding slice of \(B\). Let \(Q^{\pi^*_w}(s,a) = [Q_1^{\pi^*_w}(s,a), \cdots, Q_K^{\pi^*_w}(s,a)]^\top \in \mathbb{R}^K\) be the action-value function evaluated with \(\pi^*_w\) for vector reward \(r\). Then we have

\[B^{\pi^*_w}(s) = \mathbb{E}_{a \sim \pi^*_w(\cdot|s)} \bigg[ ( Q^{\pi^*_w}(s,a) - \mathbb{E}_{a' \sim \pi^*_w(\cdot|s)} [Q^{\pi^*_w}(s,a')] ) ( Q^{\pi^*_w}(s,a) - \mathbb{E}_{a' \sim \pi^*_w(\cdot|s)} [Q^{\pi^*_w}(s,a')] )^\top \bigg]\]

which is the covariance matrix of \(Q^{\pi^*_w}(s,\cdot)\) over the probability distribution \(\pi^*_w(\cdot|s)\). Let \(s_k\) correspond to the \(k\)-th row of \(T^{\pi^*_w} ~ (1 \leq k \leq p)\). Then we have the following Hessian formulation for \(s_k\): \[H[v^*_{w}(s_k)] = \frac{1}{\beta} \sum_{l=1}^p [(I_p - \gamma T^{\pi^*_w})^{-1}]_{kl} B^{\pi^*_w}(s_l).\]

For the case of \(L \geq 1\), the only difference is that \(\pi^*_{w}\) is changed to \[\pi^*_{u,w}(a | s) = \frac{ \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a) + \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v^{*}_{u,w}(s') \} ]}{\sum_{a'} \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a') + \sum_{k=1}^K w_k r^{(k)}(s,a') + \gamma \sum_{s'}T(s'|s,a') v^{*}_{u,w}(s') \} ]}\] where \(v^{*}_{u, w}\) is the fixed point of the operator \(\mathcal{T}_{u,w}\): \[\forall s, ~ [\mathcal{T}_{u,w} v](s) = \beta \log \sum_a \exp [ \frac{1}{\beta} \{ \sum_{l=1}^L u_l c^{(l)}(s,a) + \sum_{k=1}^K w_k r^{(k)}(s,a) + \gamma \sum_{s'}T(s'|s,a) v(s') \} ]\] and the size of \(B^{\pi^*_{u,w}}(s)\) is \((L+K) \times (L+K)\), not \(K \times K\), defined by \(Q^{\pi^*_{u,w}}(s,a) \in \mathbb{R}^{L+K}\) which is the action-value function evaluated with \(\pi^*_{u,w}\) for the concatenated vector function of constrained reward \(c\) and unconstrained reward \(r\). Finally, we have \[H[v^*_{u,w}(s_k)] = \frac{1}{\beta} \sum_{l=1}^p [(I_p - \gamma T^{\pi^*_{u,w}})^{-1}]_{kl} B^{\pi^*_{u,w}}(s_l).\] ◻

12 Convergence Analysis↩︎

12.1 Assumption for Action-value Nondegeneracy↩︎

Assumption There exists at least one state \(s \in \mathcal{S}\) such that \(\rho^{\pi^*_{u,w}}(s) > 0\) and the centered action-value vectors \(\bigl\{\,Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot\mid s)}[\,Q^{\pi^*_{u,w}}(s,a')\,]\;:\;a\in\mathcal{A}\bigr\}\) span \(\mathbb{R}^{K+L}\).

This condition fails only in degenerate multi-objective settings when for every state \(s\in\mathcal{S}\) with \(\rho^{\pi^*_{u,w}}(s) > 0\), the set \(\bigl\{\,Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot\mid s)}[\,Q^{\pi^*_{u,w}}(s,a')\,]: a\in \mathcal{A}\bigr\}\) lies entirely within an affine subspace of dimension less than \(K+L\) (e.g., the size of an action set is smaller than the number of objectives).

Then \(B^{\pi^*_{u,w}}(s) = \mathbb{E}_{a \sim \pi^*_{u,w}(\cdot|s)} \bigg[ ( Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot|s)} [Q^{\pi^*_{u,w}}(s,a')] ) ( Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot|s)} [Q^{\pi^*_{u,w}}(s,a')] )^\top \bigg] \in \mathbb{R}^{(L+K) \times (L+K)}\) is positive definite. This is because (i) \(\pi^*_{u,w}(a|s) > 0\) for all \(a\) from 7 (which has this favorable property that facilitates analysis), and (ii) for any \(y \in \mathbb{R}^{K+L} ~ \text{with} ~ y \neq \mathbf{0}\), \(y^\top B^{\pi^*_{u,w}}(s) y = \sum_{a} \pi^*_{u,w}(a|s) \bigg( y^\top ( Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot|s)} [Q^{\pi^*_{u,w}}(s,a')] ) \bigg)^2 > 0\) as at least one \(a\) should satisfy \(y^\top ( Q^{\pi^*_{u,w}}(s,a) - \mathbb{E}_{a' \sim \pi^*_{u,w}(\cdot|s)} [Q^{\pi^*_{u,w}}(s,a')] ) \neq 0\).

By Proposition 5, we have the Hessian of \(\mathcal{L}(u,w)\) as \(H[\mathcal{L}(u,w)] = \frac{1}{\beta} \sum_{l=1}^p [ \mu_0^\top(I_p - \gamma T^{\pi^*_{u,w}})^{-1}]_l B^{\pi^*_{u,w}}(s_l) = \frac{1}{\beta} \sum_s \rho^{\pi^*_{u,w}}(s) B^{\pi^*_{u,w}}(s)\) where \(p = |\mathcal{S}|\) and \(\rho^{\pi^*_{u,w}}(s) = \sum_{t=0}^\infty \gamma^t \text{Pr}(s_t=s | \pi^*_{u,w}, T, \mu_0)\). Therefore, \(H[\mathcal{L}(u,w)]\) is positive definite under the assumption.

12.2 Existence of \(\lambda\)↩︎

Let \(R = \{ \rho \in \mathbb{R}^{|\mathcal{S}| |\mathcal{A}| }_+ | \sum_{a'} \rho(s',a') = \mu_0(s') + \gamma \sum_{(s,a)} T(s' | s,a) \rho(s,a) ,~\forall s' \}\) be the set of occupancy measures, which is convex and closed. The set \(R\) is also bounded since \(\rho(s,a) \leq \frac{1}{1-\gamma} ~ \forall (s,a)\) for \(\gamma \in [0,1)\). Under Slater condition with strict feasibility, there exists an occupancy measure \(\bar{\rho} \in R\) and positive constants of \(\{ \nu_l \}_{l=1}^L\) such that \[\sum_{(s,a)} c^{(l)}(s,a) \bar{\rho}(s,a) = J_{K+l}(\pi^{\bar{\rho}}) \geq C^{(l)} + \nu_l, ~~ l =1, \cdots, L.\] Although \(\sum_{(s,a)} c^{(l)}(s,a)\rho(s,a)\) is linear in \(\rho\), so that satisfying equality already implies Slater condition, assuming strict feasibility is a mild condition and is standard in prior constrained MDP literature [29].

Define the Lagrangian of the primal problem as: \[\tilde{L}(\rho, u) = \min_{1 \leq k \leq K} \bigg( \sum_{(s,a)} r^{(k)}(s,a) \rho(s,a) \bigg) + \beta \sum_{s} \mathcal{H}_\rho(s) \rho(s) + \sum_{l=1}^L u_l ( \sum_{(s,a)} c^{(l)}(s,a) \rho(s,a) - C^{(l)} )\] with \(\rho \in R\) and \(u \in \mathbb{R}^L_+\). Let the Lagrangian dual be \(D(u) = \max_{\rho \in R} \tilde{L}(\rho, u)\). Let \(P^* = \min_{1 \leq k \leq K} \bigg( \sum_{(s,a)} r^{(k)}(s,a) \rho^*(s,a) \bigg) + \beta \sum_{s} \mathcal{H}_{\rho^*}(s) \rho^*(s)\) denote the optimal value of the primal problem, where \(\rho^* \in R\) is an optimal solution that satisfies the \(L\) constraints.

By strong duality, we have \(P^* = \min_{u \in \mathbb{R}^L_+} D(u) = \tilde{L}(\rho^*, u^*)\) where \(u^* \in \mathbb{R}^L_+\) is an optimal dual variable corresponding to the constraints: \(\sum_{(s,a)} c^{(l)}(s,a) \rho(s,a) \geq C^{(l)}, ~ l =1, \cdots, L\). Also, \[P^* = \max_{\rho \in R} \tilde{L}(\rho, u^*) \geq \tilde{L}(\bar{\rho}, u^*) \geq \min_{1 \leq k \leq K} \bigg( \sum_{(s,a)} r^{(k)}(s,a) \bar{\rho}(s,a) \bigg) + \beta \sum_{s} \mathcal{H}_{\bar{\rho}}(s) \bar{\rho}(s) + \sum_{l=1}^L u_l^* \nu_l\] where the second inequality follows from the definition of \(\{ \nu_l \}_{l=1}^L\). Thus, \[0 \leq u_l^* \leq \frac{P^* - (\min_{1 \leq k \leq K} J_k(\pi^{\bar{\rho}}) + \beta \sum_{s} \mathcal{H}_{\bar{\rho}}(s) \bar{\rho}(s))}{\nu_l} := \bar{u}_l, \forall l.\]

Therefore, it suffices to restrict attention to the compact set \(u \in \Pi_{l=1}^L [0, \bar{u}_l] =: \mathcal{C}\) instead of \(\mathbb{R}^L_+\) when searching for an optimal dual variable corresponding to the constraints \(\sum_{(s,a)} c^{(l)}(s,a) \rho(s,a) \geq C^{(l)}, ~ l =1, \cdots, L\).

Next, the Hessian is explicitly written as follows: \[H[\mathcal{L}(u,w)] = \frac{1}{\beta} \sum_{l=1}^p [ \mu_0^\top (I - \gamma T^{\pi^*_{u,w}})^{-1} ]_{l} B^{\pi^*_{u,w}}(s_l).\]

From the differentiability of \(v^*_{u,w}\) established in Theorem 3.3, \(v^*_{u,w}\) is continuous in \((u,w)\). By their definitions, this continuity extends to \(Q^*_{u,w}\), \(\pi^*_{u,w}\), each entry of \(B^{\pi^*_{u,w}}\) and \(H[\mathcal{L}(u,w)]\). For two real symmetric matrices \(A\) and \(B\), the minimum eigenvalue \(\lambda_{\text{min}}(\cdot)\) satisfies: \[|\lambda_{\text{min}}(A) - \lambda_{\text{min}}(B)| \leq \| A - B \|_2 \leq \| A - B \|_F,\] where \(\| \cdot \|_F\) is the Frobenius norm.

Therefore, \(\lambda_{\text{min}}(H[\mathcal{L}(u,w)])\) is continuous in \((u,w)\) over the compact set \(\mathcal{C} \times \Delta^K\). By the extreme value theorem, there exists \(\lambda = \min_{(u,w) \in \mathcal{C} \times \Delta^K} \lambda_{\text{min}}(H[\mathcal{L}(u,w)]) > 0\) (with \(\lambda \leq \alpha\)).

12.3 Proof of Convergence Analysis↩︎

Let \(\lambda_{\text{min}}( A )\) be the minimum eigenvalue of a real symmetric matrix \(A\). For simplicity, we denote \(\lambda := \lambda_{\text{min}}( H[\mathcal{L}(u,w)] )\). Then \(0 < \lambda \leq \alpha\) [35] and \(\mathcal{L}(u,w)\) is \(\lambda\)-strongly convex.

Theorem 6 Let \((u^*,w^*)\) denote an optimal solution to ?? . For each outer-loop index \(m \geq 1\) in Algorithm 1, let \(Q^*_{u^m, w^m}\) denote the fixed point of 10 with \([u;w] = [u^m;w^m]\), and let \(Q^m\) denote the Q-function after completing the \(m\)-th inner-loop update. For each \(m\), assume \(\| Q^m - Q^*_{u^m, w^m} \|_\infty < \epsilon\) for some \(\epsilon > 0\). Then for \(m \geq 1\), \[\| [u^m;w^m] - [u^*;w^*] \|_{2} \leq (1-\frac{\lambda}{\alpha})^m \| [u^0;w^0] - [u^*;w^*] \|_{2} + \frac{\sqrt{|\mathcal{S}|}}{\lambda} \sqrt{\sum_{i=1}^{K+L} \{ r^{(i)}_{\text{max}} \}^2} \frac{1+\gamma }{(1-\gamma)^2} \epsilon.\]

Proof. By the definition in 7 , we have the optimal policy \(\pi^*_{u^m, w^m}(a | s) = \frac{ \exp ( \frac{1}{\beta} Q^*_{u^m, w^m}(s,a) ) }{\sum_{a'} \exp ( \frac{1}{\beta} Q^*_{u^m, w^m}(s,a') ) }\) when \((u,w) = (u^m,w^m)\). According to Theorem 3, we have \(\nabla_{(u,w)}\mathcal{L}(u^m,v^m) = [\sum_s \mu_0(s) v_c^{\pi^*_{u^m,v^m}}(s) - [C^{(1)},\cdots,C^{(L)}]^\top ; \sum_s \mu_0(s) v_r^{\pi^*_{u^m,v^m}}(s) ] \in \mathbb{R}^{L+K}\).

We also have \(\tilde{\nabla}_{(u,w)}\mathcal{L}(u^m,v^m) := [\sum_s \mu_0(s) v_c^{\pi^m}(s) - [C^{(1)},\cdots,C^{(L)}]^\top ; \sum_s \mu_0(s) v_r^{\pi^m}(s) ] \in \mathbb{R}^{L+K}\), an estimated gradient of \(\nabla_{(u,w)}\mathcal{L}(u^m,w^m)\) using \(\pi^m\) where \(\pi^m(a | s) = \frac{ \exp ( \frac{1}{\beta} Q^m(s,a) ) }{\sum_{a'} \exp ( \frac{1}{\beta} Q^m(s,a') ) }\).

Let \(e_m := \tilde{\nabla}_{(u,w)}\mathcal{L}(u^m,v^m) - \nabla_{(u,w)}\mathcal{L}(u^m,w^m)\). For each \(s\), let \(v_{r, i}^{\pi}(s) ~ (1 \leq i \leq K)\) and \(v_{c, j}^{\pi}(s) ~ (1 \leq j \leq L)\) denote the elements of the \(i\)-th dimension of \(v_{r}^{\pi}(s) \in \mathbb{R}^K\) and the \(j\)-th dimension of \(v_{c}^{\pi}(s) \in \mathbb{R}^L\), respectively. Then we have \[\begin{align} \label{eq:error95eq1} \| e_m \|^2_2 &= \| [\sum_s \mu_0(s) (v_c^{\pi^m}(s) - v_c^{\pi^*_{u^m,v^m}}(s)) ; \sum_s \mu_0(s) (v_r^{\pi^m}(s) - v_r^{\pi^*_{u^m,v^m}}(s)) ] \|^2_2 \nonumber \\ &= \sum_{i=1}^K \left( \sum_s \mu_0(s) (v_{r, i}^{\pi^m}(s) - v_{r, i}^{\pi^*_{u^m, w^m}}(s)) \right)^2 + \sum_{j=1}^L \left( \sum_s \mu_0(s) (v_{c, j}^{\pi^m}(s) - v_{c, j}^{\pi^*_{u^m, w^m}}(s)) \right)^2 \nonumber \\ &\leq \| \mu_0 \|^2_2 \sum_s \bigg[ \sum_{i=1}^K (v_{r, i}^{\pi^m}(s) - v_{r, i}^{\pi^*_{u^m, w^m}}(s))^2 + \sum_{j=1}^L (v_{c, j}^{\pi^m}(s) - v_{c, j}^{\pi^*_{u^m, w^m}}(s))^2 \bigg] \end{align}\tag{27}\] where \(\| \mu_0 \|^2_2 = \sum_s (\mu_0(s))^2\) and the inequality holds by Cauchy-Schwarz.

Since both \(\pi^m\) and \(\pi^*_{u^m, w^m}\) use softmax parameterization with \(Q^m\) and \(Q^*_{u^m, w^m}\), respectively, we have \[\label{eq:softmax951} \forall s, ~ |v_{r, i}^{\pi^m}(s) - v_{r, i}^{\pi^*_{u^m, w^m}}(s)| \leq \frac{(1+\gamma)r^{(i)}_{\text{max}}}{(1-\gamma)^2} \| Q^m - Q^*_{u^m, w^m} \|_\infty ~ (1 \leq i \leq K)\tag{28}\] and \[\label{eq:softmax952} \forall s, ~ |v_{c, j}^{\pi^m}(s) - v_{c, j}^{\pi^*_{u^m, w^m}}(s)| \leq \frac{(1+\gamma)r^{(K+j)}_{\text{max}}}{(1-\gamma)^2} \| Q^m - Q^*_{u^m, w^m} \|_\infty ~ (1 \leq j \leq L)\tag{29}\] according to the property of equation (261) in [62]. Combining 28 , 29 , and \(\| \mu_0 \|_2 \leq 1\) with 27 gives \[\begin{align} \label{eq:error95eq95final} \| e_m \|_2 &\leq \sqrt{|\mathcal{S}|} \sqrt{\sum_{i=1}^{K+L} \{ r^{(i)}_{\text{max}} \}^2} \frac{1+\gamma }{(1-\gamma)^2} \| Q^m - Q^*_{u^m, w^m} \|_\infty \nonumber \\ &< \sqrt{|\mathcal{S}|} \sqrt{\sum_{i=1}^{K+L} \{ r^{(i)}_{\text{max}} \}^2} \frac{1+\gamma }{(1-\gamma)^2} \epsilon. \end{align}\tag{30}\]

Next, we view the projected gradient descent for each outer loop as a proximal gradient descent. We reformulate the optimization in ?? of \[\min_{u \in \mathbb{R}_{+}^L, w \in \Delta^K} \mathcal{L}(u,w)\] as follows: \[\label{eq:proximal95conversion} \min_{ (u,w) \in \mathbb{R}^{L+K}} \mathcal{L}(u,w) + I_{\mathbb{R}_{+}^L \times \Delta^K}(u,w)\tag{31}\] where \(I_{\mathbb{R}_{+}^L \times \Delta^K}(u,w)\) is the indicator function with its value 0 if \((u,w) \in \mathbb{R}_{+}^L \times \Delta^K\) and \(+\infty\) otherwise. \(I_{\mathbb{R}_{+}^L \times \Delta^K}\) is convex because its epigraph \(\{(u,w,t_e) | t_e \geq 0, (u,w) \in \mathbb{R}_{+}^L \times \Delta^K \}\) is convex. We note that according to Theorem 4, the smoothness of \(\mathcal{L}(u,w)\) is satisfied on \(\mathbb{R}^{L+K}\), which makes 31 valid. We also note that we computed the smoothness coefficient \(\alpha = \frac{1}{\beta (1-\gamma)} \sum_{i=1}^{K+L} \left( \frac{r^{(i)}_{\text{max}}}{1-\gamma} \right)^2\) of \(\mathcal{L}\) in Appendix 10.

Applying the error bound in 30 to the analysis of inexact proximal gradient method [63], we have \[\begin{align} \| [u^m;w^m] - [u^*;w^*] \|_{2} &\leq (1-\frac{\lambda}{\alpha})^m \| [u^0;w^0] - [u^*;w^*] \|_{2} + \frac{1}{\alpha} \sum_{i=1}^m (1-\frac{\lambda}{\alpha})^{m-i} \| e_i \|_2 \nonumber \\ &\leq (1-\frac{\lambda}{\alpha})^m \| [u^0;w^0] - [u^*;w^*] \|_{2} + \frac{\sqrt{|\mathcal{S}|}}{\lambda} \sqrt{\sum_{i=1}^{K+L} \{ r^{(i)}_{\text{max}} \}^2} \frac{1+\gamma }{(1-\gamma)^2} \epsilon. \end{align}\]

This is achieved because we use the convex optimization method from [36] for projection onto the simplex \(\Delta^K\), and apply non-negativity clipping for projection onto \(\mathbb{R}_+^L\), both of them induce zero error in each phase of proximal objective update as it is fully deterministic and avoids randomized procedures.

It remains to check whether \(I_{\mathbb{R}_{+}^L \times \Delta^K}\) in 31 is a lower semi-continuous proper convex function [63]. \(I_{\mathbb{R}_{+}^L \times \Delta^K}\) is lower semi-continuous because \(\mathbb{R}_{+}^L \times \Delta^K\) is closed, and it is also proper convex since \(I_{\mathbb{R}_{+}^L \times \Delta^K}\) never attains \(-\infty\) and \(\mathbb{R}_{+}^L \times \Delta^K\) is non-empty. ◻

12.4 Convergence Analysis for Degenerate Case↩︎

Theorem 7. Let \((u^*,w^*)\) denote an optimal solution to ?? . For each outer-loop index \(m \geq 1\) in Algorithm 1, let \(Q^*_{u^m, w^m}\) denote the fixed point of 10 with \([u;w] = [u^m;w^m]\), and let \(Q^m\) denote the Q-function after completing the \(m\)-th inner-loop update. For each \(m\), assume \(\| Q^m - Q^*_{u^m, w^m} \|_\infty < \epsilon_m\) for some \(\epsilon_m > 0\). Then for \(m \geq 1\), \[\mathcal{L}( \frac{1}{m} \sum_{i=1}^m (u^i, w^i) ) - \mathcal{L}(u^*,w^*) \leq \frac{\alpha}{2m} ( \| [u^0;w^0] - [u^*;w^*] \|_{2} ~ + \frac{2 M}{\alpha} \sum_{i=1}^m \epsilon_i )^2\] where \(M = \sqrt{|\mathcal{S}|} \sqrt{\sum_{j=1}^{K+L} \{ r^{(j)}_{\text{max}} \}^2} \frac{1+\gamma }{(1-\gamma)^2}\).

Proof. Using an analysis of inexact proximal gradient method [63] using the same logic in the proof of Theorem 6 (Appendix 12.3), we have

\[\mathcal{L}( \frac{1}{m} \sum_{i=1}^m (u^i, w^i) ) - \mathcal{L}(u^*,w^*) \leq \frac{\alpha}{2m} ( \| [u^0;w^0] - [u^*;w^*] \|_{2} ~ + \frac{2}{\alpha} \sum_{i=1}^m \| e_i \|_2 )^2\] where \(e_i := \tilde{\nabla}_{(u,w)}\mathcal{L}(u^i,w^i) - \nabla_{(u,w)}\mathcal{L}(u^i,w^i)\) is the \(i\)-th gradient error and \[\| e_i \|_2 < \sqrt{|\mathcal{S}|} \sqrt{\sum_{j=1}^{K+L} \{ r^{(j)}_{\text{max}} \}^2} \frac{1+\gamma }{(1-\gamma)^2} \epsilon_i = M \epsilon_i\] from 30 . ◻

We note that the error of \(\mathcal{L}( \frac{1}{m} \sum_{i=1}^m (u^i, w^i) ) - \mathcal{L}(u^*,w^*)\) decreases at rate \(O(\frac{1}{m})\) when \(\{\epsilon_i\}_{i=1}^\infty\) is summable (e.g., \(\epsilon_m = O(\frac{1}{m^{1+\delta}})\) with \(\delta > 0\)).

13 Experimental Details: Tabular Settings↩︎

When generating structured MOMDPs randomly, we first verify whether the generated instances are feasible. To do this, We first consider the following unregularized convex optimization: \[\label{eq:optimal95obj} \max_{\rho \geq 0} \min_{1 \leq k \leq K} \bigg( \sum_{(s,a)} r^{(k)}(s,a) \rho(s,a) \bigg)\tag{32}\] \[\label{eq:optimal95flow} \sum_{a'} \rho(s',a') = \mu_0(s') + \gamma \sum_{(s,a)} T(s' | s,a) \rho(s,a) ,~\forall s'\tag{33}\] \[\label{eq:optimal95const95c} \sum_{(s,a)} c^{(l)}(s,a) \rho(s,a) \geq C^{(l)}, ~~ l =1, \cdots, L\tag{34}\] which is equivalently expressed as the following LP by using additional scalar variable \(\tilde{c} \in \mathbb{R}\): \[\label{eq:eqiv95optimal95obj} \max_{\rho \geq 0, \tilde{c}} \tilde{c}\tag{35}\] \[\label{eq:eqiv95optimal95flow} \sum_{a'} \rho(s',a') = \mu_0(s') + \gamma \sum_{(s,a)} T(s' | s,a) \rho(s,a) ,~\forall s'\tag{36}\] \[\label{eq:eqiv95optimal95const95r} \sum_{(s,a)} r^{(k)}(s,a) \rho(s,a) \geq \tilde{c}, ~~ k =1, \cdots, K,\tag{37}\] \[\label{eq:eqiv95optimal95const95c} \sum_{(s,a)} c^{(l)}(s,a) \rho(s,a) \geq C^{(l)}, ~~ l =1, \cdots, L.\tag{38}\]

We want to generate \(\mu_0, T, r\), and \(c\) in structured MOMDPs to satisfy feasibility and Slater condition by solving the following LP using the pywraplp function from the OR-Tools library:

\[\label{eq:feasible95obj} \max_{\rho \geq \epsilon_{\text{low}}} ~ 0\tag{39}\] \[\label{eq:feasible95flow} \sum_{a'} \rho(s',a') = \mu_0(s') + \gamma \sum_{(s,a)} T(s' | s,a) \rho(s,a) ,~\forall s'\tag{40}\] \[\label{eq:feasible95const95r} \sum_{(s,a)} r^{(k)}(s,a) \rho(s,a) \geq \tilde{c} + \epsilon_{\text{low}}, ~~ k =1, \cdots, K,\tag{41}\] \[\label{eq:feasible95const95c} \sum_{(s,a)} c^{(l)}(s,a) \rho(s,a) \geq C^{(l)} + \epsilon_{\text{low}}, ~~ l =1, \cdots, L\tag{42}\]

where \(\epsilon_{\text{low}}\) is used to guarantee the strict feasibility for Slater condition, and we set \(\epsilon_{\text{low}}=10^{-4}\). If the LP solver does not find a feasible solution, we regenerate the constrained MOMDP until a feasible instance is found. Once any feasible solution is found, we solve the LP of 35 , 36 , 37 , and 38 by using LP solver to acquire the optimal max-min value \(\tilde{c}^*\).

Each algorithm is updated iteratively until the maximum change in the \(Q\)-function between successive iterations falls below \(10^{-4}\). We use the following settings: \(\gamma=0.8\), \(l_{\text{w}}=0.001\), and \(\text{ITER}=3000\). For each algorithm, we independently disable the learning of \(u\), of \(w\), and of both \((u,w)\), initializing \(u\) to the zero vector and \(w=[1/K, \cdots, 1/K] \in \Delta^K\). Each algorithm is evaluated using three random seeds for each constrained MOMDP setting, resulting in six runs when averaged across the MOMDP classes. All experiments are conducted on an Intel Core i9-10900X CPU @ 3.70GHz.

14 Experimental Details: Applications↩︎

14.1 Implementation of Our Algorithm for Applications↩︎

We now leverage the usage of neural network for our algorithm. If we differentiate the both side of \(v^{*}_{u, w}(s) = [\mathcal{T}_{u,w} v^{*}_{u, w}](s)\) w.r.t. \(u\) and \(w\) for all \(s\), then we have the following formula: \[\label{eq:grad95w95v95star} \forall s, ~ \nabla_w v^{*}_{u, w}(s) = \sum_a \pi^*_{u,w}(a | s) \bigg( r(s,a) + \gamma \sum_{s'}T(s'|s,a) \nabla_w v^{*}_{u, w}(s') \bigg).\tag{43}\]

\[\label{eq:grad95u95v95star} \forall s, ~ \nabla_u v^{*}_{u, w}(s) = \sum_a \pi^*_{u,w}(a | s) \bigg( c(s,a) + \gamma \sum_{s'}T(s'|s,a) \nabla_u v^{*}_{u, w}(s') \bigg).\tag{44}\]

Here, \(\pi^*_{u,w}(a | s)\) is defined as in 7 . To ensure stable gradient estimation in continuous state spaces, we parameterize a gradient network to estimate \(\nabla_u v^{*}_{u, w}(s)\) and \(\nabla_w v^{*}_{u, w}(s)\). For continuous action environments, we also employ an actor network \(\pi_\theta\) and implement Algorithm 2. To further stabilize the estimation of the gradient, we add an additional linear layer after the penultimate layer of the actor network \(\pi_\theta\), and use its \((L+K)\)-dimensional output as the gradient network \(g_\theta(s)\). We use the notation \(g_\theta\) to indicate that the actor network and the gradient network share parameters and jointly update their lower-layer weights.

Figure 2: Proposed Constrained Max-min Algorithm for Continuous Action
Figure 3: Critic Update
Figure 4: Actor Update

14.2 Unconstrained Max-min MORL Algorithm↩︎

Figure 5: Max-min GS for Continuous Action (Our modification from [12])

14.3 Building Thermal Control↩︎

Table 7: Hyperparameters for algorithms in building thermal control environment
Parameter Value
Shared
optimizer Adam (Kingma & Ba, 2015)
discount (\(\gamma\)) 0.99
target smoothing ratio (\(\tau\)) 0.001
reward dimension 3
max episode step 480
replay buffer size \(8\times 10^5\)
hidden units per layer 64
minibatch size 32
activation function ReLU
entropy coefficient 0.05
weight learning rate \(1 \times 10^{-6}\)
weight scheduling \(1/\sqrt{t}\)
Constrained Max-min MORL (Ours)
constraint type maximize
constraint dimension 1
constraint epsilon 1.0
constraint threshold \(-180\)
main learning rate \(7.5 \times 10^{-4}\)
gradient steps for critic update 3
gradient estimation learning rate \(5 \times 10^{-5}\)
gradient estimation steps 1
gradient target smoothing ratio 0.001
Unconstrained Max-min MORL (Gaussian)
main learning rate \(7.5 \times 10^{-4}\)
perturbation \(q\) learning rate 0.073
perturbation gradient steps 1
gradient steps for critic update 3
perturbation \(q\)-copies 10
perturbation noise std-dev 0.01
Unconstrained Max-min MORL (ARAM)
main learning rate \(7.5 \times 10^{-4}\)
CI coefficient \(\eta\) 0.2
MD coefficient \(\lambda\) 0.03
Max-average SAC with a Lagrangian Relaxation
constraint type minimize
initial lambda 1.0
main learning rate (actor/critic) \(3 \times 10^{-4}\)
constraint threshold 180
entropy coefficient 0.05
lambda learning rate 0.005
Unconstrained Max-average SAC
main learning rate (actor/critic) \(3 \times 10^{-4}\)

We use a building thermal control environment from SustainGym [54], which simulates a commercial building with \(N_{\text{zone}}=23\) thermal zones. We group these zones into three sections (bottom, middle, and top) and define a comfort objective for each section. Accordingly, in our experiments we set \(K=3\) and \(L=1\), i.e., three reward objectives and a single cost constraint.

At each timestep \(t\), the agent observes a normalized continuous state \(s_t \in \mathbb{R}^{29}\). The underlying (unnormalized) state is constructed by concatenating: (i) the temperatures of all 23 zones, (ii) outdoor temperature, (iii) global horizontal irradiance (GHI), (iv) ground temperature, (v) occupancy power, (vi) carbon intensity, and (vii) electricity price. The environment clips each component to predefined bounds and linearly rescales it to \([0,1]\) before returning \(s_t\).

The action is a continuous vector \(a_t \in \mathbb{R}^{23}\) with each component bounded in \([-1,1]\), representing zone-level heating/cooling power commands (cooling is negative and heating is positive). The reward is a \(K\)-dimensional vector \(\mathbf{r}_t \in \mathbb{R}^{3}\) that measures comfort for three zone groups. The 23 zones are partitioned into bottom (zones 1–9), middle (zones 10–16), and top (zones 17–23), and each reward entry increases when the temperatures in the corresponding group stay closer to the target temperatures. We define a single scalar energy cost signal \(c_t\) from the aggregated magnitude of the zone-level power commands, specifically \(c_t = \|a_t\|_1 / s\), where we tune the scaling factor \(s\) (set to \(s=5\)) to match the reward scale, and use \(c_t\) as the cost for the energy constraint (\(L=1\)).

Each episode consists of 480 timesteps. The total training budget is \(8\times 10^{5}\) timesteps, with evaluations conducted once every two training episodes. During each evaluation, one episode is run and the cumulative discounted sums of the \((L+K)\)-dimensional vector are computed. These experiments were conducted using an NVIDIA RTX 4000 SFF Ada GPU (20GB) across five random seeds.

14.4 Effect of \(\beta\) and Wall-Clock Time Comparison↩︎

Table 8: Performance comparison with different values of \(\beta\) in Building environment (\(C_{th}=180\)).
\(\beta\) Cost sum (\(C_{th} = 180\)) Minimum return (\(\uparrow\))
0.1 152.6 625.6
0.05 178.7 639.8
0.01 269.8 563.6
Table 9: Wall-clock time comparison per episode in Building environment
Algorithm Training Time (s)
Random \(0.2\)
MA-SAC \(7.3\)
MA-SAC-L \(13.5\)
Ours \(18.4\)
Max-min GS \(48.6\)
ARAM \(1.7\)

The entropy term is used not only to encourage exploration but also to resolve the indeterminacy issue in solving the dual of the max-min optimization problem [12]. The impact of \(\beta\) is seen in Table 8: The max-min performance in the Building environment is higher for moderate values of \(\beta\) (e.g., 0.1, 0.05) than for very small values (e.g., \(\beta=0.01\)).

Table 9 reports the wall-clock training time per episode for each algorithm. Incorporating constraints increases the training time from MA-SAC to MA-SAC-L. Although our algorithm requires more computation than MA-SAC-L, it achieves a better balance between max-min fairness and constraint satisfaction, as demonstrated in Table 3. Furthermore, our algorithm requires significantly less computation than Max-min GS.

14.5 Multi-Objective Locomotion Control↩︎

Table 10: Hyperparameters for algorithms in multi-objective locomotion control environment
Parameter Value
Shared
optimizer Adam (Kingma & Ba, 2015)
discount (\(\gamma\)) 0.99
target smoothing ratio (\(\tau\)) 0.001
reward dimension 2
max episode step 1000
replay buffer size \(1\times 10^6\)
hidden units per layer 64
minibatch size 32
activation function ReLU
entropy coefficient 0.05
weight learning rate 0.001
weight scheduling \(1/\sqrt{t}\)
Constrained Max-min MORL (Ours)
constraint type maximize
constraint dimension 1
constraint epsilon 1.0
constraint threshold \(-50\)
main learning rate \(7.5 \times 10^{-4}\)
gradient steps for critic update 3
gradient estimation learning rate \(2.5 \times 10^{-5}\)
gradient estimation steps 1
gradient target smoothing ratio 0.001
Unconstrained Max-min MORL (Gaussian)
main learning rate \(7.5 \times 10^{-4}\)
perturbation \(q\) learning rate 0.073
perturbation gradient steps 1
gradient steps for critic update 3
perturbation \(q\)-copies 10
perturbation noise std-dev 0.01
Unconstrained Max-min MORL (ARAM)
main learning rate \(7.5 \times 10^{-4}\)
CI coefficient \(\eta\) 0.2
MD coefficient \(\lambda\) 0.03
Max-average SAC with a Lagrangian Relaxation
constraint type minimize
initial lambda 1.0
main learning rate (actor/critic) \(3 \times 10^{-4}\)
constraint threshold 50
entropy coefficient 0.05
lambda learning rate 0.001
Unconstrained Max-average SAC
main learning rate (actor/critic) \(3 \times 10^{-4}\)

14.6 Evaluation with Different \(C_{th}\)↩︎

Table 11: MoAnt-v5 results over five seeds, with the two constraint-satisfying algorithms highlighted in bold (\(C_{th}=40\)).
Algorithm
\((C_{th}=40)\)
return \((\uparrow)\)
Random \(146.5\) \(48.2\)
MA-SAC \(275.3\) \(98.8\)
MA-SAC-L \(\mathbf{36.8}\) \(78.8\)
Ours \(\mathbf{36.5}\) \(92.9\)
Max-min GS \(111.7\) \(92.7\)
ARAM \(620.7\) \(101.3\)

We set the threshold value \(C_{th} = 40\) in the MoAnt environment, which is more stringent than the value \(C_{th} = 50\) used in Table 5. Our method consistently balances max-min fairness with constraint satisfaction, meeting the constraint while achieving superior max-min performance compared to MA-SAC-L.

14.7 Greenhouse-gas-emission-aware Traffic Management↩︎

Table 12: Hyperparameters for traffic signal control environment
Parameter Value
Shared
optimizer Adam (Kingma & Ba, 2015)
discount (\(\gamma\)) 0.99
target smoothing ratio (\(\tau\)) 0.001
reward dimension 16
total seconds per episode 9000
delta time (seconds) 30
total timesteps \(1\times 10^5\)
replay buffer size \(1\times 10^5\)
hidden units per layer 64
minibatch size 32
activation function ReLU
entropy coefficient 0.05
weight scheduling \(1/\sqrt{t}\)
Constrained Max-min MORL (Ours)
constraint type minimize
constraint dimension 1
constraint epsilon -1.0
constraint threshold \(7.0 \times 10^{4}\)
main learning rate \(7.5 \times 10^{-4}\)
weight learning rate 0.01
gradient steps for critic update 3
gradient estimation learning rate \(1.0 \times 10^{-5}\)
gradient estimation steps 1
gradient target smoothing ratio 0.001
Unconstrained Max-min MORL (Gaussian)
main learning rate \(7.5 \times 10^{-4}\)
weight learning rate 0.01
perturbation \(q\) learning rate 0.073
perturbation gradient steps 1
gradient steps for critic update 3
perturbation \(q\)-copies 20
perturbation noise std-dev 0.01
Unconstrained Max-min MORL (ARAM)
main learning rate 0.001
CI coefficient \(\eta\) 0.00202
MD coefficient \(\lambda\) 0.2
Constrained Max-average PGO
main learning rate 0.01
constraint type minimize
constraint threshold \(7.0 \times 10^{4}\)
constraint learning rate 0.001
Unconstrained Max-average PGO
main learning rate 0.001

Here, \(|\mathcal{A}|=4\). MA-PGO is implemented using the PPO algorithm [64] to maximize the average reward over \(K=16\) objectives. Based on MA-PGO, MA-CPGO applies clipping only to the unconstrained reward part (i.e., \(w \cdot r\)), while leaving the constrained reward part unclipped (i.e., \(u \cdot c\)) and applies the Lagrangian update, following [25] to improve constraint satisfaction. We found that applying clipping to \(u \cdot c + w \cdot r\) does not guarantee constraint satisfaction. Each method is run for 100k timesteps per seed using five random seeds.

15 Limitation, Future Work, and Discussion↩︎

In this section, we discuss several limitations of our work and related future research avenues, although our method offers a promising direction for developing constrained MORL algorithms.

First, there is a lack of well-established benchmarks for MORL compared to standard RL settings [3], and even fewer environments are specifically designed for constrained MORL. Additionally, most existing MORL environments have low-dimensional reward spaces (typically fewer than four dimensions) [42], which limits the ability to evaluate our algorithm in high-dimensional settings. Developing practical benchmarks for both MORL and constrained MORL is therefore a critical research direction for the community.

Second, while it is common in the constrained MDP literature to assume that feasibility is ensured by appropriately chosen thresholds [26], [31], determining such thresholds, that is, setting the constraint set \(\{ C^{(l)} \}_{l=1}^L\), is non-trivial in practice outside of simple or tabular domains. Unlike trial-and-error reward design, constraint threshold design is often infeasible or unsafe due to the potential risks and costs involved. Leveraging external sources of information, such as human demonstrations or natural language descriptions, offers a promising path for setting constraint thresholds in constrained RL and MORL. Another possible approach is to infer the constraint values from expert demonstrations, commonly referred to as inverse constrained RL [65], [66].

Third, while our setting clearly distinguishes rewards from costs, this distinction may be ambiguous in other domains. Determining which objectives should be treated as constraints versus unconstrained rewards can be challenging. As with constraint threshold design, incorporating external guidance could help better structure constrained MORL problems.

Fourth, several constrained RL studies have explored more conservative formulations than those based on expected cumulative cost, for example, using outage probability or quantile-based constraints to manage rare but critical failures in domains such as finance or insurance [56], [67]. While our current framework and analysis rely on expected cumulative cost, extending it to support such conservative constraint formulations presents a valuable direction for future work.

Lastly, although we assume the convergence of the (action) value function for each weight pair \((u,w)\), it is well known that the combination of function approximation, bootstrapped updates, and off-policy learning can lead to instability and even divergence during training [61], [68]. A theoretical investigation into this so-called deadly triad, along with additional convergence guarantees, would further improve the robustness of our algorithm and broaden its applicability to other domains.

Scalarization-based methods are highly valuable, especially because of their interpretability and flexibility in expressing designer preferences. In particular, when incorporating constraints, these methods also make it straightforward to assess constraint satisfaction through the corresponding dual variables. However, linear scalarization cannot recover nonlinear or concave regions of the Pareto frontier, potentially missing desirable trade-off solutions [1], [3]. While mixtures of convex scalarization functions can help approximate concave regions, this often requires careful tuning and may increase computational effort. Addressing the limitations of scalarization-based approaches is indeed valuable.

Finally, we note that the purpose of our theoretical analysis is not to imply that all assumptions will be checked analytically in practice, but rather to provide predictable behavior and guidance for practical usage of our algorithm.

References↩︎

[1]
D. M. Roijers, P. Vamplew, S. Whiteson, and R. Dazeley, “A survey of multi-objective sequential decision-making,” J. Artif. Intell. Res., vol. 48, pp. 67–113, 2013, doi: 10.1613/JAIR.3987.
[2]
R. Yang, X. Sun, and K. Narasimhan, “A generalized algorithm for multi-objective reinforcement learning and policy adaptation,” in Advances in neural information processing systems 32: Annual conference on neural information processing systems 2019, NeurIPS 2019, december 8-14, 2019, vancouver, BC, canada, 2019, pp. 14610–14621, [Online]. Available: https://proceedings.neurips.cc/paper/2019/hash/4a46fbfca3f1465a27b210f4bdfe6ab3-Abstract.html.
[3]
C. F. Hayes et al., “A practical guide to multi-objective reinforcement learning and planning,” Auton. Agents Multi Agent Syst., vol. 36, no. 1, p. 26, 2022, doi: 10.1007/S10458-022-09552-Y.
[4]
L. N. Alegre, A. L. Bazzan, D. M. Roijers, A. Nowé, and B. C. da Silva, “Sample-efficient multi-objective learning via generalized policy improvement prioritization,” arXiv preprint arXiv:2301.07784, 2023.
[5]
J. Teoh, P. Varakantham, and P. Vamplew, “On generalization across environments in multi-objective reinforcement learning,” in The thirteenth international conference on learning representations, 2025, [Online]. Available: https://openreview.net/forum?id=tuEP424UQ5.
[6]
R. Liu et al., “Efficient discovery of pareto front for multi-objective reinforcement learning,” in The thirteenth international conference on learning representations, 2025, [Online]. Available: https://openreview.net/forum?id=fDGPIuCdGi.
[7]
D. Kim, M. Hong, J. Park, and S. Oh, “Conflict-averse gradient aggregation for constrained multi-objective reinforcement learning,” in The thirteenth international conference on learning representations, 2025, [Online]. Available: https://openreview.net/forum?id=ogXkmugNZw.
[8]
W. Kim, J. Lee, J. Lee, and B.-J. Lee, “FairDICE: Fairness-driven offline multi-objective reinforcement learning,” in The thirty-ninth annual conference on neural information processing systems, 2025, [Online]. Available: https://openreview.net/forum?id=2jQJ7aNdT1.
[9]
P. Li, H. Tang, Y. Yuan, J. HAO, Z. Dong, and Y. ZHENG, COLA: Towards efficient multi-objective reinforcement learning with conflict objective regularization in latent space,” in The thirty-ninth annual conference on neural information processing systems, 2025, [Online]. Available: https://openreview.net/forum?id=Cldpn7H3NN.
[10]
U. Siddique, P. Weng, and M. Zimmer, “Learning fair policies in multi-objective (deep) reinforcement learning with average and discounted rewards,” in Proceedings of the 37th international conference on machine learning, ICML 2020, 13-18 july 2020, virtual event, 2020, vol. 119, pp. 8905–8915, [Online]. Available: http://proceedings.mlr.press/v119/siddique20a.html.
[11]
C. Cousins, K. Asadi, E. Lobo, and M. Littman, “On welfare-centric fair reinforcement learning,” in Reinforcement learning conference, 2024, [Online]. Available: https://openreview.net/forum?id=aWGtAqYBie.
[12]
G. Park, W. Byeon, S. Kim, E. Havakuk, A. Leshem, and Y. Sung, “The max-min formulation of multi-objective reinforcement learning: From theory to a model-free algorithm,” in Forty-first international conference on machine learning, ICML 2024, vienna, austria, july 21-27, 2024, 2024, [Online]. Available: https://openreview.net/forum?id=cY9g0bwiZx.
[13]
M. Raeis and A. Leon-Garcia, “A deep reinforcement learning approach for fair traffic signal control,” in 2021 IEEE international intelligent transportation systems conference (ITSC), 2021, pp. 2512–2518.
[14]
E. Zehavi, A. Leshem, R. Levanda, and Z. Han, “Weighted max-min resource allocation for frequency selective channels,” IEEE Trans. Signal Process., vol. 61, no. 15, pp. 3723–3732, 2013, doi: 10.1109/TSP.2013.2262278.
[15]
U. U. Khan, N. Dilshad, M. H. Rehmani, and T. Umer, “Fairness in cognitive radio networks: Models, measurement methods, applications, and future research directions,” J. Netw. Comput. Appl., vol. 73, pp. 12–26, 2016, doi: 10.1016/J.JNCA.2016.07.008.
[16]
K. Regan and C. Boutilier, “Robust policy computation in reward-uncertain MDPs using nondominated policies,” in Proceedings of the twenty-fourth AAAI conference on artificial intelligence, AAAI 2010, atlanta, georgia, USA, july 11-15, 2010, 2010, pp. 1127–1133, doi: 10.1609/AAAI.V24I1.7740.
[17]
A. Saifullah, D. Ferry, J. Li, K. Agrawal, C. Lu, and C. D. Gill, “Parallel real-time scheduling of DAGs,” IEEE Trans. Parallel Distributed Syst., vol. 25, no. 12, pp. 3242–3252, 2014, doi: 10.1109/TPDS.2013.2297919.
[18]
K. Wang, X. Jiang, N. Guan, D. Liu, W. Liu, and Q. Deng, “Real-time scheduling of DAG tasks with arbitrary deadlines,” ACM Trans. Design Autom. Electr. Syst., vol. 24, no. 6, pp. 66:1–66:22, 2019, doi: 10.1145/3358603.
[19]
S. Chakraborty et al., “MaxMin-RLHF: Alignment with diverse human preferences,” in Forty-first international conference on machine learning, ICML 2024, vienna, austria, july 21-27, 2024, 2024, [Online]. Available: https://openreview.net/forum?id=8tzjEMF0Vq.
[20]
E. Eaton, M. Hussing, M. Kearns, A. Roth, S. B. Sengupta, and J. Sorrell, “Intersectional fairness in reinforcement learning with large state and constraint spaces,” in Proceedings of the 42nd international conference on machine learning, 2025, vol. 267, pp. 14918–14933, [Online]. Available: https://proceedings.mlr.press/v267/eaton25a.html.
[21]
X. Chen et al., “Cooling-aware resource allocation and load management for mobile edge computing systems,” arXiv preprint arXiv:2006.10978, 2020.
[22]
C. Jiang et al., “Energy aware edge computing: A survey,” Computer Communications, vol. 151, pp. 556–580, 2020.
[23]
Z. Fan, N. Peng, M. Tian, and B. Fain, “Welfare and fairness in multi-objective reinforcement learning,” in Proceedings of the 2023 international conference on autonomous agents and multiagent systems, AAMAS 2023, london, united kingdom, 29 may 2023 - 2 june 2023, 2023, pp. 1991–1999, doi: 10.5555/3545946.3598870.
[24]
N. Peng, M. Tian, and B. Fain, “Multi-objective reinforcement learning with nonlinear preferences: Provable approximation for maximizing expected scalarized return.” 2025, [Online]. Available: https://arxiv.org/abs/2311.02544.
[25]
Y. Liu, J. Ding, and X. Liu, “IPO: Interior-point policy optimization under constraints.” 2019, [Online]. Available: https://arxiv.org/abs/1910.09615.
[26]
S. Ha, P. Xu, Z. Tan, S. Levine, and J. Tan, “Learning to walk in the real world with minimal human effort,” arXiv preprint arXiv:2002.08550, 2020.
[27]
S. Vaswani, L. Yang, and C. Szepesvari, “Near-optimal sample complexity bounds for constrained MDPs,” in Advances in neural information processing systems, 2022, [Online]. Available: https://openreview.net/forum?id=ZJ7Lrtd12x_.
[28]
M. Calvo-Fullana, S. Paternain, L. F. Chamon, and A. Ribeiro, “State augmented constrained reinforcement learning: Overcoming the limitations of learning with rewards,” IEEE Transactions on Automatic Control, vol. 69, no. 7, pp. 4275–4290, 2023.
[29]
A. Müller, P. Alatur, V. Cevher, G. Ramponi, and N. He, “Truly no-regret learning in constrained MDPs,” in Forty-first international conference on machine learning, 2024, [Online]. Available: https://openreview.net/forum?id=hrWte3nlzr.
[30]
[31]
C. Tessler, D. J. Mankowitz, and S. Mannor, “Reward constrained policy optimization,” arXiv preprint arXiv:1805.11074, 2018.
[32]
S. P. Boyd and L. Vandenberghe, Convex optimization. Cambridge university press, 2004.
[33]
J. Lee, W. Jeon, B.-J. Lee, J. Pineau, and K.-E. Kim, “OptiDICE: Offline policy optimization via stationary distribution correction estimation,” in Proceedings of the 38th international conference on machine learning, ICML 2021, 18-24 july 2021, virtual event, 2021, vol. 139, pp. 6120–6130, [Online]. Available: http://proceedings.mlr.press/v139/lee21f.html.
[34]
T. Haarnoja, H. Tang, P. Abbeel, and S. Levine, “Reinforcement learning with deep energy-based policies,” in Proceedings of the 34th international conference on machine learning, ICML 2017, sydney, NSW, australia, 6-11 august 2017, 2017, vol. 70, pp. 1352–1361, [Online]. Available: http://proceedings.mlr.press/v70/haarnoja17a.html.
[35]
S. Bubeck, “Convex optimization: Algorithms and complexity,” 2015, [Online]. Available: https://arxiv.org/abs/1405.4980.
[36]
W. Wang and M. Á. Carreira-Perpiñán, “Projection onto the probability simplex: An efficient algorithm with a simple proof, and an application,” CoRR, vol. abs/1309.1541, 2013, [Online]. Available: http://arxiv.org/abs/1309.1541.
[37]
C. Boutilier, T. L. Dean, and S. Hanks, “Decision-theoretic planning: Structural assumptions and computational leverage,” J. Artif. Intell. Res., vol. 11, pp. 1–94, 1999, doi: 10.1613/JAIR.575.
[38]
A. Abels, D. M. Roijers, T. Lenaerts, A. Nowé, and D. Steckelmacher, “Dynamic weights in multi-objective deep reinforcement learning,” in Proceedings of the 36th international conference on machine learning, ICML 2019, 9-15 june 2019, long beach, california, USA, 2019, vol. 97, pp. 11–20, [Online]. Available: http://proceedings.mlr.press/v97/abels19a.html.
[39]
T. Basaklar, S. Gumussoy, and Ümit Y. Ogras, PD-MORL: Preference-driven multi-objective reinforcement learning algorithm,” in The eleventh international conference on learning representations, ICLR 2023, kigali, rwanda, may 1-5, 2023, 2023, [Online]. Available: https://openreview.net/pdf?id=zS9sRyaPFlJ.
[40]
W. Hung, B.-K. Huang, P.-C. Hsieh, and X. Liu, “Q-pensieve: Boosting sample efficiency of multi-objective RL through memory sharing of q-snapshots,” in The eleventh international conference on learning representations, ICLR 2023, kigali, rwanda, may 1-5, 2023, 2023, [Online]. Available: https://openreview.net/pdf?id=AwWaBXLIJE.
[41]
H. Lu, D. Herman, and Y. Yu, “Multi-objective reinforcement learning: Convexity, stationarity and pareto optimality,” in The eleventh international conference on learning representations, ICLR 2023, kigali, rwanda, may 1-5, 2023, 2023, [Online]. Available: https://openreview.net/pdf?id=TjEzIsyEsQ6.
[42]
G. Park and Y. Sung, “Reward dimension reduction for scalable multi-objective reinforcement learning,” in The thirteenth international conference on learning representations, 2025, [Online]. Available: https://openreview.net/forum?id=ssRdQimeUI.
[43]
Y. E. Nesterov and V. G. Spokoiny, “Random gradient-free minimization of convex functions,” Found. Comput. Math., vol. 17, no. 2, pp. 527–566, 2017, doi: 10.1007/S10208-015-9296-2.
[44]
W. Byeon, G. Park, J. Chae, A. Leshem, and Y. Sung, “Multi-objective reinforcement learning with max-min criterion: A game-theoretic approach,” in The thirty-ninth annual conference on neural information processing systems, 2025, [Online]. Available: https://openreview.net/forum?id=YFb7KFE39x.
[45]
C. Daskalakis and I. Panageas, “Last-iterate convergence: Zero-sum games and constrained min-max optimization,” arXiv preprint arXiv:1807.04252, 2018.
[46]
S. Miryoosefi, K. Brantley, H. Daume III, M. Dudik, and R. E. Schapire, “Reinforcement learning with convex constraints,” in Advances in neural information processing systems, 2019, vol. 32, [Online]. Available: https://proceedings.neurips.cc/paper_files/paper/2019/file/873be0705c80679f2c71fbf4d872df59-Paper.pdf.
[47]
J. Achiam, D. Held, A. Tamar, and P. Abbeel, “Constrained policy optimization,” in International conference on machine learning, 2017, pp. 22–31.
[48]
S. Paternain, L. Chamon, M. Calvo-Fullana, and A. Ribeiro, “Constrained reinforcement learning has zero duality gap,” Advances in Neural Information Processing Systems, vol. 32, 2019.
[49]
J. Lee et al., COptiDICE: Offline constrained reinforcement learning via stationary distribution correction estimation,” in International conference on learning representations, 2022, [Online]. Available: https://openreview.net/forum?id=FLA55mBee6Q.
[50]
S. H. Huang et al., “A constrained multi-objective reinforcement learning framework,” in Conference on robot learning, 8-11 november 2021, london, UK, 2021, vol. 164, pp. 883–893, [Online]. Available: https://proceedings.mlr.press/v164/huang22a.html.
[51]
Q. Lin, Z. Liu, D. Mo, and C. Yu, “An offline adaptation framework for constrained multi-objective reinforcement learning,” in Advances in neural information processing systems 38: Annual conference on neural information processing systems 2024, NeurIPS 2024, vancouver, BC, canada, december 10 - 15, 2024, 2024.
[52]
M. L. Littman, “Markov games as a framework for multi-agent reinforcement learning,” in Machine learning proceedings 1994, Elsevier, 1994, pp. 157–163.
[53]
T. G. Dietterich, “Hierarchical reinforcement learning with the MAXQ value function decomposition,” Journal of artificial intelligence research, vol. 13, pp. 227–303, 2000.
[54]
C. Yeh et al., “SustainGym: Reinforcement learning environments for sustainable energy systems,” Advances in Neural Information Processing Systems, vol. 36, pp. 59464–59476, 2023.
[55]
T. Haarnoja, A. Zhou, P. Abbeel, and S. Levine, “Soft actor-critic: Off-policy maximum entropy deep reinforcement learning with a stochastic actor,” in Proceedings of the 35th international conference on machine learning, ICML 2018, stockholmsmässan, stockholm, sweden, july 10-15, 2018, 2018, vol. 80, pp. 1856–1865, [Online]. Available: http://proceedings.mlr.press/v80/haarnoja18b.html.
[56]
Q. Yang, T. D. Simão, S. H. Tindemans, and M. T. J. Spaan, WCSAC: Worst-case soft actor critic for safety-constrained reinforcement learning,” in Thirty-fifth AAAI conference on artificial intelligence, AAAI 2021, thirty-third conference on innovative applications of artificial intelligence, IAAI 2021, the eleventh symposium on educational advances in artificial intelligence, EAAI 2021, virtual event, february 2-9, 2021, 2021, pp. 10639–10646, doi: 10.1609/AAAI.V35I12.17272.
[57]
F. Felten et al., “A toolkit for reliable benchmarking and research in multi-objective reinforcement learning,” in Proceedings of the 37th conference on neural information processing systems (NeurIPS 2023), 2023.
[58]
R. Fox, A. Pakman, and N. Tishby, “Taming the noise in reinforcement learning via soft updates,” in Proceedings of the thirty-second conference on uncertainty in artificial intelligence, UAI 2016, june 25-29, 2016, new york city, NY, USA, 2016, [Online]. Available: http://auai.org/uai2016/proceedings/papers/219.pdf.
[59]
R. A. Horn and C. R. Johnson, Matrix analysis. Cambridge university press, 2012.
[60]
D. Silver, “Lectures on reinforcement learning.” 2015.
[61]
R. S. Sutton and A. G. Barto, Reinforcement learning: An introduction. MIT press, 2018.
[62]
T. Yang, S. Cen, Y. Wei, Y. Chen, and Y. Chi, “Federated natural policy gradient and actor critic methods for multi-task reinforcement learning,” 2024, [Online]. Available: https://arxiv.org/abs/2311.00201.
[63]
M. Schmidt, N. L. Roux, and F. Bach, “Convergence rates of inexact proximal-gradient methods for convex optimization,” 2011, [Online]. Available: https://arxiv.org/abs/1109.2415.
[64]
J. Schulman, F. Wolski, P. Dhariwal, A. Radford, and O. Klimov, “Proximal policy optimization algorithms,” CoRR, vol. abs/1707.06347, 2017, [Online]. Available: http://arxiv.org/abs/1707.06347.
[65]
S. Malik, U. Anwar, A. Aghasi, and A. Ahmed, “Inverse constrained reinforcement learning,” in Proceedings of the 38th international conference on machine learning, 2021, vol. 139, pp. 7390–7399, [Online]. Available: https://proceedings.mlr.press/v139/malik21a.html.
[66]
S. G. Subramanian, G. Liu, M. Elmahgiubi, K. Rezaee, and P. Poupart, “Confidence aware inverse constrained reinforcement learning,” in Forty-first international conference on machine learning, 2024, [Online]. Available: https://openreview.net/forum?id=6TCeizkLJV.
[67]
W. Jung, M. Cho, J. Park, and Y. Sung, “Quantile constrained reinforcement learning: A reinforcement learning framework constraining outage probability,” in Advances in neural information processing systems 35: Annual conference on neural information processing systems 2022, NeurIPS 2022, new orleans, LA, USA, november 28 - december 9, 2022, 2022, [Online]. Available: http://papers.nips.cc/paper\_files/paper/2022/hash/2a07348a6a7b2c208ab5cb1ee0e78ab5-Abstract-Conference.html.
[68]
F. Che et al., “Target networks and over-parameterization stabilize off-policy bootstrapping with function approximation,” in Forty-first international conference on machine learning, ICML 2024, vienna, austria, july 21-27, 2024, 2024, [Online]. Available: https://openreview.net/forum?id=R6GT1UDcOW.