Graph Neural Network-Based Task Migration Strategy for Battery Run Time Optimization of Smart Medical Device Cluster

The interconnection of smart medical devices is becoming an important trend in healthcare. Due to the differences in task requirements and power reserves of smart medical device clusters, if some devices run out of energy prematurely, it may lead to a decrease in the performance and reliability of the whole system. Thus, it is particularly important to globally optimize the run time of smart medical device clusters. This paper proposes a task migration strategy for battery run time collaborative optimization of smart medical device clusters. Firstly, a feature factor of device run time is proposed to evaluate the difference in the distribution of remaining power and pending tasks in the device cluster. Secondly, each device regulates the local scheduling according to the feature factor, so the dominant device reserves more free time for task migration. Finally, a reinforcement learning-based task migration strategy is proposed, where a graph neural network embeds graph features on the feature factor and its associated device and sub-task features. The task migration actions are selected based on the graph features for collaborative optimization of run time for smart medical device clusters. Validated in the distributed battery-drive medical device hardware-in-the-loop experimental platform, the results show that in the cluster with balanced battery run time, the proposed method effectively extends the overall battery run time of the cluster by up to 4.26%. In the cluster with unbalanced battery run time/task requirements, the proposed method can effectively extend the cluster run time by more than 14.64%.