skip to main content
10.1145/3502827.3502836acmotherconferencesArticle/Chapter ViewAbstractPublication PagesicaipConference Proceedingsconference-collections
research-article

Attention Augmented Convolutional Neural Network for Fine-Grained Plant Disease Classification and Visualization Using Stochastic Sample Transformations

Authors Info & Claims
Published:27 January 2022Publication History

ABSTRACT

Deep convolutional neural network (DCNN) image analysis improved plant health management. DCNN methods require train samples with diverse feature distribution and discriminative features for the models to generalize well on unseen categories. The limited train sample coined with the locality nature of the convolution operator and commonly shared kernel weights led CNN models to a sub-optimal solution. To address these challenges, we trained a multi-head self-attention augmented CNN (AAC) and explored other attention augmented networks to jointly exploit global salient and local shared features for fine-grained plant disease classification and visualization. We further proposed stochastic train sample transformations to enlarge and improve train sample diversity and feature distribution. The experiment evaluations on the PlantVillage dataset confirmed that the attention augmentation methods demonstrated robust performance in disease classification and visualization. Specifically, the AAC network scored 99.56% with stochastic transformed train samples, 98.01% with simple geometric transformed samples, and 97.34% without transformation. Therefore, attention augmentation in general, AAC in particular, and stochastic sample transformation can address discriminative plant disease classification and visualization.

