A Label Propagation Strategy for CutMix in Multi-Label Remote Sensing Image Classification

TL;DR

This paper introduces a label propagation strategy to improve CutMix data augmentation for multi-label remote sensing image classification, effectively handling label noise and enhancing model performance.

Contribution

It proposes a novel label propagation method that leverages pixel-level class positional information to adapt CutMix for multi-label remote sensing tasks, reducing label noise issues.

Findings

Improves mAP macro by 2-4% over standard CutMix.

Demonstrates robustness with noisy class positional data.

Effective in both simulated and real scenarios.

Abstract

The development of supervised deep learning-based methods for multi-label scene classification (MLC) is one of the prominent research directions in remote sensing (RS). However, collecting annotations for large RS image archives is time-consuming and costly. To address this issue, several data augmentation methods have been introduced in RS. Among others, the CutMix data augmentation technique, which combines parts of two existing training images to generate an augmented image, stands out as a particularly effective approach. However, the direct application of CutMix in RS MLC can lead to the erasure or addition of class labels (i.e., label noise) in the augmented (i.e., combined) training image. To address this problem, we introduce a label propagation (LP) strategy that allows the effective application of CutMix in the context of MLC problems in RS without being affected by label noise. To this end, our proposed LP strategy exploits pixel-level class positional information to update the multi-label of the augmented training image. We propose to access such class positional information from reference maps (e.g., thematic products) associated with each training image or from class explanation masks provided by an explanation method if no reference maps are available. Similarly to pairing two training images, our LP strategy carries out a pairing operation on the associated pixel-level class positional information to derive the updated multi-label for the augmented image. Experimental results show the effectiveness of our LP strategy in general (e.g., an improvement of 2% to 4% mAP macro compared to standard CutMix) and its robustness in the case of various simulated and real scenarios with noisy class positional information in particular. Code is available at https://git.tu-berlin.de/rsim/cutmix_lp.