arrow
返回

AdaAug+: A Reinforcement Learning-Based Adaptive Data Augmentation for Change Detection

delete2024-01-01
delete0
PRE
AI
R
Rui Huang
J
Jieda Wei
Y
Yan Xing *
Q
Qing Guo
DOI:10.1109/TGRS.2024.3478218delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Data augmentation (DA) increases the diversity of training data to improve the model generalization ability. Most of the DA methods focus on image classification or object detection. Directly using the existing DA methods on change detection tasks not only falls short of fully exploring the specificity of the change image pairs but also leads to longer training times. In this article, we first propose a mask-guided mixing (MGM) DA for change detection, which mixes the change regions of the current training sample based on prediction results and labels to generate high-quality samples with more positive samples. We then propose a new reinforcement learning (RL)-based Adaptive DA method, AdaAug(+), to adaptively select the optimal DA policy for the training samples. An actor selects the best augmentation operation from the operation set according to the image pair. The augmented image pairs make it easier for the change detector to learn the optimal parameters and improve the final detection performance. To reduce the training time, we identify and remove the redundant training samples during the training process by our redundancy searching policy. We have conducted various experiments on four remote sensing change detection datasets with different change detectors. The experimental results demonstrate that AdaAug+ achieves promising performance compared to the state-of-the-art DA methods and requires less training time.
Keyword:
Data augmentation
Training
Detectors
Feature extraction
Data models
Transformers
Training data
Object detection
Image classification
Deep learning
Change detection
data augmentation (DA)
reinforcement learning (RL)

期刊

IEEE Transactions on Geoscience and Remote Sensing 封面图
IEEE Transactions on Geoscience and Remote Sensing
IF:
8.6
论文数:
2.1W
被引数:
10.7W

机构

C
Civil Aviation University of China
学者数:
3.0K
论文数: 1.9K
被引数: 1.5K
A
agency for science technology & research (a*star)
学者数:
2.2W
论文数: 1.9W
被引数: 57
引用论文

引用论文

err分享
err收藏
Local Patch AutoAugment With Multi-Agent Collaboration
err2024-01-01
err1
errOAAI
errLin, Shiqi; Yu, Tao; Feng, Ruoyu; Li, Xin; Yu, Xiaoyuan; Xiao, Lei; Chen, Zhibo
err分享
err收藏
Stability of adaptive behaviors in middle-school children with autism spectrum disorders
err2007-10-01
err0
PREAI
errRobin L. Gabriels; Bonnie Jean Ivers; Dina E. Hill; John A. Agnew; John McNeill
err分享
err收藏
学者 查看更多内容