Bias-Compensation Augmentation Learning for Semantic Segmentation in UAV Networks

Abstract

In the realm of emergency disaster relief, it is paramount to attain a thorough comprehension of the semantic information associated with the local disaster scene for strategic rescue path planning and immediate rescue operations for affected individuals. Unmanned aerial vehicle (UAV) networks are widely utilized for rapid data collection in the aftermath of disasters due to their flexibility and maneuverability, assisting in rescue decision-making. However, some disasters, such as seismic events and floods have disrupted the initially structured ground shape information, leading to a disparate distribution of data collected by various UAV groups. This exposes traditional semantic segmentation models susceptible to shortcut bias, posing challenges in adapting to semantic segmentation tasks in disaster scenarios. Thus, this paper proposes a bias-compensation augmentation learning based semantic segmentation framework, which substantially enhances the extraction capability of semantic information. Initially, we exploit an artificial augmentation neural network for bias-awareness to determine the relative bias values of the collected image data. Subsequently, considering the limited computing power resources in UAV networks, we present a bias compensation computation offloading strategy to achieve a relatively balanced distribution of semantic information across UAV nodes, optimizing the trade-off between network scheduling efficiency and model accuracy. We conduct reconstruction validation on the FloodNet dataset, and a plethora of experimental results demonstrate that, compared to traditional methods, this approach greatly improves the accuracy of pixel-level semantic segmentation by over 86.5%. Moreover, the average combined processing time is also reduced by over 50%, enhancing the utilization efficiency of limited computational resources.