DeClEx-Processing Pipeline for Tumor Classification

Authors: Gaurav Shinde, Sai Charan Gongiguntla, Prajwal Shirur, Ahmed Hambaba

Abstract:

Health issues are significantly increasing, putting a substantial strain on healthcare services. This has accelerated the integration of machine learning in healthcare, particularly following the COVID-19 pandemic. The utilization of machine learning in healthcare has grown significantly. We introduce DeClEx, a pipeline which ensures that data mirrors real-world settings by incorporating gaussian noise and blur and employing autoencoders to learn intermediate feature representations. Subsequently, our convolutional neural network, paired with spatial attention, provides comparable accuracy to state-of-the-art pre-trained models while achieving a threefold improvement in training speed. Furthermore, we provide interpretable results using explainable AI techniques. We integrate denoising and deblurring, classification and explainability in a single pipeline called DeClEx.

Keywords: Machine learning, healthcare, classification, explainability.

References:

[1] Nanyue, W. et al. Comparative study of pulse- diagnosis signals between 2 kinds of liver disease patients based on the combination of unsupervised learning and supervised learning. 2013 IEEE International Conference on Bioinformatics and Biomedicine 2013, pp. 260-262. (Online). Available: https://doi.org/10.1109/BIBM.2013.6732688
[2] Kim, S. K., Yoo, T. K., Oh, E., & Kim, D. W. Osteoporosis risk prediction using machine learning and conventional methods. Annual International Conference of the IEEE Engineering in Medicine and Biology Society 2013, pp. 188-191. (Online). Available: https://doi.org/10.1109/EMBC.2013.6609469
[3] Seixas, J. L., Barbon, S., & Mantovani, R. G. Pattern Recognition of Lower Member Skin Ulcers in Medical Images with Machine Learning Algorithms. 2015 IEEE 28th International Symposium on Computer-Based Medical Systems 2015, pp. 50-53. (Online). Available: https://doi.org/10.1109/CBMS.2015.48
[4] van Tulder, G. & de Bruijne, M. Combining Generative and Discriminative Representation Learning for Lung CT Analysis with Convolutional Restricted Boltzmann Machines. IEEE Transactions on Medical Imaging 2016, 35(5), pp. 1262-1272. (Online). Available: https://doi.org/10.1109/TMI.2016.2526687
[5] Gerazov, B. & Conceicao, R. C. Deep learning for tumour classification in homogeneous breast tissue in medical microwave imaging. IEEE EUROCON 2017 - 17th International Conference on Smart Technologies 2017, pp. 564-569. (Online). Available: https://doi.org/10.1109/EUROCON.2017.8011175
[6] Tyagi, A., Mehra, R., & Saxena, A. Interactive Thyroid Disease Prediction System Using Machine Learning Technique. 2018 Fifth International Conference on Parallel, Distributed and Grid Computing (PDGC) 2018, pp. 689-693. (Online). Available: https://doi.org/10.1109/PDGC.2018.8745910
[7] Anastasiou, A., Pitoglou, S., Androutsou, T., Kostalas, E., Matsopoulos, G., & Koutsouris, D. MODELHealth: An Innovative Software Platform for Machine Learning in Healthcare Leveraging Indoor Localization Services. 2019 20th IEEE International Conference on Mobile Data Management (MDM) 2019, pp. 443-446. (Online). Available: https://doi.org/10.1109/MDM.2019.000-5
[8] Tsarapatsani, K. et al. Machine Learning Models for Cardiovascular Disease Events Prediction. 2022 44th Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC) 2022, pp. 1066-1069. (Online) Available: https://doi.org/10.1109/EMBC48229.2022.9871121
[9] Sabir, M.W., Khan, Z., Saad, N.M., Khan, D.M., Al-Khasawneh, M.A., Perveen, K., Qayyum, A., & Ali, S.S. Azhar. Segmentation of Liver Tumor in CT scan Using ResU-Net. Applied Sciences 2022, 12, 8650. (Online). Available: https://doi.org/10.3390/app12178650
[10] Zhao, X., Fang, C., Fan, D.-J., Lin, X., Gao, F., & Li, G. Cross-Level Contrastive Learning and Consistency Constraint for Semi-Supervised Medical Image Segmentation. 2022 IEEE 19th International Symposium on Biomedical Imaging (ISBI) 2022, pp. 1-5. (Online). Available: https://doi.org/10.1109/ISBI52829.2022.9761710
[11] Li, X., Fan, Y., Zheng, H., Gao, J., Wei, X., & Yu, M. Balanced and Discriminative Contrastive Learning For Class-Imbalanced Medical Images. ICASSP 2024 - 2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) 2024, pp. 3735-3739. (Online). Available: https://doi.org/10.1109/ICASSP48485.2024.10447586
[12] Workman, T. E. et al. Explainable Deep Learning Applied to Understanding Opioid Use Disorder and Its Risk Factors. 2019 IEEE International Conference on Big Data (Big Data) 2019, pp. 4883-4888. (Online). Available: https://doi.org/10.1109/BigData47090.2019.9006297
[13] Hossain, M. S., Muhammad, G., & Guizani, N. Explainable AI and Mass Surveillance System-Based Healthcare Framework to Combat COVID-19 Like Pandemics. IEEE Network 2020, 34(4), pp. 126-132, July/August. (Online). Available: https://doi.org/10.1109/MNET.011.2000458
[14] Marvin, G. & Alam, M. G. R. Explainable Feature Learning for Predicting Neonatal Intensive Care Unit (NICU) Admissions. 2021 IEEE International Conference on Biomedical Engineering, Computer and Information Technology for Health 2021, pp. 69-74. (Online). Available: https://doi.org/10.1109/BECITHCON54710.2021.9893719
[15] Ammar, N. et al. SPACES: Explainable Multimodal AI for Active Surveillance, Diagnosis, and Management of Adverse Childhood Experiences (ACEs). 2021 IEEE International Conference on Big Data (Big Data) 2021, pp. 5843-5847. (Online). Available: https://doi.org/10.1109/BigData52589.2021.9671303
[16] She, W. J. et al. Investigation of a Web-Based Explainable AI Screening for Prolonged Grief Disorder. IEEE Access 2022, 10, pp. 41164-41185. (Online). Available: https://doi.org/10.1109/ACCESS.2022.3163311
[17] Rahman, A. & Alam, M. G. Rabiul. Explainable AI-based Maternal Health Risk Prediction using Machine Learning and Deep Learning. 2023 IEEE World AI IoT Congress (AIIoT) 2023, pp. 0013-0018. (Online). Available: https://doi.org/10.1109/AIIoT58121.2023.10174540
[18] Palkar, A., Dias, C. C., Chadaga, K., & Sampathila, N. Empowering Glioma Prognosis with Transparent Machine Learning and Interpretative Insights Using Explainable AI. IEEE Access 2024, 12, pp. 31697-31718. (Online). Available: https://doi.org/10.1109/ACCESS.2024.3370238
[19] Nickparvar, M. Brain Tumor MRI Dataset. Kaggle. https://www.kaggle.com/datasets/masoudnickparvar/brain-tumor-mri-dataset
[20] Shinde, G., Mohapatra, R., Krishan, P., Garg, H., Prabhu, S., Das, S., Masum, M., & Sengupta, S. The State of Lithium-Ion Battery Health Prognostics in the CPS Era. to be submitted.
[21] Chen, D., Hong, W., & Zhou, X. Transformer Network for Remaining Useful Life Prediction of Lithium-Ion Batteries. IEEE Access 2022, 10, pp. 19621-19628. (Online). Available: https://doi.org/10.1109/ACCESS.2022.3151975
[22] Yu, J., Shen, Y., Liu, N., & Pan, Q. Frequency-Enhanced Channel-Spatial Attention Module for Grain Pests Classification. Agriculture 2022, 12, 2046. (Online). Available: https://doi.org/10.3390/agriculture12122046
[23] Ribeiro, M. T., Singh, S., & Guestrin, C. "Why Should I Trust You?": Explaining the Predictions of Any Classifier 2016. arXiv preprint arXiv:1602.04938.
[24] Lundberg, S. & Lee, S.-I. A Unified Approach to Interpreting Model Predictions 2017. arXiv preprint arXiv:1705.07874.
[25] Radford, A. & Narasimhan, K. Improving Language Understanding by Generative Pre-Training 2018.
[26] Touvron, H., Lavril, T., Izacard, G., Martinet, X., Lachaux, M.-A., Lacroix, T., Rozière, B., Goyal, N., Hambro, E., Azhar, F., Rodriguez, A., Joulin, A., Grave, E., & Lample, G. LLaMA: Open and Efficient Foundation Language Models 2023. arXiv preprint arXiv:2302.13971.
[27] Singhal, K., Azizi, S., Tu, T. et al. Large language models encode clinical knowledge. Nature 620, 172–180 2023. https://doi.org/10.1038/s41586-023-06291-2
[28] Lewis, P., Perez, E., Piktus, A., Petroni, F., Karpukhin, V., Goyal, N., Küttler, H., Lewis, M., Yih, W.-t., Rocktäschel, T., Riedel, S., & Kiela, D. Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasks 2021. arXiv preprint arXiv:2005.11401.
[29] Chen, Y., Zhang, M., Li, J., & Kuang, X. Adversarial Attacks and Defenses in Image Classification: A Practical Perspective. 2022 7th International Conference on Image, Vision, and Computing (ICIVC) 2022, pp. 424-430. (Online). Available: https://doi.org/10.1109/ICIVC55077.2022.9886997