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...

Full description

Bibliographic Details
Main Authors: Kamarudin, Mohd. Hider, Ismail, Zool Hilmi, Saidi, Noor Baity, Hanada, Kousuke
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