Affiliation:
1. Computer Science and Engineering Department, American University of Sharjah, Sharjah P.O. Box 26666, United Arab Emirates
Abstract
This study presents a comprehensive analysis of utilizing TensorFlow Lite on mobile phones for the on-edge medical diagnosis of lung diseases. This paper focuses on the technical deployment of various deep learning architectures to classify nine respiratory system diseases using X-ray imagery. We propose a simple deep learning architecture that experiments with six different convolutional neural networks. Various quantization techniques are employed to convert the classification models into TensorFlow Lite, including post-classification quantization with floating point 16 bit representation, integer quantization with representative data, and quantization-aware training. This results in a total of 18 models suitable for on-edge deployment for the classification of lung diseases. We then examine the generated models in terms of model size reduction, accuracy, and inference time. Our findings indicate that the quantization-aware training approach demonstrates superior optimization results, achieving an average model size reduction of 75.59%. Among many CNNs, MobileNetV2 exhibited the highest performance-to-size ratio, with an average accuracy loss of 4.1% across all models using the quantization-aware training approach. In terms of inference time, TensorFlow Lite with integer quantization emerged as the most efficient technique, with an average improvement of 1.4 s over other conversion approaches. Our best model, which used EfficientNetB2, achieved an F1-Score of approximately 98.58%, surpassing state-of-the-art performance on the X-ray lung diseases dataset in terms of accuracy, specificity, and sensitivity. The model experienced an F1 loss of around 1% using quantization-aware optimization. The study culminated in the development of a consumer-ready app, with TensorFlow Lite models tailored to mobile devices.
Reference30 articles.
1. Kufel, J., Bargieł, K., Koźlik, M., Czogalik, Ł., Dudek, P., Jaworski, A., Magiera, M., Bartnikowska, W., Cebula, M., and Nawrat, Z. (2022). Usability of mobile solutions intended for diagnostic images—A systematic review. Healthcare, 10.
2. Digital diagnosis: Privacy and the regulation of mobile phone health applications;Flaherty;Am. J. Law Med.,2014
3. Use of mobile devices for medical imaging;Hirschorn;J. Am. Coll. Radiol.,2014
4. Diagnostic concordance between mobile interfaces and conventional workstations for emergency imaging assessment;Venson;Int. J. Med. Inform.,2018
5. Privacy and security issues with mobile health research applications;Tovino;J. Law Med. Ethics,2020
Cited by
1 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献