A Framework for Imbalanced SAR Ship Classification: Curriculum Learning, Weighted Loss Functions, and a Novel Evaluation Metric
Accurate ship classification is essential for maritime traffic monitoring applications but is significantly hindered by imbalanced datasets. In this paper, we propose a novel methodology that combines curriculum learning with weighted loss functions to address class imbalance in the FUSAR-Ship dataset, facilitating the accurate classification of its nine classes. Our method achieved notable improvements, including a 6.53% average increase in F1-scores compared to baseline models, and successfully identified all classes, including previously misclassified ones. To better evaluate model performance on long-tailed datasets, we introduce a novel evaluation metric that provides a more nuanced assessment of classification ability across underrepresented classes. While demonstrated on the FUSAR-Ship dataset, our approach and metric are broadly applicable to other imbalanced classification problems.
Authors
- Ch Muhammad Awais
- Davide Moroni (ORCID: https://orcid.org/0000-0002-5175-5126)
- Marco Reggiannini (ORCID: https://orcid.org/0000-0002-4872-9541)
Institutions
- University of Pisa (IT)
- Istituto di Scienza e Tecnologie dell'Informazione "Alessandro Faedo" (IT)
Publication Details
- Published
- 2025-02-28
- DOI
- https://doi.org/10.1109/wacvw65960.2025.00171
- Citations
- 1
- Primary Topic
- Maritime Navigation and Safety
- Type
- article
- Field-Weighted Citation Impact
- 1.15