Multitask Fine-Grained Feature Mining for Multilabel Remote Sensing Image Classification.

Jie Guo,Hao Sun,Jinheng Han,Bin Song,Yuhao Chi,Bingxi Song
DOI: https://doi.org/10.1109/tgrs.2024.3426473
IF: 8.2
2024-01-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Multilabel remote sensing image classification can provide comprehensive object-level semantic descriptions of remote sensing images. However, most existing methods cannot fully mine the fine-grained features of images and labels, resulting in low classification accuracy. To address this issue, we propose a novel multitask framework for multilabel remote sensing image classification. The framework establishes the class-specific feature extraction as a binary classification auxiliary task to assist the main multilabel classification task, which can improve the model's local and global feature extraction ability. Meanwhile, the framework updates the label correlation graph using the graph transformer layer to accurately identify label node pairs with potential correlation, which effectively mines the correlation of multiple labels to generate more accurate label co-occurrence embedding for image label prediction. Experimental results on UCM, AID, and DFC15 multilabel datasets show that the proposed method outperforms existing state-of-the-art methods.
What problem does this paper attempt to address?