Effective Data Augmentation for Active Sonar Classification Using Attention-Based Complementary Learning With Uncertainty Measure

Youngsang Hwang

doi:10.1109/ACCESS.2025.3555420

Effective Data Augmentation for Active Sonar Classification Using Attention-Based Complementary Learning With Uncertainty Measure

2025-01-01

0

OA

AI

DOI：10.1109/ACCESS.2025.3555420

Original

Original request for help

Save

Abstract

En 中文

We enhance the performance of active sonar classification by integrating a validated attention-based complementary learning model (ABCL) with an active learning (AL) framework, following modifications to adapt the model for AL. Acoustic data from active sonar systems strongly depend on ocean environmental conditions and deep learning (DL) models trained on limited active sonar data often demonstrate poor performance when tested on data from new environments. To enhance performance, a small number of test data samples with high uncertainties are selected to augment the existing training set. A deep ensemble approach is employed to measure the uncertainty of each sample, quantified as the variance of predictions from models with independently optimized weights. ABCL, initially designed for robust generalization with limited active sonar data, was modified to produce two predictions per sample. This modification enhances the reliability of uncertainty measurement and is referred to as deeper ABCL (DABCL). Two datasets from different experimental conditions serve as the training and test sets. Most high-uncertainty test samples identified through AL are found in regions with a mix of target and non-target instances, allowing DABCL to adapt effectively to the shifted test data distribution. This approach achieves superior performance compared to other DL models, including VGG16, ResNet18, and Swin Transformer, both before and after applying AL. Although mislabeling occurs in 20 percent of the uncertainty samples during data augmentation, the fine-tuned DABCL still outperforms the version without AL.

Keywords:

Sonar

Data models

Uncertainty

Training

Sonar measurements

Oceans

Sea measurements

Measurement uncertainty

Predictive models

Data augmentation

Active sonar classification

active learning

uncertainty measure

data augmentation

Effective Data Augmentation for Active Sonar Classification Using Attention-Based Complementary Learning With Uncertainty Measure

Abstract

Journal

Organization