MEKAWIRE
ExpertarXiv

Reinforcement Learning Based Heterogeneous Task Allocation Under Spatio-Temporal Environmental Uncertainty for Multi-Robot Teams

Xi Lin, Yaxuan Li, Paul Szenher, Brendan Englot
From the 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.

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

Xi Lin, Yaxuan Li, Paul Szenher, Brendan Englot

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

Fig. 1: Our task allocation problem. The base station of each robot species is marked by a pink square. The capability vectors of robot species and the capability requirements of tasks are shown as color palettes in pink squares and next to robot and task icons, where each color represents a specific capability. Gray, lime and cyan arrows indicate the task location which the robot is en-route to executing, transporting items towards, and ready for (or under) execution, respectively. Background colors, from blue (low) to red (high), show local uncertainty levels that are unknown to robots. White contour lines indicate locations with the same uncertainties.

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]
(1)

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]
(2)

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.

Fig. 2: Policy network architecture. Observations of robots and tasks are passed through corresponding gated encoders to obtain robust embedded features. Then a self-attention module and a cross-attention module are used to fuse the ego robot feature with that of other robots and tasks to obtain a comprehensive understanding. The policy outputs a probability vector that reflects the attentions of the ego robot on all available 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\}
(4)

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)]
(5)

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}
(6)

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
(7)
\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}
(8)
Fig. 3: Examples of initial scenes in experiment episodes. The capability requirements of active tasks at the current moment are shown as color palettes, while hexagons and pentagons with lime colors indicate the start and goal locations of transportation tasks. Markers for tasks that have not yet appeared are the same as those in Fig. 1.

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.

TABLE I: Training hyperparameters.
  • range of n
    [75,150]
    [3,5]
    learning rate
    batch size
    1e-5
    1024
  • range of k
    [75,150]
    [4,6]
    learning rate
    eval ep num
    1e-5
    256
  • max ep time
    [75,150]
    200
    learning rate
    total ep num
    1e-5
    5000
Fig. 4: Training performance. The solid line and bandwidth represent the mean and standard deviation of evaluation results over ten models from ten different training trials.

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

TABLE II: Makespan data in experiments. Each entry shows the mean and standard deviation of results from 100 experiments. (a) 50\% info gathering, 25\% maintenance, 25\% transportation (b) 25\% info gathering, 50\% maintenance, 25\% transportation (c) 25\% info gathering, 25\% maintenance, 50\% transportation
  • RL (Ours)
    50 3 4
    75.92
    100 4 5
    94.72
    200 4 8
    109.50
    200 6 8
    82.10
    400 7 10
    109.26
    400 10 10
    87.25
  • —
    50 3 4
    \bf\pm 22.45
    100 4 5
    \bf\pm 30.29
    200 4 8
    \bf\pm 28.40
    200 6 8
    \bf\pm 18.59
    400 7 10
    \bf\pm 32.39
    400 10 10
    \bf\pm 25.05
  • HBTA
    50 3 4
    101.78
    100 4 5
    119.54
    200 4 8
    129.20
    200 6 8
    96.38
    400 7 10
    121.33
    400 10 10
    91.44
  • —
    50 3 4
    \pm 40.44
    100 4 5
    \pm 39.43
    200 4 8
    \pm 31.49
    200 6 8
    \pm 26.12
    400 7 10
    \pm 36.83
    400 10 10
    \pm 24.72
  • D-HBTA
    50 3 4
    85.79
    100 4 5
    108.95
    200 4 8
    129.36
    200 6 8
    94.91
    400 7 10
    122.66
    400 10 10
    93.57
  • —
    50 3 4
    \pm 28.88
    100 4 5
    \pm 30.30
    200 4 8
    \pm 33.68
    200 6 8
    \pm 25.84
    400 7 10
    \pm 38.05
    400 10 10
    \pm 23.42
  • CBBA
    50 3 4
    90.53
    100 4 5
    116.74
    200 4 8
    139.68
    200 6 8
    105.75
    400 7 10
    134.94
    400 10 10
    105.25
  • —
    50 3 4
    \pm 27.43
    100 4 5
    \pm 29.98
    200 4 8
    \pm 35.41
    200 6 8
    \pm 27.12
    400 7 10
    \pm 30.78
    400 10 10
    \pm 30.36
  • Hungarian
    50 3 4
    108.95
    100 4 5
    134.47
    200 4 8
    153.75
    200 6 8
    109.20
    400 7 10
    136.54
    400 10 10
    99.49
  • —
    50 3 4
    \pm 38.12
    100 4 5
    \pm 38.99
    200 4 8
    \pm 41.47
    200 6 8
    \pm 30.26
    400 7 10
    \pm 38.36
    400 10 10
    \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.

