A Novel Approach to Defect Detection in Arabica Coffee Beans Using Deep Learning: Investigating Data Augmentation and Model Optimization

Yusriel Ardian; Novta Danyel Irawan; Sutoko Sutoko; I Nyoman Gede Arya Astawa; Ida Bagus Irawan Purnama; Felix Andika Dwiyanto

doi:10.17977/um018v7i12024p117-127

A Novel Approach to Defect Detection in Arabica Coffee Beans Using Deep Learning: Investigating Data Augmentation and Model Optimization

Yusriel Ardian, Novta Danyel Irawan, Sutoko Sutoko, I Nyoman Gede Arya Astawa, Ida Bagus Irawan Purnama, Felix Andika Dwiyanto

Abstract

Arabica coffee beans have valuable market worth because of their taste and quality, and there are defects like wholly and partially black beans that can lower the standards of a product, especially in the premium coffee sector. However, the manual processes used to detect the defects take an inordinate amount of time and are inefficient. This study aims to bridge the knowledge gap on the automated detection and recognition of the defects present in the Arabica coffee beans by creating and optimizing a CNN model based on a modified VGG16 architecture. The model applies data augmentation, rotation, cropping, and Bayesian hyperparameter optimization to improve defect detectability and expedite the training period. During testing, the defined model demonstrated excellent efficiency in defect detection, with a 97.29% confidence level, which was higher than that of the modified VGG16 and Slim-CNN models. The goal of the second optimization was an improvement of the practical application of the model. In terms of the time it takes for a model to be trained, approximately 30% of the time was saved. These findings present a consistent and effective way for the mass production processes of coffee to have quality control procedures automated. The model's ability to detect defects in other agricultural items makes it attractive, thus serving as a practical example of how AI can impact effective management in the inspection processes. The research further enriches the study of deep learning applications in agriculture by demonstrating how to efficiently address specific defect detection problems through an optimized convolutional neural network model.

Full Text:

PDF

References

J. E. Loppies et al., “Physical quality and flavor profile of arabica coffee beans (Coffea arabica) from Seko, South Sulawesi as a specialty coffee,” IOP Conf. Ser. Earth Environ. Sci., vol. 1338, no. 1, p. 012048, May 2024.

J. Li, “What Determines Coffee Aroma and Flavor?,” Berkeley Sci. J., vol. 26, no. 2, Aug. 2022.

Y. Abebe, B. Juergen, B. Endashaw, H. Kitessa, and G. Heiner, “The major factors influencing coffee quality in Ethiopia: The case of wild Arabica coffee (Coffea arabica L.) from its natural habitat of southwest and southeast afromontane rainforests,” African J. Plant Sci., vol. 14, no. 6, pp. 213–230, Jun. 2020.

C. de M. Carvalho, E. da S. Moreira, L. Zuim, M. C. de Paula, T. Lima Filho, and S. M. Della Lucia, “Does the presence of defective beans in Arabica healthy coffee beans affect the sensory perception and acceptance of coffee beverages?,” J. Sens. Stud., vol. 38, no. 6, Dec. 2023.

K. A. Maspul, “Pricing Consistency in Coffee Business Management: A Comprehensive Study,” J. Arbitr. Econ. Manag. Account., vol. 2, no. 02, pp. 78–86, 2024.

Z. Ren, F. Fang, N. Yan, and Y. Wu, “State of the Art in Defect Detection Based on Machine Vision,” Int. J. Precis. Eng. Manuf. Technol., vol. 9, no. 2, pp. 661–691, Mar. 2022.

R. M. Ariong, D. M. Okello, M. H. Otim, and P. Paparu, “The cost of inadequate postharvest management of pulse grain: Farmer losses due to handling and storage practices in Uganda,” Agric. Food Secur., vol. 12, no. 1, p. 20, Aug. 2023.

H.-D. Thai, H.-J. Ko, and J.-H. Huh, “Coffee Bean Defects Automatic Classification Realtime Application Adopting Deep Learning,” IEEE Access, vol. 12, pp. 126503–126517, 2024.

M. Agarwal, S. K. Gupta, and K. K. Biswas, “Development of Efficient CNN model for Tomato crop disease identification,” Sustain. Comput. Informatics Syst., vol. 28, p. 100407, Dec. 2020.

N. Shelar, S. Shinde, S. Sawant, S. Dhumal, and K. Fakir, “Plant Disease Detection Using Cnn,” ITM Web Conf., vol. 44, p. 03049, May 2022.

Z. Saeed, A. Raza, A. H. Qureshi, and M. Haroon Yousaf, “A Multi-Crop Disease Detection and Classification Approach using CNN,” in 2021 International Conference on Robotics and Automation in Industry (ICRAI), Oct. 2021, pp. 1–6.

M. Faisal, F. Albogamy, H. Elgibreen, M. Algabri, and F. A. Alqershi, “Deep Learning and Computer Vision for Estimating Date Fruits Type, Maturity Level, and Weight,” IEEE Access, vol. 8, pp. 206770–206782, 2020.

K.-W. Hsieh et al., “Fruit maturity and location identification of beef tomato using R-CNN and binocular imaging technology,” J. Food Meas. Charact., vol. 15, no. 6, pp. 5170–5180, Dec. 2021.

S. Chen, J. Xiong, J. Jiao, Z. Xie, Z. Huo, and W. Hu, “Citrus fruits maturity detection in natural environments based on convolutional neural networks and visual saliency map,” Precis. Agric., vol. 23, no. 5, pp. 1515–1531, Oct. 2022.

