27 February 2018 Reduction in training time of a deep learning model in detection of lesions in CT
Author Affiliations +
Deep learning (DL) emerged as a powerful tool for object detection and classification in medical images. Building a well-performing DL model, however, requires a huge number of images for training, and it takes days to train a DL model even on a cutting edge high-performance computing platform. This study is aimed at developing a method for selecting a “small” number of representative samples from a large collection of training samples to train a DL model for the could be used to detect polyps in CT colonography (CTC), without compromising the classification performance. Our proposed method for representative sample selection (RSS) consists of a K-means clustering algorithm. For the performance evaluation, we applied the proposed method to select samples for the training of a massive training artificial neural network based DL model, to be used for the classification of polyps and non-polyps in CTC. Our results show that the proposed method reduce the training time by a factor of 15, while maintaining the classification performance equivalent to the model trained using the full training set. We compare the performance using area under the receiveroperating- characteristic curve (AUC).
© (2018) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Nazanin Makkinejad, Nima Tajbakhsh, Amin Zarshenas, Ashfaq Khokhar, Kenji Suzuki, "Reduction in training time of a deep learning model in detection of lesions in CT", Proc. SPIE 10575, Medical Imaging 2018: Computer-Aided Diagnosis, 105753V (27 February 2018); doi: 10.1117/12.2293679; https://doi.org/10.1117/12.2293679

Back to Top