Back to Search Start Over

Interpretability-Mask: a label-preserving data augmentation scheme for better classification.

Authors :
Zhao, Hao
Wang, Jikai
Chen, Zonghai
Lin, Shiqi
Bao, Peng
Xu, Meng
Source :
Signal, Image & Video Processing; Sep2023, Vol. 17 Issue 6, p2799-2808, 10p
Publication Year :
2023

Abstract

Data augmentation effectively alleviates the over-fitting problem in convolutional neural network-based (CNN-based) models, especially in the limited dataset. However, the inconsistency problem between the augmented sample and its original label is still a critical challenge during the augmentation operation. In this paper, we propose a novel data augmentation scheme named Interpretability-Mask (IM), which exploits the interpretability of the classifier to obtain the most discriminative regions and preserve label invariance. Concretely, we first construct a set-based representation for a sample and its label by superpixel segmentation and the local interpretable model-agnostic explanations (LIME) operator. Secondly, the sample represented by the superpixel set is utilized to synthesize the region-level disturbance augmentation sample with a random removal strategy. Then, the label constructed by the most interpretive superpixel set is applied to maintain the consistency between the augmented sample and its original label. Lastly, the augmentation scheme will be randomly used to the training stage. Extensive experiments are conducted on challenging datasets. A significant improvement in classification performance has achieved with the IM scheme. On the CIFAR-10 dataset, the Top-1 error rate drops by 2.15% at most. On the CIFAR-100 dataset, the Top-1 error rate decreases by up to 3.69%. And the maximum decline of the Top-1 error rate is 3.35% on the Mini-ImageNet. Experimental results manifest the effectiveness and generality of the proposed method. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
18631703
Volume :
17
Issue :
6
Database :
Complementary Index
Journal :
Signal, Image & Video Processing
Publication Type :
Academic Journal
Accession number :
164372022
Full Text :
https://doi.org/10.1007/s11760-023-02497-z