An augmented attention-based lightweight CNN model for plant water stress detection
Recently, deep learning techniques specifically the Convolutional Neural Networks (CNNs) have reported outstanding results from the application for plant water stress detection based on computer vision system compared to other machine learning methods. However, the size of the conventional CNN model...
Main Authors: | , , , |
---|---|
Format: | Article |
Published: |
Springer
2023
|
Subjects: |
_version_ | 1811132101969313792 |
---|---|
author | Kamarudin, Mohd. Hider Ismail, Zool Hilmi Saidi, Noor Baity Hanada, Kousuke |
author_facet | Kamarudin, Mohd. Hider Ismail, Zool Hilmi Saidi, Noor Baity Hanada, Kousuke |
author_sort | Kamarudin, Mohd. Hider |
collection | ePrints |
description | Recently, deep learning techniques specifically the Convolutional Neural Networks (CNNs) have reported outstanding results from the application for plant water stress detection based on computer vision system compared to other machine learning methods. However, the size of the conventional CNN models is generally too large for its deployment on resource-limited devices such as mobile smartphone or embedded devices. In this study, a lightweight CNN is proposed by incorporating attention mechanism as an augmentation module into the model. The model was trained, validated, and tested using plant images of Setaria grass undergone three water stress treatments. Experimental results show that the proposed method improved the interclass precision, recall, F1-score, and the overall accuracy by more than 9%. Compared to the established lightweight CNN models, the proposed lightweight CNN achieved faster computational time with comparable parameters. In addition, the proposed lightweight model is also efficient when trained on small plant dataset with limited overfitting. |
first_indexed | 2024-04-09T03:46:01Z |
format | Article |
id | utm.eprints-105058 |
institution | Universiti Teknologi Malaysia - ePrints |
last_indexed | 2024-09-23T23:56:58Z |
publishDate | 2023 |
publisher | Springer |
record_format | dspace |
spelling | utm.eprints-1050582024-06-30T00:44:08Z http://eprints.utm.my/105058/ An augmented attention-based lightweight CNN model for plant water stress detection Kamarudin, Mohd. Hider Ismail, Zool Hilmi Saidi, Noor Baity Hanada, Kousuke T Technology (General) Recently, deep learning techniques specifically the Convolutional Neural Networks (CNNs) have reported outstanding results from the application for plant water stress detection based on computer vision system compared to other machine learning methods. However, the size of the conventional CNN models is generally too large for its deployment on resource-limited devices such as mobile smartphone or embedded devices. In this study, a lightweight CNN is proposed by incorporating attention mechanism as an augmentation module into the model. The model was trained, validated, and tested using plant images of Setaria grass undergone three water stress treatments. Experimental results show that the proposed method improved the interclass precision, recall, F1-score, and the overall accuracy by more than 9%. Compared to the established lightweight CNN models, the proposed lightweight CNN achieved faster computational time with comparable parameters. In addition, the proposed lightweight model is also efficient when trained on small plant dataset with limited overfitting. Springer 2023 Article PeerReviewed Kamarudin, Mohd. Hider and Ismail, Zool Hilmi and Saidi, Noor Baity and Hanada, Kousuke (2023) An augmented attention-based lightweight CNN model for plant water stress detection. Applied Intelligence, 53 (18). pp. 20828-20843. ISSN 0924-669X http://dx.doi.org/10.1007/s10489-023-04583-8 DOI : 10.1007/s10489-023-04583-8 |
spellingShingle | T Technology (General) Kamarudin, Mohd. Hider Ismail, Zool Hilmi Saidi, Noor Baity Hanada, Kousuke An augmented attention-based lightweight CNN model for plant water stress detection |
title | An augmented attention-based lightweight CNN model for plant water stress detection |
title_full | An augmented attention-based lightweight CNN model for plant water stress detection |
title_fullStr | An augmented attention-based lightweight CNN model for plant water stress detection |
title_full_unstemmed | An augmented attention-based lightweight CNN model for plant water stress detection |
title_short | An augmented attention-based lightweight CNN model for plant water stress detection |
title_sort | augmented attention based lightweight cnn model for plant water stress detection |
topic | T Technology (General) |
work_keys_str_mv | AT kamarudinmohdhider anaugmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection AT ismailzoolhilmi anaugmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection AT saidinoorbaity anaugmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection AT hanadakousuke anaugmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection AT kamarudinmohdhider augmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection AT ismailzoolhilmi augmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection AT saidinoorbaity augmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection AT hanadakousuke augmentedattentionbasedlightweightcnnmodelforplantwaterstressdetection |