arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2506.17455cs.CV

AQUA20:一种用于在挑战性条件下水下物种分类的基准数据集

AQUA20: A Benchmark Dataset for Underwater Species Classification under Challenging Conditions

Islamic University of Technology(伊斯兰理工大学)

Taufikur Rahman Fuad, Sabbir Ahmed, Shahriar Ivan

AI总结:

本文提出AQUA20数据集,包含20种海洋物种的8171张水下图像,评估了13种先进深度学习模型在水下物种分类中的性能,发现ConvNeXt模型在准确率和F1分数上表现最佳。

摘要:

在水下环境中实现稳健的视觉识别仍是一个重大挑战,由于复杂畸变如浑浊、低光照和遮挡,严重降低了标准视觉系统的表现。本文介绍了AQUA20,一个包含8171张水下图像的综合基准数据集,涵盖20种海洋物种,反映了现实环境中的光照、浑浊、遮挡等挑战,为水下视觉理解提供有价值的资源。十三种最先进的深度学习模型,包括轻量级CNN(SqueezeNet、MobileNetV2)和基于Transformer的架构(ViT、ConvNeXt),被评估以基准其在挑战性条件下分类海洋物种的性能。我们的实验结果表明,ConvNeXt在Top-3准确率98.82%、Top-1准确率90.69%以及整体F1分数88.92%方面表现最佳,参数规模适中。其他基准模型的结果也展示了复杂性与性能之间的权衡。我们还通过GRAD-CAM和LIME提供了详尽的可解释性分析,以解释模型的优缺点。我们的结果揭示了水下物种识别仍有显著改进空间,并展示了AQUA20作为未来研究基础的价值。该数据集在https://huggingface.co/datasets/taufiktrf/AQUA20公开可用。

英文摘要:

Robust visual recognition in underwater environments remains a significant challenge due to complex distortions such as turbidity, low illumination, and occlusion, which severely degrade the performance of standard vision systems. This paper introduces AQUA20, a comprehensive benchmark dataset comprising 8,171 underwater images across 20 marine species reflecting real-world environmental challenges such as illumination, turbidity, occlusions, etc., providing a valuable resource for underwater visual understanding. Thirteen state-of-the-art deep learning models, including lightweight CNNs (SqueezeNet, MobileNetV2) and transformer-based architectures (ViT, ConvNeXt), were evaluated to benchmark their performance in classifying marine species under challenging conditions. Our experimental results show ConvNeXt achieving the best performance, with a Top-3 accuracy of 98.82% and a Top-1 accuracy of 90.69%, as well as the highest overall F1-score of 88.92% with moderately large parameter size. The results obtained from our other benchmark models also demonstrate trade-offs between complexity and performance. We also provide an extensive explainability analysis using GRAD-CAM and LIME for interpreting the strengths and pitfalls of the models. Our results reveal substantial room for improvement in underwater species recognition and demonstrate the value of AQUA20 as a foundation for future research in this domain. The dataset is publicly available at: https://huggingface.co/datasets/taufiktrf/AQUA20.

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

↑