Computer-Aided Diagnosis of Vertebral Compression Fractures Using Convolutional Neural Networks and Radiomics

Abstract

Vertebral Compression Fracture (VCF) occurs when the vertebral body partially collapses under the action of compressive forces. Non-traumatic VCFs can be secondary to osteoporosis fragility (benign VCFs) or tumors (malignant VCFs). The investigation of the etiology of non-traumatic VCFs is usually necessary, since treatment and prognosis are dependent on the VCF type. Currently, there has been great interest in using Convolutional Neural Networks (CNNs) for the classification of medical images because these networks allow the automatic extraction of useful features for the classification in a given problem. However, CNNs usually require large datasets that are often not available in medical applications. Besides, these networks generally do not use additional information that may be important for classification. A different approach is to classify the image based on a large number of predefined features, an approach known as radiomics. In this work, we propose a hybrid method for classifying VCFs that uses features from three different sources: i) intermediate layers of CNNs; ii) radiomics; iii) additional clinical and image histogram information. In the hybrid method proposed here, external features are inserted as additional inputs to the first dense layer of a CNN. A Genetic Algorithm is used to: i) select a subset of radiomic, clinical, and histogram features relevant to the classification of VCFs; ii) select hyper-parameters of the CNN. Experiments using different models indicate that combining information is interesting to improve the performance of the classifier. Besides, pre-trained CNNs presents better performance than CNNs trained from scratch on the classification of VCFs.

Keywords: Computer aided diagnosis; Convolutional neural network; Genetic algorithm; Vertebral compression fracture.