Enhancing Ship Classification in Optical Satellite Imagery: Integrating Convolutional Block Attention Module with ResNet for Improved Performance
In this study, we present an advanced convolutional neural network (CNN) architecture for ship classification based on optical satellite imagery, which significantly enhances performance through the integration of a convolutional block attention module (CBAM) and additional architectural innovations...
Gespeichert in:
Hauptverfasser: | , , , , , |
---|---|
Format: | Artikel |
Sprache: | eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | In this study, we present an advanced convolutional neural network (CNN)
architecture for ship classification based on optical satellite imagery, which
significantly enhances performance through the integration of a convolutional
block attention module (CBAM) and additional architectural innovations.
Building upon the foundational ResNet50 model, we first incorporated a standard
CBAM to direct the model's focus toward more informative features, achieving an
accuracy of 87% compared to 85% of the baseline ResNet50. Further augmentations
involved multiscale feature integration, depthwise separable convolutions, and
dilated convolutions, culminating in an enhanced ResNet model with improved
CBAM. This model demonstrated a remarkable accuracy of 95%, with precision,
recall, and F1 scores all witnessing substantial improvements across various
ship classes. In particular, the bulk carrier and oil tanker classes exhibited
nearly perfect precision and recall rates, underscoring the enhanced capability
of the model to accurately identify and classify ships. Attention heatmap
analyses further validated the efficacy of the improved model, revealing more
focused attention on relevant ship features regardless of background
complexities. These findings underscore the potential of integrating attention
mechanisms and architectural innovations into CNNs for high-resolution
satellite imagery classification. This study navigates through the class
imbalance and computational costs and proposes future directions for
scalability and adaptability in new or rare ship-type recognition. This study
lays the groundwork for applying advanced deep learning techniques in remote
sensing, offering insights into scalable and efficient satellite image
classification. |
---|---|
DOI: | 10.48550/arxiv.2404.02135 |