GWS: Rotation object detection in aerial remote sensing images based on Gauss–Wasserstein scattering

Abstract

The majority of existing rotating target detectors inherit the horizontal detection paradigm and design the rotational regression loss based on the inductive paradigm. But the loss design limitation of the inductive paradigm makes these detectors hardly detect effectively tiny targets with high accuracy, particularly for large-aspect-ratio objects. Therefore, in view of the fact that horizontal detection is a special scenario of rotating target detection and based on the relationship between rotational and horizontal detection, we shift from an inductive to a deductive paradigm of design to develop a new regression loss function named Gauss–Wasserstein scattering (GWS). First, the rotating bounding box is transformed into a two-dimensional Gaussian distribution, and then the regression losses between Gaussian distributions are calculated as the Wasserstein scatter; By analyzing the gradient of centroid regression, centroid regression is shown to be able to adjust gradients dynamically based on object characteristics, and small targets requiring high accuracy detection rely on this mechanism, and more importantly, it is further demonstrated that GWS is scale-invariant while possessing an explicit regression logic. The method is performed on a large public remote sensing dataset DOTA and two popular detectors and achieves a large accuracy improvement in both large aspect ratio targets and small targets detection compared to similar methods.