arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.02135cs.CVeess.IV

增强光学卫星图像中的船舶分类:将卷积块注意力模块与ResNet集成以提升性能

Enhancing Ship Classification in Optical Satellite Imagery: Integrating Convolutional Block Attention Module with ResNet for Improved Performance

Sungkyunkwan University School of Medicine(成均馆大学医学院);The Coala Inc.(考拉公司);Korea Science Academy of Korea Advanced Institute of Science and Technology(韩国科学技术院韩国科学学院);Seoul National University School of Earth and Environmental Sciences(首尔国立大学地球与环境科学学院)

Ryan Donghan Kwon, Gangjoo Robin Nam, Jisoo Tak, Junseob Shin, Hyerin Cha, Seung Won Lee

AI总结:

本研究通过集成CBAM及多尺度、深度可分离和空洞卷积等创新,改进ResNet50模型,在光学卫星图像船舶分类中达到95%准确率,显著提升各类船舶识别性能。

摘要:

在本研究中,我们提出了一种基于光学卫星图像的先进卷积神经网络(CNN)架构用于船舶分类,通过集成卷积块注意力模块(CBAM)和额外的架构创新,显著提升了性能。在基础ResNet50模型之上,我们首先加入了标准CBAM以引导模型关注更具信息量的特征,达到了87%的准确率,而基线ResNet50的准确率为85%。进一步的增强涉及多尺度特征集成、深度可分离卷积和空洞卷积,最终形成了一个带有改进CBAM的增强ResNet模型。该模型展现了95%的显著准确率,且各类船舶的精确率、召回率和F1分数均大幅提升。特别是,散货船和油轮类别的精确率和召回率几乎达到完美,凸显了模型准确识别和分类船舶能力的增强。注意力热图分析进一步验证了改进模型的有效性,显示无论背景复杂性如何,模型都能更集中地关注相关船舶特征。这些发现强调了将注意力机制和架构创新集成到CNN中用于高分辨率卫星图像分类的潜力。本研究解决了类别不平衡和计算成本问题,并提出了在新颖或稀有船舶类型识别中实现可扩展性和适应性的未来方向。本研究为在遥感中应用先进深度学习技术奠定了基础,为可扩展且高效的卫星图像分类提供了见解。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