References

  1. 1. Tagel Aboneh, Abebe Rorissa, Ramasamy Srinivasagan, and Ashenafi Gemechu. 2021. Computer Vision Framework for Wheat Disease Identification and Classification Using Jetson GPU Infrastructure. Technologies 9, 3: 47. https://doi.org/10.3390/technologies9030047Google ScholarGoogle ScholarCross RefCross Ref
  2. 2. Marko Arsenovic, Mirjana Karanovic, Srdjan Sladojevic, Andras Anderla, and Darko Stefanovic. 2019. Solving Current Limitations of Deep Learning Based Approaches for Plant Disease Detection. Symmetry 11, 7: 939. https://doi.org/10.3390/sym11070939Google ScholarGoogle ScholarCross RefCross Ref
  3. 3. Irwan Bello, Barret Zoph, Quoc Le, Ashish Vaswani, and Jonathon Shlens. 2019. Attention augmented convolutional networks. In Proceedings of the IEEE International Conference on Computer Vision, 3285–3294. https://doi.org/10.1109/ICCV.2019.00338Google ScholarGoogle ScholarCross RefCross Ref
  4. 4. Quan Huu Cap, Hiroyuki Uga, Satoshi Kagiwada, and Hitoshi Iyatomi. 2020. LeafGAN: An Effective Data Augmentation Method for Practical Plant Disease Diagnosis. IEEE Transactions on Automation Science and Engineering: 1–10.Google ScholarGoogle ScholarCross RefCross Ref
  5. 5. Junde Chen, Defu Zhang, Md Suzauddola, Yaser Ahangari Nanehkaran, and Yuandong Sun. 2021. Identification of plant disease images via a squeeze-and-excitation MobileNet model and twice transfer learning. IET Image Processing 15, 5: 1115–1127.Google ScholarGoogle ScholarCross RefCross Ref
  6. 6. Ting Chen, Simon Kornblith, Mohammad Norouzi, and Geoffrey Hinton. 2020. A simple framework for contrastive learning of visual representations. In 37th International Conference on Machine Learning, ICML 2020, 1575–1585.Google ScholarGoogle ScholarDigital LibraryDigital Library
  7. 7. Halil Durmus, Ece Olcay Gunes, and Murvet Kirci. 2017. Disease detection on the leaves of the tomato plants by using deep learning. In 2017 6th International Conference on Agro-Geoinformatics, Agro-Geoinformatics 2017, 1–5. https://doi.org/10.1109/Agro-Geoinformatics.2017.8047016Google ScholarGoogle ScholarCross RefCross Ref
  8. 8. Kumie Gedamu, Yanli Ji, Yang Yang, Ling Ling Gao, and Heng Tao Shen. 2021. Arbitrary-view human action recognition via novel-view action generation. Pattern Recognition 118, 2021: 108043. https://doi.org/10.1016/j.patcog.2021.108043Google ScholarGoogle Scholar
  9. 9. Jie Hu, Li Shen, and Gang Sun. 2018. Squeeze-and-Excitation Networks. In Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, 7132–7141. https://doi.org/10.1109/CVPR.2018.00745Google ScholarGoogle Scholar
  10. 10. Gilbert Gutabaga Hungilo, Gahizi Emmanuel, and Andi W.R. Emanuel. 2019. Image processing techniques for detecting and classification of plant disease – A review. ACM International Conference Proceeding Series: 48–52. https://doi.org/10.1145/3332340.3332341Google ScholarGoogle ScholarDigital LibraryDigital Library
  11. 11. DIng Jiang, Fudong Li, Yuequan Yang, and Song Yu. 2020. A Tomato Leaf Diseases Classification Method Based on Deep Learning. In Proceedings of the 32nd Chinese Control and Decision Conference, CCDC 2020, 1446–1450. https://doi.org/10.1109/CCDC49329.2020.9164457Google ScholarGoogle ScholarCross RefCross Ref
  12. 12. R. Karthik, M. Hariharan, Sundar Anand, Priyanka Mathikshara, Annie Johnson, and R Menaka. 2020. Attention embedded residual CNN for disease detection in tomato leaves. Applied Soft Computing Journal 86. https://doi.org/10.1016/j.asoc.2019.105933Google ScholarGoogle ScholarDigital LibraryDigital Library
  13. 13. Sharada P. Mohanty, David P. Hughes, and Marcel Salathé. 2016. Using Deep Learning for Image-Based Plant Disease Detection. Frontiers in Plant Science 7, September: 1–10. https://doi.org/10.3389/fpls.2016.01419Google ScholarGoogle Scholar
  14. 14. Aravind Krishnaswamy Rangarajan, Raja Purushothaman, and Aniirudh Ramesh. 2018. Tomato crop disease classification using pre-trained deep learning algorithm. In Procedia Computer Science, 1040–1047. https://doi.org/10.1016/j.procs.2018.07.070Google ScholarGoogle Scholar
  15. 15. Davinder Singh, Naman Jain, Pranjali Jain, Pratik Kayal, Sudhakar Kumawat, and Nipun Batra. 2020. PlantDoc: A dataset for visual plant disease detection. ACM International Conference Proceeding Series: 249–253. https://doi.org/10.1145/3371158.3371196Google ScholarGoogle ScholarDigital LibraryDigital Library
  16. 16. Sanghyun Woo, Jongchan Park, Joon-Young Lee, and In So Kweon. 2018. CBAM: Convolutional Block Attention Module. In Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), 3–19. https://doi.org/10.1007/978-3-030-01234-2_1Google ScholarGoogle ScholarDigital LibraryDigital Library
  17. 17. Getinet Yilma, Seid Belay, Kumie Gedamu, Maregu Assefa, and Melese Ayalew. 2021. Attention augmented residual network for tomato disease detection and classification. TURKISH JOURNAL OF ELECTRICAL ENGINEERING & COMPUTER SCIENCES 29, 2021/SI–1: 2869–2885. https://doi.org/10.3906/elk-2105-115Google ScholarGoogle ScholarCross RefCross Ref
  18. 18. Getinet Yilma, Seid Belay, Zhiguang Qin, Kumie Gedamu, and Melese Ayalew. 2020. Plant Disease Classification Using Two Pathway Encoder GAN Data Generation. In 2020 17th International Computer Conference on Wavelet Active Media Technology and Information Processing (ICCWAMTIP), 67–72. https://doi.org/10.1109/ICCWAMTIP51612.2020.9317494Google ScholarGoogle Scholar
  19. 19. Getinet Yilma, Kumie Gedamu, Maregu Assefa, Ariyo Oluwasanmi, and Zhiguang Qin. 2021. Generation and Transformation Invariant Learning for Tomato Disease Classification. In 2021 IEEE 2nd International Conference on Pattern Recognition and Machine Learning (PRML), 121–128. https://doi.org/10.1109/prml52754.2021.9520693Google ScholarGoogle Scholar
  20. 20. Hengshuang Zhao, Jiaya Jia, and Vladlen Koltun. 2020. Exploring Self-attention for Image Recognition. In Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, 10073–10082. https://doi.org/10.1109/CVPR42600.2020.01009Google ScholarGoogle ScholarCross RefCross Ref

Index Terms

  1. Attention Augmented Convolutional Neural Network for Fine-Grained Plant Disease Classification and Visualization Using Stochastic Sample Transformations
          Index terms have been assigned to the content through auto-classification.

          Recommendations

          Comments

          Login options

          Check if you have access through your login credentials or your institution to get full access on this article.

          Sign in
          • Published in

            cover image ACM Other conferences
            ICAIP '21: Proceedings of the 5th International Conference on Advances in Image Processing
            November 2021
            112 pages
            ISBN:9781450385183
            DOI:10.1145/3502827

            Copyright © 2021 ACM

            Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than ACM must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]

            Publisher

            Association for Computing Machinery

            New York, NY, United States

            Publication History

            • Published: 27 January 2022

            Permissions

            Request permissions about this article.

            Request Permissions

            Check for updates

            Qualifiers

            • research-article
            • Research
            • Refereed limited

          PDF Format

          View or Download as a PDF file.

          PDF

          eReader

          View online with eReader.

          eReader

          HTML Format

          View this article in HTML Format .

          View HTML Format