Development of an X-ray image processing and diagnosis tool for breast imaging using deep learning neural networks

Thumbnail Image
Χατζάκης, Δημήτριος
Journal Title
Journal ISSN
Volume Title
Recently a multitude of machine learning methods and deep learning architectures have been under development for deployment in the field of medical imaging. Presented forward, a thorough analysis of breast cancer imaging analysis techniques is done with focus on the use of AI. Methods achieving results of highest accuracy of breast tumor segmentation and classification are presented with the aim of developing a robust, accurate and efficient deep learning architecture, to aid medical doctors accurately detect malignancies in an early stage. Extensive research with the inclusion of more than 350 related published papers was performed. It contained results published between 2016 to 2020 that resulted in a review of selected models, all at the vanguard of state-of-the-art progress. One easy to use, customizable and very efficient architecture, that of the UNet, was selected for benchmarking to find the optimal hyperparameter tuning and layer optimization. Two different public datasets were tested, both quantitatively and qualitatively, the MIAS and the CBIS-DDSM. The optimal results discovered for the UNet are a four-layer downsampling stream with 64 initial filters and a dropout of 0.3. Obtained results are presented in which although tumors are detected in most of the cases , other regions ,mainly in the breast’s edges, are miss detected as suspicious areas. Further work could improve on it with the addition of one or more of backbone networks in the UNet architecture.
Breast imaging, Tumor segmentation, Tumor classification, Neural networks