Underwater Image Feature Extraction and Classification Using a Multi-Scale Vision Transformer with Cross-Scale Biased Attention Fusion
Underwater image analysis is affected by light scattering, wavelength-dependent attenuation, low contrast, and suspended particles, which reduce the discriminative visual features. Current multi-scale Vision Transformers are not well-suited to these degradations because they cannot effectively fuse features across scales to achieve accurate classification. Although Vision Transformers (ViTs) can model long-range interactions, single-scale patch tokenization remains suboptimal for underwater images, where both fine-grained textures and global structures are important. This study proposes a Multi-Scale Vision Transformer (MS-ViT) with Cross-Scale Biased Attention Fusion (CSBAF) for underwater image classification. Before transformer encoding, the CSBAF introduces a learnable source–target scale-pair bias and an input-dependent scale-reliability gate. This differs from standard multi-scale fusion and cross-attention methods, which mainly concatenate features or exchange information between scale branches. The proposed design enables the model to emphasize reliable scales while suppressing degraded-scale responses. A hybrid dataset containing 14,000 images from the Roboflow Aquarium and RUIE datasets across five classes was used for evaluation. MS-ViT with CSBAF achieved 88.9% accuracy and an 88.8% F1-score, outperforming the CNN baseline by 7.6% and state-of-the-art transformer models, including UWFormer, DP-ViT, and CvT, by 2.3–4.2%. Ablation studies showed a 1.7% accuracy improvement over simple multi-scale concatenation, whereas cross-dataset testing achieved 84.4% accuracy, indicating reasonable cross-dataset robustness. These results demonstrate that explicit scale–aware fusion can improve transformer-based underwater visual understanding.
Authors
- Abdullah Faiz
- Ping Chen (ORCID: https://orcid.org/0000-0003-0684-9665)
- Kun Li (ORCID: https://orcid.org/0000-0002-4439-7586)
Institutions
- North University of China (CN)
Publication Details
- Journal
- Applied Sciences
- Published
- 2026-05-27
- DOI
- https://doi.org/10.3390/app16115358
- Primary Topic
- Image Enhancement Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- China Postdoctoral Science Foundation
- Shanxi Provincial Key Research and Development Project