Title:

Enhancing Leaf Area Segmentation by Using Attention Gates and Knowledge Distillation in UNet Architecture, Journal of Telecommunications and Information Technology, 2025, nr 3

Group publication title:

2025, nr 3, JTIT-artykuły

Creator:

Shamim Banu, A. ; Deivalakshmi, S.

Subject and Keywords:

attention gate ; knowledge distillation ; modified light weight UNet ; semantic segmentation

Description:

kwartalnik

Abstrakt:

Accurate segmentation of leaf regions plays a vital role in plant phenotyping and agricultural analysis. This paper presents AKDUNet, a lightweight UNet-based architecture that integrates attention gates and knowledge distillation to improve segmentation performance while minimizing computational complexity. The architecture replaces traditional skip connections with attention gates to focus on salient spatial features and employs a two-stage training pipeline, where a compact student model learns from a deeper teacher model using a tailored distillation loss function. AKDUNet is evaluated on two benchmark datasets (CWFID and Sunflower) and outperforms a range of state-of-the-art models, including UNet++, Inception UNet, VGG-based UNets, SDUNet, INSCA UNet, and SegFormer. Ablation studies confirm the advantages of attention modules, and qualitative analyses using Grad-CAM visualizations reveal the model's ability to effectively focus on crucial leaf structures. The results demonstrate that AKDUNet is not only computationally efficient but also highly accurate, making it suitable for real-time deployment in resource-constrained agricultural environments.

Number:

3

Publisher:

National Institute of Telecommunications

Date:

2025, nr 3

Resource Type:

artykuł

DOI:

10.26636/jtit.2025.3.2079

eISSN:

on-line: ISSN 1899-8852

Source:

Journal of Telecommunications and Information Technology

Language:

ang

Rights Management:

Biblioteka Naukowa Instytutu Łączności

License:

CC BY 4.0