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.

A framework for imbalanced SAR ship classification: curriculum learning, weighted loss functions, and a novel evaluation metric

Awais Ch Muhammad;Reggiannini M.;Moroni D.
2025

Abstract

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.
2025
Istituto di Scienza e Tecnologie dell'Informazione "Alessandro Faedo" - ISTI
979-8-3315-3662-6
SAR ship classification, Deep learning, Imbalanced dataset, Loss function, Evaluation metrics
File in questo prodotto:
File Dimensione Formato  
A_Framework_for_Imbalanced_SAR_Ship_Classification_Curriculum_Learning_Weighted_Loss_Functions_and_a_Novel_Evaluation_Metric.pdf

solo utenti autorizzati

Tipologia: Versione Editoriale (PDF)
Licenza: NON PUBBLICO - Accesso privato/ristretto
Dimensione 633.77 kB
Formato Adobe PDF
633.77 kB Adobe PDF   Visualizza/Apri   Richiedi una copia

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/20.500.14243/544001
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus 1
  • ???jsp.display-item.citation.isi??? ND
social impact