A DEEP LEARNING ENSEMBLE APPROACH FOR X-RAY IMAGE CLASSIFICATION

Abstract

The application of deep learning-based intelligent systems for X-ray imaging in various settings, including transportation, customs inspections, and public security, to identify hidden or prohibited objects are discussed in this study. In busy environments, x-ray inspections face challenges due to time limitations and a lack of qualified personnel. Deep learning algorithms can automate the imaging process, enhancing object detection and improving safety. This study uses a dataset of 5094 x-ray images of laptops with hidden foreign circuits and normal ones, training 11 deep learning algorithms with the 10-fold cross-validation method. The predictions of deep learning models selected based on the 70% threshold value have been combined using a meta-learner. ShuffleNet has the highest individual performance with 83.56%, followed by InceptionV3 at 81.30%, Darknet19 at 78.92%, DenseNet201 at 77.70% and Xception at 71.26%. Combining these models into an ensemble achieved a remarkable classification success rate of 85.97%, exceeding the performance of any individual model. The ensemble learning approach provides a more stable prediction output, reducing standard deviation among folds as well. This research highlights the potential for safer and more effective X-ray inspections through advanced machine learning techniques.

Keywords

• Deep Learning
• Ensemble Learning
• Object Classification
• X-Ray

References

1. S. Akçay, M. E. Kundegorski, M. Devereux, and T. P. Breckon, "Transfer learning using convolutional neural networks for object classification within X-ray baggage security imagery," in 2016 IEEE International Conference on Image Processing (ICIP), 25-28 Sept. 2016 2016, pp. 1057-1061, doi: 10.1109/ICIP.2016.7532519.
2. E. Benedykciuk, M. Denkowski, and K. Dmitruk, "Material classification in X-ray images based on multi-scale CNN," Signal, Image and Video Processing, vol. 15, no. 6, pp. 1285-1293, 2021/09/01 2021, doi: 10.1007/s11760-021-01859-9.
3. C. Miao et al., "Sixray: A large-scale security inspection x-ray benchmark for prohibited item discovery in overlapping images," in Proc. IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) 2019, pp. 2119-2128.
4. A. Chang, Y. Zhang, S. Zhang, L. Zhong, and L. Zhang, "Detecting prohibited objects with physical size constraint from cluttered X-ray baggage images," Knowledge-Based Systems, vol. 237, p. 107916, 2022.
5. F. Shao, J. Liu, P. Wu, Z. Yang, and Z. Wu, "Exploiting foreground and background separation for prohibited item detection in overlapping X-Ray images," Pattern Recognition, vol. 122, p. 108261, 2022.
6. C. Zhao et al., "BoostTree and BoostForest for ensemble learning," IEEE Transactions on Pattern Analysis and Machine Intelligence, 2022.
7. Z.-H. Zhou, Ensemble learning. Springer, 2021.
8. I. H. Witten, E. Frank, and M. A. Hall, "Data mining: Practical machine learning tools and techniques," ed: Morgan Kaufmann, 2016.

9. A. Ait Nasser and M. A. Akhloufi, "A review of recent advances in deep learning models for chest disease detection using radiography," Diagnostics, vol. 13, no. 1, p. 159, 2023.
10. M. Radak, H. Y. Lafta, and H. Fallahi, "Machine learning and deep learning techniques for breast cancer diagnosis and classification: a comprehensive review of medical imaging studies," Journal of Cancer Research and Clinical Oncology, pp. 1-19, 2023.
11. W. Khan, N. Zaki, and L. Ali, "Intelligent pneumonia identification from chest x-rays: A systematic literature review," IEEE Access, vol. 9, pp. 51747-51771, 2021.
12. H. Wang, H. Li, W. Gao, and J. Xie, "PrUb-EL: A hybrid framework based on deep learning for identifying ubiquitination sites in Arabidopsis thaliana using ensemble learning strategy," Analytical Biochemistry, vol. 658, p. 114935, 2022.
13. E. Putin et al., "Deep biomarkers of human aging: application of deep neural networks to biomarker development," Aging (Albany NY), vol. 8, no. 5, p. 1021, 2016.
14. P. Peng, C. Marceau, and D. M. Villeneuve, "Attosecond imaging of molecules using high harmonic spectroscopy," Nature Reviews Physics, vol. 1, no. 2, pp. 144-155, 2019.
15. R. Xie and M. Marsili, "A random energy approach to deep learning," Journal of Statistical Mechanics: Theory and Experiment, vol. 2022, no. 7, p. 073404, 2022.
16. S. Kolte, N. Bhowmik, and Dhiraj, "Threat Object-based anomaly detection in X-ray images using GAN-based ensembles," Neural Computing and Applications, pp. 1-16, 2022.
17. Q. Kong, N. Akira, B. Tong, Y. Watanabe, D. Matsubara, and T. Murakami, "Multimodal Deep Neural Networks Based Ensemble Learning for X-Ray Object Recognition," 2019: Springer, pp. 523-538.
18. R. Zhou, F. Wang, X. Fang, A. Fenster, and H. Gan, "An adaptively weighted ensemble of multiple CNNs for carotid ultrasound image segmentation," Biomedical Signal Processing and Control, vol. 83, p. 104673, 2023.
19. A. H. Ahmed, M. Al Radi, and N. Werghi, "An Ensemble Learning Method Based on Deep Neural and Pca-Based Svm Network for Baggage Threat and Smoke Recognition," in Proc. Advances in Science and Engineering Technology International Conferences, Dubai, 2023, pp. 1-6.
20. A. Kumar and J. Mayank, "Ensemble learning for AI developers," BApress: Berkeley, CA, USA, 2020.
21. E. O. Kiyak, "Data Mining and Machine Learning for Software Engineering," Data Mining-Methods, Applications and Systems, 2020.
22. X. Zhang, X. Zhou, M. Lin, and J. Sun, "Shufflenet: An extremely efficient convolutional neural network for mobile devices," in Proc. IEEE conference on computer vision and pattern recognition, 2018, pp. 6848-6856.
23. J. Redmon and A. Farhadi, "YOLO9000: better, faster, stronger," in Proc. IEEE conference on computer vision and pattern recognition, 2017, pp. 7263-7271.
24. J. Redmon and A. Farhadi, "Yolov3: An incremental improvement," arXiv preprint arXiv:1804.02767, 2018.
25. C. Szegedy et al., "Going deeper with convolutions," in Proc. IEEE conference on computer vision and pattern recognition, 2015, pp. 1-9.
26. G. Huang, Z. Liu, L. Van Der Maaten, and K. Q. Weinberger, "Densely connected convolutional networks,", in Proc. IEEE conference on computer vision and pattern recognition, 2017, pp. 4700-4708.
27. F. Chollet, "Xception: Deep learning with depthwise separable convolutions," in Proc. IEEE conference on computer vision and pattern recognition, 2017, pp. 1251-1258.
28. Ž. Vujović, "Classification model evaluation metrics," International Journal of Advanced Computer Science and Applications, vol. 12, no. 6, pp. 599-606, 2021.