Enhanced EfficientNet-B0 with Dual Attention Mechanisms for Food Category Classification in X-ray Images
Source abstract
Abstract
Abstract To improve the operational efficiency of food foreign object inspection systems, this study proposes a food category recognition model for X-ray images. 6800 X-ray images containing 34 types of food were collected. Image labels were automatically assigned according to the folder corresponding to each food category. The dataset was divided into training, validation, and testing sets in an 8:1:1 ratio. The model was based on EfficientNet-B0. To enhance the representation capability of key image regions, Convolutional Block Attention Module (CBAM) and Efficient Channel Attention Module (ECA) have been added to EfficientNet-B0. To improve the robustness of the model to noise and image variability, the images were enhanced through various methods such as random scaling, horizontal flipping, rotation, occlusion and so on. The proposed model achieved a test accuracy of 96.60%, which is 2.03% higher than ResNet-50, 1.56% higher than the baseline EfficientNet-B0, and 2.14% higher than MobileNet-V2. The test results indicate that the proposed method can provide accurate, efficient, and automated technical solutions for food classification in X-ray images.
Source: openaire · Quality: candidate unverified · Source record ↗
Research context
OpenAlex topics: Nutritional Studies and Diet · Identification and Quantification in Food · Consumer Attitudes and Food Labeling
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (31)
Show source-provided references
1. Garcia, S.N., Osburn, B.I., Jay-Russell, M.T.: One health for food safety, food security, and sustainable food production. Front. Sustain. Food Syst. 4, 1 (2020)10.3389/fsufs.2020.00001 ↗
2. Gizaw, Z.: Public health risks related to food safety issues in the food market: a systematic literature review. Environ. Health Prev. Med. 24(1), 68 (2019)10.1186/s12199-019-0825-5 ↗
3. Organization WH.: WHO global strategy for food safety 2022–2030: towards stronger food safety systems and global cooperation. World Health Organization (2022)
4. De Prados, M., Fulladosa, E., Gou, P., et al.: Non-destructive determination of fat content in green hams using ultrasound and X-rays. Meat Sci. 104, 37–43 (2015)10.1016/j.meatsci.2015.01.015 ↗
5. Kotwaliwale, N., Weckler, P.R., Brusewitz, G.H., et al.: Non-destructive quality determination of pecans using soft X-rays. Postharvest Biol. Technol. 45(3), 372–380 (2007)10.1016/j.postharvbio.2007.03.008 ↗
6. Yao, J., Wu, Z., Wang, P., et al.: Research on X-ray image fusion algorithm for food foreign object detection. J. Nondestr. Eval. 44(2), 63 (2025)10.1007/s10921-025-01203-y ↗
7. Gill, H.S., Murugesan, G., Mehbodniya, A., et al.: Fruit type classification using deep learning and feature fusion. Comp. Electron. Agric. 211, 107990 (2023)10.1016/j.compag.2023.107990 ↗
8. Xue, G., Liu, S., Ma, Y.: A hybrid deep learning-based fruit classification using attention model and convolution autoencoder. Complex Intell. Syst. 9(3), 2209–2219 (2023)10.1007/s40747-020-00192-x ↗
9. Allken, V., Handegard, N.O., Rosen, S., et al.: Fish species identification using a convolutional neural network trained on synthetic data. ICES J. Mar. Sci. 76(1), 342–349 (2019)10.1093/icesjms/fsy147 ↗
10. He W, Liu M, Liang Q.: Capturing fine-grained food image features through iterative clustering and attention mechanisms. IEEE Access (2025)10.1109/ACCESS.2025.3561465 ↗
11. Lee, D.H., Kim, E.S., Cho, J.S., et al.: A two-stage automatic labeling method for detecting abnormal food items in X-ray images. J. Food Meas. Charact. 16(4), 2999–3009 (2022)10.1007/s11694-022-01387-1 ↗
12. Nazir, N., Sarwar, A., Saini, B.S., et al.: A Robust deep learning approach for accurate segmentation of cytoplasm and nucleus in noisy pap smear images. Computation. 11(10), 195 (2023)10.3390/computation11100195 ↗
13. Shoaib, M., Shah, B., Ei-Sappagh, S., et al.: Corrigendum: an advanced deep learning models-based plant disease detection: a review of recent research. Front. Plant Sci. 14, 1158933 (2023)10.3389/fpls.2023.1158933 ↗
14. Dang, M., Wang, H., Li, Y., et al.: Computer vision for plant disease recognition: a comprehensive review. Bot. Rev. 90(3), 251–311 (2024)10.1007/s12229-024-09299-z ↗
15. Einarsson G, Jensen JN, Paulsen RR, et al.: Foreign object detection in multispectral x-ray images of food items using sparse discriminant analysis. Scandinavian conference on image analysis. Cham: Springer International Publishing p. 350–61 (2017)10.1007/978-3-319-59126-1_29 ↗
16. Lee, M.S., Park, C.H., Kang, M.G.: Edge enhancement algorithm for low-dose X-ray fluoroscopic imaging. Comput. Methods Programs Biomed. 152, 45–52 (2017)10.1016/j.cmpb.2017.09.010 ↗
17. Samei, E., Dobbins, J.T., III., Lo, J.Y., et al.: A framework for optimising the radiographic technique in digital X-ray imaging. Radiat. Prot. Dosimetry. 114(1–3), 220–229 (2005)10.1093/rpd/nch562 ↗
18. Mahasin, M., Dewi, I.A.: Comparison of cspdarknet53, cspresnext-50, and efficientnet-b0 backbones on yolo v4 as object detector. Int. J. Eng. Sci. Inform. Technol. 2(3), 64–72 (2022)10.52088/ijesty.v2i3.291 ↗
19. Kansal, K., Chandra, T.B., Singh, A.: ResNet-50 vs. EfficientNet-B0: multi-centric classification of various lung abnormalities using deep learning. Procedia Comput. Sci. 235, 70–80 (2024)10.1016/j.procs.2024.04.007 ↗
20. Tadepalli, Y., Kollati, M., Kuraparthi, S., et al.: EfficientNet-B0 based monocular dense-depth map estimation. Trait. Signal (2021). https://doi.org/10.18280/ts.38052410.18280/ts.380524 ↗
21. UNICOMP: UNICOMP small and medium packaging/bagged food x-ray foreign object detection equipment UNX4015-N. https://www.unicomp.cn/FOD-X-Rayi01714.html (2024). Accessed 4 August 2025
22. Kayhan OS, Gemert JCv.: On translation invariance in cnns: Convolutional layers can exploit absolute spatial location. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. p. 14274–85 (2020)
23. Purwono, P., Ma’arif, A., Rahmaniar, W., et al.: Understanding of convolutional neural network (cnn): A review. Int. J. Robot. Control Syst. 2(4), 739–748 (2022)10.31763/ijrcs.v2i4.888 ↗
24. Madaan, V., Roy, A., Gupta, C., et al.: XCOVNet: chest X-ray image classification for COVID-19 early detection using convolutional neural networks. N. Gener. Comput. 39(3), 583–597 (2021)10.1007/s00354-021-00121-7 ↗
25. Shamrat, F.J.M., Azam, S., Karim, A., et al.: High-precision multiclass classification of lung disease through customized MobileNetV2 from chest X-ray images. Comput. Biol. Med. 155, 106646 (2023)10.1016/j.compbiomed.2023.106646 ↗
26. Alshmrani, G.M.M., Ni, Q., Jiang, R., et al.: A deep learning architecture for multi-class lung diseases classification using chest X-ray (CXR) images. Alex. Eng. J. 64, 923–935 (2023)10.1016/j.aej.2022.10.053 ↗
27. Jiang, N.F., Shao, Y.: Learning invariant representation using synthetic imagery for object detection. AI Commun. 36(1), 13–25 (2022). https://doi.org/10.3233/AIC-22003910.3233/AIC-220039 ↗
28. Nfor, K.A.A., Theodore Armand, T.P., Ismaylovna, K.P., Joo, M.I., Kim, H.C.: An explainable CNN and vision transformer-based approach for real-time food recognition. Nutrients 17(2), 362 (2025)10.3390/nu17020362 ↗
29. Yang, K., He, Z., Pei, W., et al.: SiamCorners: Siamese corner networks for visual tracking. IEEE Trans. Multimedia 24, 1956–1967 (2021)10.1109/TMM.2021.3074239 ↗
30. Yang, K., Li, Q., Tian, C., et al.: Defort: deformable transformer for visual tracking. Neural Netw. 176, 106380 (2024)10.1016/j.neunet.2024.106380 ↗
31. Huang, Y.W., Li, X.: Food image segmentation using multi-modal imaging sensors with color and thermal data. Sensors 23(2), 560 (2023). https://doi.org/10.3390/s2302056010.3390/s23020560 ↗