References

  1. [1] J. A. Knauss and N. Garfield (2016) Introduction to physical oceanography. Waveland Press.
  2. [2] T. I. Fossen (2011) Handbook of marine craft hydrodynamics and motion control. John Wiley & Sons.
  3. [3] A. Vagale, R. Oucheikh, R. T. Bye, O. L. Osen, and T. I. Fossen (2021) Path planning and collision avoidance for autonomous surface vehicles i: a review. Journal of Marine Science and Technology 26 (4), pp. 1292–1306.
  4. [4] Y. Liu, R. Song, R. Bucknall, and X. Zhang (2019) Intelligent multi-task allocation and planning for multiple unmanned surface vehicles (USVs) using self-organising maps and fast marching method. Information Sciences 496, pp. 180–197.
  5. [5] C. Wei, Z. Ji, and B. Cai (2020) Particle swarm optimization for cooperative multi-robot task allocation: a multi-objective approach. IEEE Robotics and Automation Letters 5 (2), pp. 2530–2537.
  6. [6] P. Mahato, S. Saha, C. Sarkar, and M. Shaghil (2023) Consensus-based fast and energy-efficient multi-robot task allocation. Robotics and Autonomous Systems 159. Note: Article 104270
  7. [7] B. Fu, W. Smith, D. M. Rizzo, M. Castanier, M. Ghaffari, and K. Barton (2022) Robust task scheduling for heterogeneous robot teams under capability uncertainty. IEEE Transactions on Robotics 39 (2), pp. 1087–1105.
  8. [8] W. Dai, U. Rai, J. Chiun, Y. Cao, and G. Sartoretti (2025) Heterogeneous multi-robot task allocation and scheduling via reinforcement learning. IEEE Robotics and Automation Letters 10 (3), pp. 2654–2661.
  9. [9] M. Liu, H. Ma, J. Li, and S. Koenig (2019) Task and path planning for multi-agent pickup and delivery. In Proceedings of the International Joint Conference on Autonomous Agents and Multiagent Systems (AAMAS), pp. 1152–1160.
  10. [10] Z. Chen, J. Alonso-Mora, X. Bai, D. D. Harabor, and P. J. Stuckey (2021) Integrated task assignment and path planning for capacitated multi-agent pickup and delivery. IEEE Robotics and Automation Letters 6 (3), pp. 5816–5823.
  11. [11] A. Agrawal, A. S. Bedi, and D. Manocha (2023) RTAW: an attention inspired reinforcement learning method for multi-robot task allocation in warehouse environments. In 2023 IEEE International Conference on Robotics and Automation (ICRA), Vol. , pp. 1393–1399.
  12. [12] R. S. Sutton and A. G. Barto (1998) Reinforcement learning: an introduction. MIT Press.
  13. [13] L. Zhou and P. Tokekar (2018) An approximation algorithm for risk-averse submodular optimization. In Workshop on the Algorithmic Foundations of Robotics (WAFR), pp. 144–159.
  14. [14] O. Peltzer, K. Brown, M. Schwager, M. J. Kochenderfer, and M. Sehr (2020) STT-CBS: a conflict-based search algorithm for multi-agent path finding with stochastic travel times. arXiv preprint arXiv:2004.08025.
  15. [15] S. Choudhury, J. K. Gupta, M. J. Kochenderfer, D. Sadigh, and J. Bohg (2022) Dynamic multi-robot task allocation under uncertainty and temporal constraints. Autonomous Robots 46 (1), pp. 231–247.
  16. [16] H. Park and S. Hutchinson (2018) Robust rendezvous for multi-robot system with random node failures: an optimization approach. Autonomous Robots 42 (8), pp. 1807–1818.
  17. [17] X. Huo, H. Zhang, Z. Wang, C. Huang, and H. Yan (2024) A self-learning approach to heterogeneous multi-robot coalition formation under uncertainty. IEEE Transactions on Automation Science and Engineering 22, pp. 3445–3457.
  18. [18] B. P. Gerkey and M. J. Matarić (2004) A formal analysis and taxonomy of task allocation in multi-robot systems. The International Journal of Robotics Research 23 (9), pp. 939–954.
  19. [19] S. Ma, W. Guo, R. Song, and Y. Liu (2021) Unsupervised learning based coordinated multi-task allocation for unmanned surface vehicles. Neurocomputing 420, pp. 227–245.
  20. [20] J. Banfi, A. Messing, C. Kroninger, E. Stump, S. Hutchinson, and N. Roy (2022) Hierarchical planning for heterogeneous multi-robot routing problems via learned subteam performance. IEEE Robotics and Automation Letters 7 (2), pp. 4464–4471.
  21. [21] A. Samiei and L. Sun (2023) Distributed matching-by-clone Hungarian-based algorithm for task allocation of multiagent systems. IEEE Transactions on Robotics 40, pp. 851–863.
  22. [22] W. Babincsak, A. Aswale, and C. Pinciroli (2023) Ant colony optimization for heterogeneous coalition formation and scheduling with multi-skilled robots. In 2023 International Symposium on Multi-Robot and Multi-Agent Systems (MRS), pp. 121–127.
  23. [23] G. Neville, A. Messing, H. Ravichandar, S. Hutchinson, and S. Chernova (2021) An interleaved approach to trait-based task allocation and scheduling. In 2021 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), pp. 1507–1514.
  24. [24] J. Huang, X. Li, and L. Gao (2025) A novel GA-CP method for fixed-type multi-robot collaborative scheduling in flexible job shop. IEEE Transactions on Automation Science and Engineering 22, pp. 13531–13543.
  25. [25] H. Ma, J. Li, T. S. Kumar, and S. Koenig (2017) Lifelong multi-agent path finding for online pickup and delivery tasks. In Proceedings of the 16th Conference on Autonomous Agents and MultiAgent Systems (AAMAS), pp. 837–845.
  26. [26] K. Helsgaun (2017) An extension of the Lin-Kernighan-Helsgaun TSP solver for constrained traveling salesman and vehicle routing problems. Roskilde: Roskilde University 12, pp. 966–980.
  27. [27] H. Makino and S. Ito (2024) Online multi-agent pickup and delivery with task deadlines. In 2024 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), pp. 8428–8434.
  28. [28] A. Prorok, M. A. Hsieh, and V. Kumar (2017) The impact of diversity on optimal control policies for heterogeneous robot swarms. IEEE Transactions on Robotics 33 (2), pp. 346–358.
  29. [29] J. M. Wallace and P. V. Hobbs (2006) Atmospheric science: an introductory survey. Vol. 92, Elsevier.
  30. [30] G. K. Vallis (2017) Atmospheric and oceanic fluid dynamics. Cambridge University Press.
  31. [31] U. Frisch (1995) Turbulence: the legacy of a. n. kolmogorov. Cambridge University Press.
  32. [32] A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, Ł. Kaiser, and I. Polosukhin (2017) Attention is all you need. Advances in Neural Information Processing Systems 30.
  33. [33] G. A. Korsah, A. Stentz, and M. B. Dias (2013) A comprehensive taxonomy for multi-robot task allocation. The International Journal of Robotics Research 32 (12), pp. 1495–1512.
  34. [34] H. Choi, L. Brunet, and J. P. How (2009) Consensus-based decentralized auctions for robust task allocation. IEEE Transactions on Robotics 25 (4), pp. 912–926.
  35. [35] M. Braquet and E. Bakolas (2021) Greedy decentralized auction-based task allocation for multi-agent systems. IFAC-PapersOnLine 54 (20), pp. 675–680.
  36. [36] Y. Msala, O. Hamed, M. Talea, and M. Aboulfatah (2023) A new method for improving the fairness of multi-robot task allocation by balancing the distribution of tasks. Journal of Robotics and Control (JRC) 4 (6), pp. 743–753.