EAAI Journal 2026 Journal Article
Class label enhanced Wasserstein distance for classification of remote sensing smoke-related scenes
- Shikun Chen
- Xin Lu
- Xiaobo Lu
Classification of remote sensing (RS) smoke-related scenes is a tough task due to their high inter-class similarity. To improve the classification performance, it is a wise choice to better distinguish various scenes on the feature space by deep learning models. The optimal transport (OT), which measures the difference between probability distributions by Wasserstein distance (WD), fits well with this idea. Served as a loss function, WD enables closer distance between samples matched with each other in the transportation plan. Here in classification tasks, it is naturally expected samples belonging to the same scene are matched. In traditional WD methods, distances between samples are only measured on the feature space. Now with the existence of class labels, it is proposed samples could also be compared on the label space to further reduce distances between same class samples in the transportation plan. Based on the fact samples of the same scene should form a cluster on the feature space, it is proposed class labels can also be predicted according to the spatial relationships of feature representations besides the output of classifier. In this work, we utilize class labels both predicted based on the spatial relationships and the classifier to reduce mis-mappings in the transportation plan. The proposed algorithm is named as Class Label Enhanced Wasserstein Distance (CLEWD), and extensive experiments show CLEWD outperforms other state-of-the-art (SOTA) algorithms in classification of RS smoke-related scenes.