Evaluation of model generalization for growing plants using conditional learning
This paper aims to solve the lack of generalization of existing semantic segmentation models in the crop and weed segmentation domain. We compare two training mechanisms, classical and adversarial, to understand which scheme works best for a particular encoder-decoder model. We use simple U-Net, Seg...
Main Authors: | , |
---|---|
Format: | Article |
Language: | English |
Published: |
KeAi Communications Co., Ltd.
2022-01-01
|
Series: | Artificial Intelligence in Agriculture |
Subjects: | |
Online Access: | http://www.sciencedirect.com/science/article/pii/S2589721722000162 |
_version_ | 1797980219382956032 |
---|---|
author | Hafiz Sami Ullah Abdul Bais |
author_facet | Hafiz Sami Ullah Abdul Bais |
author_sort | Hafiz Sami Ullah |
collection | DOAJ |
description | This paper aims to solve the lack of generalization of existing semantic segmentation models in the crop and weed segmentation domain. We compare two training mechanisms, classical and adversarial, to understand which scheme works best for a particular encoder-decoder model. We use simple U-Net, SegNet, and DeepLabv3+ with ResNet-50 backbone as segmentation networks. The models are trained with cross-entropy loss for classical and PatchGAN loss for adversarial training. By adopting the Conditional Generative Adversarial Network (CGAN) hierarchical settings, we penalize different Generators (G) using PatchGAN Discriminator (D) and L1 loss to generate segmentation output. The generalization is to exhibit fewer failures and perform comparably for growing plants with different data distributions. We utilize the images from four different stages of sugar beet. We divide the data so that the full-grown stage is used for training, whereas earlier stages are entirely dedicated to testing the model. We conclude that U-Net trained in adversarial settings is more robust to changes in the dataset. The adversarially trained U-Net reports 10% overall improvement in the results with mIOU scores of 0.34, 0.55, 0.75, and 0.85 for four different growth stages. |
first_indexed | 2024-04-11T05:52:11Z |
format | Article |
id | doaj.art-52a2776bdd0b422cba684d160314264a |
institution | Directory Open Access Journal |
issn | 2589-7217 |
language | English |
last_indexed | 2024-04-11T05:52:11Z |
publishDate | 2022-01-01 |
publisher | KeAi Communications Co., Ltd. |
record_format | Article |
series | Artificial Intelligence in Agriculture |
spelling | doaj.art-52a2776bdd0b422cba684d160314264a2022-12-22T04:42:03ZengKeAi Communications Co., Ltd.Artificial Intelligence in Agriculture2589-72172022-01-016189198Evaluation of model generalization for growing plants using conditional learningHafiz Sami Ullah0Abdul Bais1Electronic Systems Engineering, Faculty of Engineering and Applied Science, University of Regina, Regina, Saskatchewan, CanadaCorresponding author.; Electronic Systems Engineering, Faculty of Engineering and Applied Science, University of Regina, Regina, Saskatchewan, CanadaThis paper aims to solve the lack of generalization of existing semantic segmentation models in the crop and weed segmentation domain. We compare two training mechanisms, classical and adversarial, to understand which scheme works best for a particular encoder-decoder model. We use simple U-Net, SegNet, and DeepLabv3+ with ResNet-50 backbone as segmentation networks. The models are trained with cross-entropy loss for classical and PatchGAN loss for adversarial training. By adopting the Conditional Generative Adversarial Network (CGAN) hierarchical settings, we penalize different Generators (G) using PatchGAN Discriminator (D) and L1 loss to generate segmentation output. The generalization is to exhibit fewer failures and perform comparably for growing plants with different data distributions. We utilize the images from four different stages of sugar beet. We divide the data so that the full-grown stage is used for training, whereas earlier stages are entirely dedicated to testing the model. We conclude that U-Net trained in adversarial settings is more robust to changes in the dataset. The adversarially trained U-Net reports 10% overall improvement in the results with mIOU scores of 0.34, 0.55, 0.75, and 0.85 for four different growth stages.http://www.sciencedirect.com/science/article/pii/S2589721722000162Weed detectionSemantic segmentationAdversarial trainingLate germinationSugar beetCrop segmentation |
spellingShingle | Hafiz Sami Ullah Abdul Bais Evaluation of model generalization for growing plants using conditional learning Artificial Intelligence in Agriculture Weed detection Semantic segmentation Adversarial training Late germination Sugar beet Crop segmentation |
title | Evaluation of model generalization for growing plants using conditional learning |
title_full | Evaluation of model generalization for growing plants using conditional learning |
title_fullStr | Evaluation of model generalization for growing plants using conditional learning |
title_full_unstemmed | Evaluation of model generalization for growing plants using conditional learning |
title_short | Evaluation of model generalization for growing plants using conditional learning |
title_sort | evaluation of model generalization for growing plants using conditional learning |
topic | Weed detection Semantic segmentation Adversarial training Late germination Sugar beet Crop segmentation |
url | http://www.sciencedirect.com/science/article/pii/S2589721722000162 |
work_keys_str_mv | AT hafizsamiullah evaluationofmodelgeneralizationforgrowingplantsusingconditionallearning AT abdulbais evaluationofmodelgeneralizationforgrowingplantsusingconditionallearning |