Reinforcement Learning Based Heterogeneous Task Allocation Under Spatio-Temporal Environmental Uncertainty for Multi-Robot Teams
Autonomous Surface Vehicles (ASVs) have been increasingly deployed to perform different kinds of maritime tasks in recent years, which motivates the need for a reliable task allocation policy that can adapt to heterogeneous tasks and maritime disturbances. In this work, we formulate a multi-robot task allocation problem in which a team of ASVs with heterogeneous capabilities needs to complete tasks of different types and appearance times under unknown stochastic spatio-temporal uncertainties from maritime disturbances.
From the abstract. Our summary is in progress.
Full paper
Reinforcement Learning Based Heterogeneous Task Allocation Under Spatio-Temporal Environmental Uncertainty for Multi-Robot Teams
Published under CC BY 4.0. Reproduced with attribution; original at arXiv:2610.15237 (PDF).
Abstract
Autonomous Surface Vehicles (ASVs) have been increasingly deployed to perform different kinds of maritime tasks in recent years, which motivates the need for a reliable task allocation policy that can adapt to heterogeneous tasks and maritime disturbances. In this work, we formulate a multi-robot task allocation problem in which a team of ASVs with heterogeneous capabilities needs to complete tasks of different types and appearance times under unknown stochastic spatio-temporal uncertainties from maritime disturbances. We propose a decentralized Reinforcement Learning (RL) based policy network that is trained to be robust towards uncertainty-induced observation deviations, and is able to perform individual decision making that leads to efficient execution behaviors toward heterogeneous task requirements. We evaluate the proposed policy and four baseline methods in simulated experiments with varied proportions of different tasks types, numbers of tasks, robot species and robots per species. Experiment results show that our proposed policy achieves the lowest makespan and can be deployed to address scenarios with 100 robots and 400 tasks.
I Introduction
Autonomous Surface Vehicles (ASVs) in maritime environments are typically subject to the influences of spatially and temporally varying weather and sea conditions [1]. When operating in areas covered by storms, the motions of ASVs can be substantially influenced by storm-induced wind, waves and ocean currents [2], such that it can take a longer time to travel to designated locations and perform assigned tasks. In addition, communication quality throughout the environment may degrade significantly due to rain-induced signal attenuation and antenna misalignment from significant motion disturbances in stormy regions. In this work, we focus on the task allocation problem of ASVs under the effects of spatio-temporal maritime environmental disturbances like storms.
Given that autonomous vehicles can prevent the human-induced errors and risks that exist in human-controlled vehicles, ASVs have been increasingly deployed for various kinds of maritime missions [3]. To evaluate the performance of task allocation solutions over different kinds of tasks, we consider important maritime missions with heterogeneous task requirements. As shown in Fig. 1, our problem includes information gathering tasks that are stationary single-robot tasks [4, 5, 6], maintenance tasks that require multi-robot collaboration [7, 8], and transportation tasks that are similar to those in pickup-and-delivery scenarios [9, 10, 11]. As our task allocation environment is vast and unstructured, the paths of transportation tasks are simplified as straight lines connecting start and goal locations. Furthermore, our tasks have different appearance times, such that the task allocation policy needs to perform efficient re-planning when new tasks show up in the environment [10, 11].
Due to the communication issues induced by environmental uncertainties, task allocation methods that use centralized planners may experience robot action failures when robots don’t receive commands from the planner. For decentralized methods, each robot makes individual decisions and can always act when available, but it can be challenging to perform multi-robot collaboration in a decentralized manner. We aim to take advantage of the action robustness of a decentralized method (due to individual decision making), and to foster cooperative behaviors among robots when necessary. Accordingly, we propose a decentralized decision making policy based on Deep Reinforcement Learning (DRL), which can adapt to heterogeneous tasks via trial-and-error learning [12]. Since the observations of task and robot states support the estimation of travel and task execution times, which may deviate from the actual values under the influence of environmental disturbances, we design a policy model structure that can learn to be robust to erroneous estimation of task execution time. The contributions of this work are summarized as follows:
- We formulate a multi-robot task allocation problem under the challenges of unknown stochastic spatio-temporal uncertainties in maritime environments, and tasks of heterogeneous types with different appearance times.
- We design a decentralized policy network that handles deviated observations due to environmental uncertainties via an adaptive weighting module, and performs attention-based fusions for team-and-task-aware individual decision making. An episodic reward considering episode time length and statuses of heterogeneous tasks is used to train a Reinforcement Learning (RL) based policy with the proposed model.
- The proposed RL based policy has been comprehensively evaluated against four baselines in scenarios with varying task type proportions, numbers of tasks, robot species and robots per species. Experiment results show that the proposed method demonstrates superior performance over all baselines.
- Our approach is freely available at https://github.com/RobustFieldAutonomyLab/Spatio_Temporal_MRTA.
II Related Works
Many existing works on multi-robot collaboration model uncertainties with basic probability distributions. Zhou et al. [13] proposed a sequential greedy algorithm for a multi-vehicle mobility-on-demand problem with path-related arrival time uncertainty. Peltzer et al. [14] studied a multi-agent path finding problem in warehouse scenarios where agent travel time stochasticity follows a gamma distribution. Choudhury et al. [15] proposed a stochastic conflict-based task allocation method for multi-drone package delivery missions with travel time uncertainty based on Epanechnikov and normal distributions. Park et al. [16] targeted a multi-robot collaboration problem under random robot failures modeled with the binomial distribution and proposed a distributed stochastic optimal control solution. Huo et al. [17] introduced agent observation uncertainties, modeled with Dirichlet distributions, into a multi-robot coalition formation problem and addressed it with a belief updating method. Our work presents a novel study on a multi-robot task allocation problem under spatio-temporal environmental uncertainties that affect robot travel and task execution speed, communication quality, and observations of robot and task states.
Our problem in general falls into the ST-MR-TA category (single-task robots, have multi-robot tasks, and may need time-extended assignments) [18]. Specifically, it contains heterogeneous types of tasks across different categories of task allocation problems, which involve stationary single-robot tasks, multi-robot collaboration tasks, and pickup-and-delivery style tasks with simplified paths. Relevant prior works addressing each of these three classes of tasks are summarized in the three respective paragraphs that follow.
(1) Stationary Single-Robot Tasks: Ma et al. [19] proposed a multi-ASV task allocation policy based on self-organizing maps for single-robot maritime tasks in constrained environments. Wei et al. [5] formulated the multi-robot allocation problem as multiple Traveling Salesman Problems (TSPs), applying a multi-objective particle swarm optimization solution. Banfi et al. [20] formulated a multi-robot heterogeneous area inspection mission as a vehicle routing problem, and developed a hierarchical planner which adopts a Graph Neural Network for allocation cost estimations. Mahato et al. [6] proposed a market-based method which demonstrated energy-efficient performance on a distributed multi-robot task allocation problem. Samiei et al. [21] converted a multi-task assignment problem for multi-agent systems into a linear sum assignment problem for local planning, and proposed a solution based on the Hungarian algorithm.
(2) Multi-Robot Collaboration Tasks: Fu et al. [7] developed a Mixed-Integer Programming framework for solving task assignments and scheduling for multi-robot tasks with uncertainties on robot team capabilities. Dai et al. [8] proposed a Reinforcement Learning based policy that achieves comparable makespan to optimization based methods with efficient computation time and good scalability towards large-scale problems. Babincsak et al. [22] formulate a multi-objective optimization problem for multi-robot collaboration that considers the costs of all robots’ paths and the maximum individual cost, which can be solved with Ant Colony Optimization based methods. Neville et al. [23] introduced a heuristic based graph search approach to addressing requirements of multi-robot task scheduling and motion planning, and evaluates heuristics on task allocation quality as well as plan execution time. Huang et al. [24] targeted the multi-robot collaborative scheduling problem in manufacturing scenarios and utilized genetic algorithm and constraint programming to deal with optimizations at different levels. Optimization-based methods like [7] have exponential complexity, and it can take hundreds of seconds to generate solutions for large-scale problems [8].
(3) Pickup-and-Delivery Tasks: Ma et al. [25] introduced the multi-agent pickup and delivery (MAPD) problem as an extension of a multi-agent path finding problem with continual new incoming tasks, and performed online task allocation with a Hungarian-based method. Liu et al. [9] formulated the MAPD problem as a Traveling Salesman Problem, computing task sequences for each agent with a TSP solver [26]. Chen et al. [10] aimed at the MAPD problem in warehouses and sortation centers, and presented a regret-based solution that iteratively assigns tasks to robots with minimum marginal costs. Agrawal et al. [11] proposed a Reinforcement Learning based method that trains a policy to minimize the total travel delay during task execution and showed improvement over the regret-based approach. Makino et al. [27] introduced task deadlines into the multi-agent pickup and delivery problem, and proposed algorithms that consider both task costs and deadlines, and strategies for swapping tasks among robots.
III Problem Formulation
III-A Reinforcement Learning
The task allocation problem can be formulated as a Markov Decision Process (\mathcal{S},\mathcal{A},P,R,\gamma) [8, 11], where \mathcal{S} and \mathcal{A} are the sets of environment states and agent actions, which are determined by the set of active tasks. P(s^{\prime}|s,a) is the state transition function reflecting the environment dynamics. R(s,a) is the reward function, and \gamma\in[0,1] is the discount factor. At each time step t, given the observation of the current state s_{t}, each agent chooses an action a_{t}, which causes the transition to the state s_{t+1}\sim P(\cdot|s_{t},a_{t}) and receiving the reward r_{t+1}=R(s_{t+1},a_{t+1}). The value function V^{\pi}(s) is defined as the expected cumulative reward of taking actions according to the policy \pi when starting from state s.
V^{\pi}(s)=\mathbb{E}_{\pi}[\sum_{k=0}^{\infty}\gamma^{k}r_{t+k+1}|s_{t}=s]An optimal policy \pi_{*} maximizes V^{\pi}(s), and the resulting optimal value function V^{\pi_{*}}(s) satisfies the Bellman optimality equation (2).
V^{\pi_{*}}(s)=\max_{a}\mathbb{E}[r_{t+1}+\gamma V^{\pi_{*}}(s_{t+1})|s_{t}=s,a_{t}=a]III-B Heterogeneous Robots and Tasks
Similar to relevant prior works [8, 28, 7], we assume that the whole robot team can be divided into multiple robot species, and robots within a certain species share the same capability vector. The location of a base station indicates the initial position of robots of the corresponding species, and all robots need to return to their base stations to complete the episode. As shown in Fig. 1, in this work we consider three heterogeneous types of tasks:
(1) Transportation: Its requirement is the same as the capability vector of one specific robot species and can only be performed by one of those robots. The executing robot needs to travel from the transportation start location to goal location to complete the task.
(2) Maintenance: It requires that all capabilities in its requirement need to be satisfied by the cumulative capabilities of the group of executing robots, and the execution process doesn’t start until all robots reach the task location.
(3) Information Gathering: Its requirement contains only one capability and can be performed by any robot that owns the capability upon reaching the task location.
III-C Stochastic Spatio-Temporal Environmental Uncertainty
To study the policy performance under the high variability of maritime environments, we construct a computationally efficient stochastic spatio-temporal uncertainty simulation inspired by fundamental atmospheric transport principles.
We use Gaussian ellipses to model localized high uncertainty regions like storm cores, which are commonly used as simplified representations of mesoscale atmospheric cells [29]. The uncertainties are transported by a slowly varying wind vector, approximating the classical advection equation widely used to model atmospheric scalar transport [30]. A multi-octave Gaussian-smoothed noise process is used to generate large-scale correlated background variability, approximating the multi-scale spatial structure commonly observed in atmospheric phenomena [31]. After superimposing the components mentioned above, we normalize the uncertainty values to vary between 0 and 1.
We assume that the amounts of delay in local travel speed and task execution speed of each robot are proportional to its local uncertainty level. In addition, the probabilities of communication failures among robots as well as with centralized planners (used by some of the baseline methods in experiments) are proportional to their local uncertainty levels. We also assume that centralized planners and robot onboard decentralized planners are unaware of the uncertainty value distributions over the whole working environment.
IV Methodology
IV-A Observation
When a robot is at a decision-making moment, it is given the state observation of itself, other robots the information of which are successfully obtained at the current moment, as well as active tasks. C is the robot capability vector; P is the relative position of a task (for a transportation task it is the start position) or another robot with respect to the ego robot; \textbf{{P}}_{goal} is the relative goal position of a transportation task (for other tasks it is the same as P); R is the task requirement vector; S reflects the capabilities in R that have not yet been satisfied by assigned robots; I indicates whether R is completely satisfied; t_{wait\rightarrow T} is the difference between current time and the time when the robot reached the task location; \hat{t}_{travel\rightarrow T} and \hat{t}_{travel\rightarrow T} represent the estimated travel time to assigned task T when there is no interference from the environmental uncertainty. Before passing into the encoder, robot observations are concatenated with the summary information of assigned tasks.
IV-B Reward
The objective in this problem is to obtain a policy that minimizes the total makespan when each agent performs individual decision making, and it is important to reduce myopic and greedy actions that only consider the current situation. Hence we assume that \gamma=1.0 and r=0 until the end of an episode is reached, where an episodic reward r=r_{\text{episode}} as shown below is given, such that the value function at all intermediate states in an episode would be equal to r_{\text{episode}} and thus they contribute equally to the final outcome.
\displaystyle r_{\text{episode}} \\ \displaystyle=-t_{\text{episode}}-\sum_{T_{j}\in\mathbb{F}}(I(T_{j}\notin\mathbb{T})\cdot||\Sigma_{i\in A(T_{j})}C_{i}||/||R_{j}|| \\ \displaystyle+I(T_{j}\in\mathbb{T}))\cdot r_{\alpha}-|T_{l}\notin\mathbb{F}|\cdot r_{\beta}t_{\text{episode}} is the total time step of the episode, which is terminated if all tasks have been completed and all robots have returned to base stations, or a given max time step is reached; I(\cdot) is the indicator function; \mathbb{T} is the set of transportation tasks; A(\cdot) is the set of assigned robots to a task; F is the set of finished tasks at the end of an episode. r_{\alpha} and r_{\beta} are the penalties for finished and unfinished tasks.
Under the episodic reward setting, similar to the approach in [8], we adopt a Monte-Carlo policy gradient method for policy training, which aims at maximizing the objective function R(\tau) that is the expected reward of an episode \tau.
J(\theta)=\mathbb{E}_{\tau\sim\pi_{\theta}}[R(\tau)],\ \tau=\{(i_{k},o_{k},a_{k})|k=1,\dots,K\}In our problem setting, R(\tau) is the episodic reward shown in Equation (3), and an episode is a set of decision making experiences where i, o and a in each experience represent the index of the decision making agent, its observations and action. The gradient for policy updates, \nabla_{\theta}J(\theta), is shown as follows. A baseline value b is used to reduce variance, which is computed as the average episodic reward of multiple runs of a training episode.
\nabla_{\theta}J(\theta)=\mathbb{E}[\sum_{k=1}^{K}\nabla_{\theta}\log\pi_{\theta}(a_{k}|o_{k})(R(\tau)-b)]IV-C Policy Model Structure
Given that observations contain estimated future travel and task execution information, which may deviate from the actual values that are influenced by environmental uncertainties, we adopt robot and task encoders with gate units to mitigate the effect of deviated estimations on embedded features, which is shown in Equation (6).
\mathcal{F}=\ \mathcal{F_{\text{raw}}}\odot\sigma(W_{2}\mathcal{F_{\text{raw}}}+b_{2}),\ \mathcal{F_{\text{raw}}}=W_{1}\mathcal{O}+b_{1}Raw features \mathcal{F_{\text{raw}}} are obtained by performing linear projections on observations \mathcal{O}, then we compute adaptive weights of \mathcal{F_{\text{raw}}} via another linear unit and sigmoid function \sigma(\cdot), which are applied to \mathcal{F_{\text{raw}}} via element-wise multiplication \odot to produce robust embeddings \mathcal{F}.
During the decentralized decision making, each robot needs to independently consider the status of other robots and active tasks to balance the demands such as efficiently reducing the remaining task work amount, avoiding wasting time on tasks that other robots may do better, and helping with cooperative tasks where others have been waiting. Hence, we perform feature fusions in the robot self-attention module and robot-task cross-attention module, which are based on the scaled dot-product attention function \text{Attention}(\cdot) and the multi-head attention function \text{MHA}(\cdot), which are defined in [32] and are shown below.
\text{Attention}(Q,K,V)=\text{Softmax}(\frac{QK^{T}}{\sqrt{d_{k}}})V\begin{array}[]{c}\text{MHA}(Q,K,V)=C(H_{1},\dots,H_{h})W^{O},\\[5.0pt]
\text{where }H_{i}=\text{Attention}(QW^{Q}_{i},KW^{K}_{i},VW^{K}_{i})\end{array}In the robot self-attention module, Q, K and V are the same and come from the embedded robot features such that each robot feature is fused with that of other robots according to the learned attentions at the current time step, and the module outputs team-status-aware robot features. In the robot-task cross-attention module, Q are the team-status-aware robot features, K and V come from the embedded task features, then the learned attentions to each task serve as the weights to fuse task features into the team-status-aware robot features, resulting in comprehensively fused features. The policy outputs task assignment probabilities via a scaled dot-product attention function given the final fused feature of the ego robot and task features. A categorical distribution on the task assignment probabilities is used to determine the next task during the training process, while in experiments the task with the maximum probability is chosen.
IV-D Model Training
We performed the training process on an Nvidia RTX 3090 GPU. The list of training hyperparameters we use is shown in Table I, where ep, m, n, k stand for episode, the number of tasks, robot species, and robots per species. To enable the trained policy to generalize to a variety of scenarios with different map sizes where numbers and densities of tasks and robots vary, the location coordinates in simulation environments are normalized between 0 and 1. Each episode is initialized with random m, n and k within the range, random task and robot base station locations, as well as random batches of appearance times for tasks. We evaluate the learning model after every 500 training episodes, and the training performance is summarized in Fig. 4, which shows that the model performance converges in 5000 episodes.
- range of n
- [3,5]
- batch size
- 1024
- range of k
- [4,6]
- eval ep num
- 256
- max ep time
- 200
- total ep num
- 5000
V Experiments
V-A Experiment Settings
To comprehensively evaluate the performance of our proposed policy in this problem, we consider four additional methods which serve as comparative baselines and are introduced in Sec. V-B, and perform multiple sets of simulation experiments with varying task type proportions, m, n and k, where each set of experiments (each column in Table II(c)) contains 100 experiment episodes. Each experiment episode is generated in the same manner as described in Sec. IV-D, and the visualizations of example initial environment configurations are shown in Fig. 3. Note that the initial positions of all robots of the same species are the same as the location of the corresponding base station, and we show robot icons surrounding base stations in Fig. 3 just to indicate the number of robots per species.
V-B Baselines
Given that tasks may appear at different moments in our problem, we choose the following baseline methods which are able to perform efficient online re-planning, and make necessary adaptations to our problem with heterogeneous types of tasks. The cost function used by the baselines for evaluating each robot-to-task assignment is computed as the sum of estimated travel time and estimated task time.
(1) Heuristic Based Task Allocation (HBTA): Greedy heuristic methods have been widely implemented as evaluation baselines [8, 11, 20]. To adapt to multi-robot tasks in our problem, we utilize the idea of task decomposition [33] and divide each cooperative task into single capability requirement slots. Then, for every capability requirement slot in each task, a min cost heap is constructed for all available robots, and the robot with minimum cost and without any assignment conflicts is selected.
(2) Decentralized HBTA (D-HBTA): In contrast to HBTA, for which a centralized planner considers the costs of all available robots, we also include a decentralized version that assigns a task to the ego robot only based on its marginal costs with respect to all active tasks.
(3) Consensus-Based Bundle Algorithm (CBBA): Based on the method in [34], a distributed combinatorial auction process is formulated, where each robot iteratively constructs a task bundle by greedily selecting assignments that minimize its marginal cost under feasibility constraints, followed by a consensus phase where neighboring robots exchange bid information to resolve conflicts and achieve agreement on task ownership. Similar to the idea in [35], multiple robots are allowed to win the same cooperative task until all capability requirements of the task are satisfied.
(4) Hungarian Algorithm (Hungarian): In multi-robot task allocation problems, as illustrated in [36], given a bipartite graph between robots and tasks with an associated cost matrix where each entry represents the cost of a specific robot-to-task assignment, the Hungarian algorithm computes a one-to-one assignment that minimizes the total cost by solving the linear sum assignment problem. For cooperative tasks, we also adopt the task decomposition technique mentioned above such that the one-to-one assignment constraint in the Hungarian Algorithm is satisfied.
V-C Results
- RL (Ours)
- 75.92
- 94.72
- 109.50
- 82.10
- 109.26
- 87.25
- —
- \bf\pm 22.45
- \bf\pm 30.29
- \bf\pm 28.40
- \bf\pm 18.59
- \bf\pm 32.39
- \bf\pm 25.05
- HBTA
- 101.78
- 119.54
- 129.20
- 96.38
- 121.33
- 91.44
- —
- \pm 40.44
- \pm 39.43
- \pm 31.49
- \pm 26.12
- \pm 36.83
- \pm 24.72
- D-HBTA
- 85.79
- 108.95
- 129.36
- 94.91
- 122.66
- 93.57
- —
- \pm 28.88
- \pm 30.30
- \pm 33.68
- \pm 25.84
- \pm 38.05
- \pm 23.42
- CBBA
- 90.53
- 116.74
- 139.68
- 105.75
- 134.94
- 105.25
- —
- \pm 27.43
- \pm 29.98
- \pm 35.41
- \pm 27.12
- \pm 30.78
- \pm 30.36
- Hungarian
- 108.95
- 134.47
- 153.75
- 109.20
- 136.54
- 99.49
- —
- \pm 38.12
- \pm 38.99
- \pm 41.47
- \pm 30.26
- \pm 38.36
- \pm 27.00
The experiment results are summarized in Table II(c). In general, our proposed policy shows scalability towards large robot teams (100 robots) and substantial numbers of tasks (400 tasks). Compared to baseline methods, our method achieves the lowest makespan across different sets of experiments, indicating the robustness of our proposed policy in multi-robot task allocation scenarios under environmental uncertainties and heterogeneous task requirements.
Firstly, due to the influence of unknown environmental uncertainty, observation-based travel and task time estimation may deviate from the actual values to a varied extent, which compromises the accuracy of the assignment cost function used by baseline methods, leading to performance degradation. Our proposed policy is trained to deal with these deviations in observation via adaptive weighting of the raw features, such that the policy is less vulnerable to erroneous and/or corrupted observations.
In addition, communication failures due to the effect of environmental uncertainty impacts the evaluated methods to different degrees. For centralized methods, HBTA and Hungarian, a robot can perform an action only when it successfully sends an ego observation to and receives an action command from the centralized planner. Consequently, the number of acting robots is less than that of available robots. On the contrary, for decentralized methods, RL, D-HBTA and CBBA, all available robots can make action decisions on their own. The number of acting robots can have a higher impact on algorithm performance when the total number of robots is small (n=3, k=4), and it can be seen that decentralized methods generally perform better than centralized ones.
Among decentralized methods, although CBBA enables every available robot to obtain an action, it may require multiple rounds of information exchange among robots, and thus the decision making process can be more vulnerable to communication failures. This results in higher makespan data compared to RL and D-HBTA, which require less communication among robots. Since D-HBTA is a greedy approach that only considers the costs of the ego robot, our proposed policy that also takes the statuses of other robots into account leads to better team performance, especially in experiments with higher proportions of maintenance tasks that require multi-robot collaboration.
VI Conclusion
In this work, we study a multi-robot task allocation problem in maritime environments with varying spatio-temporal uncertainties that are unknown to the robots, and heterogeneous tasks with different appearance times and requirements on robot capabilities. We proposed a decentralized RL based policy with a model structure that can learn robust team-and-task-aware features for individual decision making. In the training process, we adopt an episodic reward that evaluates the episode time cost and heterogeneous task statuses to learn an efficient policy for various working scenarios. The proposed policy shows superior performance in makespan when being evaluated against four baseline methods in simulation experiments that involve varying heterogeneous task type proportions, numbers of tasks, robot species and robots per species. The proposed policy also demonstrates scalability to scenarios with 100 robots and 400 tasks. For future work, we may further study the policy performance on energy cost, and incorporate communication uncertainties with task sites into the problem.