Bhupendra, K. Moses, A. Miglani, and P. Kumar Kankar, “Deep CNN-based damage classification of milled rice grains using a high-magnification image dataset,” Comput. Electron. Agric., vol. 195, p. 106811, Apr. 2022.

N. Genze, R. Bharti, M. Grieb, S. J. Schultheiss, and D. G. Grimm, “Accurate machine learning-based germination detection, prediction and quality assessment of three grain crops,” Plant Methods, vol. 16, no. 1, p. 157, Dec. 2020.

Y. Liu et al., “Non-Destructive Quality-Detection Techniques for Cereal Grains: A Systematic Review,” Agronomy, vol. 12, no. 12, p. 3187, Dec. 2022.

S.-Y. Chen, M.-F. Chiu, and X.-W. Zou, “Real-time defect inspection of green coffee beans using NIR snapshot hyperspectral imaging,” Comput. Electron. Agric., vol. 197, p. 106970, Jun. 2022.

J. P. Rodríguez, D. C. Corrales, J.-N. Aubertot, and J. C. Corrales, “A computer vision system for automatic cherry beans detection on coffee trees,” Pattern Recognit. Lett., vol. 136, pp. 142–153, Aug. 2020.

F. Xiong, N. Kühl, and M. Stauder, “Designing a computer-vision-based artifact for automated quality control: a case study in the food industry,” Flex. Serv. Manuf. J., Jan. 2024.

A. Badjona, R. Bradshaw, C. Millman, M. Howarth, and B. Dubey, “Faba Bean Flavor Effects from Processing to Consumer Acceptability,” Foods, vol. 12, no. 11, p. 2237, Jun. 2023.

W. Pelupessy, “The world behind the world coffee market,” Etud. Rurales, vol. 180, no. 02, pp. 187–212, 2007.

K. Maharana, S. Mondal, and B. Nemade, “A review: Data pre-processing and data augmentation techniques,” Glob. Transitions Proc., vol. 3, no. 1, pp. 91–99, Jun. 2022.

N. Madusanka, P. Jayalath, D. Fernando, L. Yasakethu, and B.-I. Lee, “Impact of H&E Stain Normalization on Deep Learning Models in Cancer Image Classification: Performance, Complexity, and Trade-Offs,” Cancers (Basel)., vol. 15, no. 16, p. 4144, Aug. 2023.

X. Pei et al., “Robustness of machine learning to color, size change, normalization, and image enhancement on micrograph datasets with large sample differences,” Mater. Des., vol. 232, p. 112086, Aug. 2023.

S. Singh et al., “A review of image fusion: Methods, applications and performance metrics,” Digit. Signal Process., vol. 137, p. 104020, Jun. 2023.

A. Mumuni and F. Mumuni, “Data augmentation: A comprehensive survey of modern approaches,” Array, vol. 16, p. 100258, Dec. 2022.

M. O. Khairandish, M. Sharma, V. Jain, J. M. Chatterjee, and N. Z. Jhanjhi, “A Hybrid CNN-SVM Threshold Segmentation Approach for Tumor Detection and Classification of MRI Brain Images,” IRBM, vol. 43, no. 4, pp. 290–299, Aug. 2022.

Z.-P. Jiang, Y.-Y. Liu, Z.-E. Shao, and K.-W. Huang, “An Improved VGG16 Model for Pneumonia Image Classification,” Appl. Sci., vol. 11, no. 23, p. 11185, Nov. 2021.

E. Hassan, M. Y. Shams, N. A. Hikal, and S. Elmougy, “The effect of choosing optimizer algorithms to improve computer vision tasks: a comparative study,” Multimed. Tools Appl., vol. 82, no. 11, pp. 16591–16633, May 2023.

J. M. Kernbach and V. E. Staartjes, “Foundations of Machine Learning-Based Clinical Prediction Modeling: Part II—Generalization and Overfitting,” 2022, pp. 15–21.

C. F. G. Dos Santos and J. P. Papa, “Avoiding Overfitting: A Survey on Regularization Methods for Convolutional Neural Networks,” ACM Comput. Surv., vol. 54, no. 10s, pp. 1–25, Jan. 2022.

R. Yacouby and D. Axman, “Probabilistic Extension of Precision, Recall, and F1 Score for More Thorough Evaluation of Classification Models,” in Proceedings of the First Workshop on Evaluation and Comparison of NLP Systems, 2020, pp. 79–91.

A. Amrani, D. Diepeveen, D. Murray, M. G. K. Jones, and F. Sohel, “Multi-task learning model for agricultural pest detection from crop-plant imagery: A Bayesian approach,” Comput. Electron. Agric., vol. 218, p. 108719, Mar. 2024.

O. O. Abayomi-Alli, R. Damaševičius, A. Qazi, M. Adedoyin-Olowe, and S. Misra, “Data Augmentation and Deep Learning Methods in Sound Classification: A Systematic Review,” Electronics, vol. 11, no. 22, p. 3795, Nov. 2022.

X. Liu, H. Zhu, W. Song, J. Wang, L. Yan, and K. Wang, “Research on Improved VGG-16 Model Based on Transfer Learning for Acoustic Image Recognition of Underwater Search and Rescue Targets,” IEEE J. Sel. Top. Appl. Earth Obs. Remote Sens., vol. 17, pp. 18112–18128, 2024.

DOI: http://dx.doi.org/10.17977/um018v7i12024p117-127