IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0349210.html

A curated dataset and lightweight deep learning framework for tea leaf disease classification

Author

Listed:
  • Sakibul Hasan Chowdhury
  • Md Shohel Arman
  • Masrafe Bin Hannan Siam
  • Md Rayhan Khan
  • Afia Hasan
  • Parvez Ahmed Moju Fahim

Abstract

Tea (Camellia sinensis) is the world’s second most consumed beverage, enjoyed daily by more than two billion people. In Bangladesh, it serves as a cornerstone agricultural export and a major sector of the domestic economy. However, commercial tea cultivation remains highly vulnerable to fungal and pest-related diseases such as Blight, Red Rust, and Helopeltis which severely reduce crop yield and compromise leaf quality. While early detection is critical to preventing widespread outbreaks, traditional manual inspection is slow, subjective, and highly error-prone. Deep learning provides a scalable alternative, yet single-branch networks often struggle to capture both minute disease lesions and broader structural degradation simultaneously. To address this, we propose a Hybrid Feature Fusion architecture that runs two highly efficient feature extractors in parallel: EfficientNetV2-Small to isolate fine-grained local textures, and MobileNetV3-Small to capture the global structural context of the leaf. The models were trained and evaluated on a real-world dataset of 2,000 annotated images, evenly distributed across the four target classes (Blight, Red Rust, Helopeltis, and Healthy). Before training, the images underwent a standardized preprocessing pipeline including resizing to 224 × 224 pixels and normalization, supplemented by a dynamic augmentation strategy featuring random rotations, horizontal flips, and brightness adjustments to improve model robustness. The proposed hybrid framework achieved an outstanding peak classification accuracy of 96.80% alongside a macro Area Under the Curve (AUC) of 0.9980. To rigorously validate its performance, the hybrid model was benchmarked against six diverse architectures: a Vision Transformer (ViT-B16 at 76.40%), a Custom CNN (89.60%), MobileNetV3 (94.40%), ResNet50 (95.60%), DenseNet121 (96.40%), and EfficientNetV2-B3 (97.60%). Although EfficientNetV2-B3 achieved a marginally higher raw accuracy, the proposed dual-branch framework delivered a superior precision-recall balance and faster convergence stability. These findings demonstrate that the proposed hybrid methodology is highly reliable and computationally balanced, making it an ideal candidate for integration into Internet of Things (IoT) edge devices for real-time disease monitoring in precision agriculture.

Suggested Citation

  • Sakibul Hasan Chowdhury & Md Shohel Arman & Masrafe Bin Hannan Siam & Md Rayhan Khan & Afia Hasan & Parvez Ahmed Moju Fahim, 2026. "A curated dataset and lightweight deep learning framework for tea leaf disease classification," PLOS ONE, Public Library of Science, vol. 21(5), pages 1-36, May.
  • Handle: RePEc:plo:pone00:0349210
    DOI: 10.1371/journal.pone.0349210
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0349210
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0349210&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0349210?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0349210. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.