Coordinating heterogeneous robots is essential for autonomous multi-robot teaming.
To execute a set of dependent tasks as quickly as possible,
and to complete tasks that cannot be addressed by individual robots,
it is necessary to form subteams that can collaboratively finish the tasks.
It is also advantageous for robots to
wait for teammates and tasks to become available
in order to form better subteams or reduce the overall completion time.
To enable both abilities,
we introduce a new graph learning approach
that formulates heterogeneous collaborative scheduling as a bipartite matching problem that maximizes a reward matrix learned via imitation learning.
We design a novel graph attention transformer network (GATN)
that represents the problem of collaborative scheduling as a bipartite graph,
and integrates both local and global graph information to estimate the reward matrix using graph attention networks and transformers.
By relaxing the constraint of one-to-one correspondence in bipartite matching,
our approach allows multiple robots to address the same task as a subteam.
Our approach also enables voluntary waiting by introducing an idle task that the robots can select to wait.
Experimental results have shown that our approach well addresses heterogeneous collaborative scheduling with dynamic subteam formation and voluntary waiting,
and outperforms the previous and baseline methods.