XNet: a convolutional neural network (CNN) implementation for medical x-ray image segmentation suitable for small datasets
Abstract
X-Ray image enhancement, along with many other medical image processing applications, requires the segmentation of images into bone, soft tissue, and open beam regions. We apply a machine learning approach to this problem, presenting an end-to-end solution which results in robust and efficient inference. Since medical institutions often do not have the resources to process and label the large quantity of X-Ray images usually needed for neural network training, we design an end-to-end solution for small datasets, while achieving state-of-the-art results. Our implementation produces an overall accuracy of 92%, F1 score of 0.92, and an AUC of 0.98, surpassing classical image processing techniques, such as clustering and entropy based methods, while improving upon the output of existing neural networks used for segmentation in non-medical contexts. The code used for this project is available online.1
- Publication:
-
Medical Imaging 2019: Biomedical Applications in Molecular, Structural, and Functional Imaging
- Pub Date:
- March 2019
- DOI:
- 10.1117/12.2512451
- arXiv:
- arXiv:1812.00548
- Bibcode:
- 2019SPIE10953E..1ZB
- Keywords:
-
- Computer Science - Computer Vision and Pattern Recognition;
- Computer Science - Artificial Intelligence;
- Physics - Medical Physics
- E-Print:
- 11 pages, 5 figures, 2 tables