Accurate material recognition with low computational overhead is critical for edge applications such as autonomous drones, mobile robots, and smart manufacturing systems. Direct fine-tuning of deep backbones often leads to early saturation in validation accuracy due to overfitting on small, domain-specific datasets. To address this, we propose a structured multi-phase fine-tuning strategy for EfficientNetV2-S, progressively unfreezing layers over four stages with adaptive learning rate scheduling. The approach also incorporates label smoothing, dropout, and data augmentation to enhance generalization. We evaluated the method on a curated dataset of 1,730 images across four material classes: glass, metal, paper, and plastic. The resulting model achieves a validation accuracy of 95.66%, demonstrating that the proposed pipeline effectively balances accuracy and computational efficiency, making it suitable for real-time deployment on resource-constrained edge devices.

Multi-stage fine-tuning of EfficientNetV2-S for material recognition on edge devices

Zese R.
Ultimo
2026

Abstract

Accurate material recognition with low computational overhead is critical for edge applications such as autonomous drones, mobile robots, and smart manufacturing systems. Direct fine-tuning of deep backbones often leads to early saturation in validation accuracy due to overfitting on small, domain-specific datasets. To address this, we propose a structured multi-phase fine-tuning strategy for EfficientNetV2-S, progressively unfreezing layers over four stages with adaptive learning rate scheduling. The approach also incorporates label smoothing, dropout, and data augmentation to enhance generalization. We evaluated the method on a curated dataset of 1,730 images across four material classes: glass, metal, paper, and plastic. The resulting model achieves a validation accuracy of 95.66%, demonstrating that the proposed pipeline effectively balances accuracy and computational efficiency, making it suitable for real-time deployment on resource-constrained edge devices.
2026
Tomic, S.; Aldhuhoori, D.; Turker, T.; Nikolic, J.; Peric, Z.; Zese, R.
File in questo prodotto:
File Dimensione Formato  
Multi-stage_fine-tuning_of_EfficientNetV2-S_for_material_recognition_on_edge_devices.pdf

accesso aperto

Descrizione: Full text editoriale
Tipologia: Full text (versione editoriale)
Licenza: Creative commons
Dimensione 584.59 kB
Formato Adobe PDF
584.59 kB Adobe PDF Visualizza/Apri

I documenti in SFERA 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/11392/2627471
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus 0
  • ???jsp.display-item.citation.isi??? 0
social impact