Comparative Analysis of Machine Learning, Deep Learning, and Transformer Models for Boulder Classification: Insights from a Novel Balanced and Imbalanced Sonar Image DatasetShow others and affiliations
2026 (English)In: 2026 2nd International Conference on Federated Learning and Intelligent Computing Systems, FLICS 2026 / [ed] Alawadi S., Giampaolo F., Institute of Electrical and Electronics Engineers (IEEE), 2026, p. 450-456Conference paper, Published paper (Refereed)
Abstract [en]
Accurate classification of underwater boulders is essential for a range of marine applications. Manual classificationmapproaches are labor-intensive, prone to subjectivity, and unsuitable for large-scale survey applications. In this study, we introduce a new benchmark dataset, SonarBoulder, constructed from high-resolution sonar imagery collected off the coast of Sweden. The dataset includes both balanced and imbalanced versions to evaluate model performance. We conducted a comparative analysis of seventeen classification models, including traditional machine learning models (e.g., Random Forest, KNN, SVM, Logistic Regression, AdaBoost, Naive Bayes, ANN, and XGBoost), deep learning methods (e.g., CNN, VGG16, ResNet50, ResNet101, EfficientNetB0, and InceptionV3), and transformer-based frameworks (e.g., DeepViT and ViT). On the balanced SonarBoulder dataset, the deep learning model ResNet101 achieved the best overall performance with 97.75% accuracy, 97.92% precision, and 97.66% F1 score, while the transformer model DeepViT achieved the highest recall at 98.59%. In the imbalanced scenario, InceptionV3 achieved the highest precision (97.26%), recall (95.83%), and F1 score (96.54%) as compared to the other methods. Our results demonstrate the effectiveness of using deep learning and transformer-based models for underwater boulder classification.
Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2026. p. 450-456
Keywords [en]
boulder, classification, deep learning, side-scan sonar data, underwater imaging, Barium compounds, Decision trees, Image classification, Learning systems, Logistic regression, Marine applications, Random forests, Rhenium compounds, Sonar, Underwater acoustics, Boulde, Comparative analyzes, F1 scores, Learning models, Machine-learning, Side scan sonar, Sonar data, Transformer modeling, Classification (of information)
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:bth-30449DOI: 10.1109/FLICS70075.2026.11621933Scopus ID: 2-s2.0-105046975238ISBN: 9798331563356 (print)OAI: oai:DiVA.org:bth-30449DiVA, id: diva2:2096310
Conference
2nd International Conference on Federated Learning and Intelligent Computing Systems, FLICS 2026, Valencia, June 9-12, 2026
2026-08-282026-08-282026-09-02Bibliographically approved