A rotation and translation invariant method for 3D organ image classification using deep convolutional neural networks

Three-dimensional (3D) medical image classification is useful in applications such as disease diagnosis and content-based medical image retrieval. It is a challenging task due to several reasons. First, image intensity values are vastly different depending on the image modality. Second, intensity values within the same image modality may vary depending on the imaging machine and artifacts may also be introduced in the imaging process. Third, processing 3D data requires high computational power. In recent years, significant research has been conducted in the field of 3D medical image classification. However, most of these make assumptions about patient orientation and imaging direction to simplify the problem and/or workwith the full 3D images. As such, they perform poorly when these assumptions are not met. In this paper, we propose a method of classification for 3D organ images that is rotation and translation invariant. To this end, we extract a representative two-dimensional (2D) slice along the plane of best symmetry from the 3D image. We then use this slice to represent the 3D image and use a 20-layer deep convolutional neural network (DCNN) to perform the classification task. We show experimentally, using multi-modal data, that our method is comparable to existing methods when the assumptions of patient orientation and viewing direction are met. Notably, it shows similarly high accuracy even when these assumptions are violated, where other methods fail. We also explore how this method can be used with other DCNN models as well as conventional classification approaches. Subjects Artificial Intelligence, Computer Vision, Data Mining and Machine Learning

[1]  Forrest N. Iandola,et al.  SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <1MB model size , 2016, ArXiv.

[2]  Byung Bok Ahn The Compact 3 D Convolutional Neural Network for Medical Images , .

[3]  Wen Gao,et al.  3D Haar-Like Features for Pedestrian Detection , 2007, 2007 IEEE International Conference on Multimedia and Expo.

[4]  José A. Gámez,et al.  Medical images modality classification using discrete Bayesian Networks , 2016 .

[5]  Geethu Mohan,et al.  MRI based medical image analysis: Survey on brain tumor grade classification , 2018, Biomed. Signal Process. Control..

[6]  Zellig S. Harris,et al.  Distributional Structure , 1954 .

[7]  David Dagan Feng,et al.  Content-Based Medical Image Retrieval: A Survey of Applications to Multidimensional and Multimodality Data , 2013, Journal of Digital Imaging.

[8]  Marka Oreškovi IMAGE INTERPOLATION TECHNIQUES , 2004 .

[9]  Bill Triggs,et al.  Histograms of oriented gradients for human detection , 2005, 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'05).

[10]  Muhammad Awais,et al.  Medical image retrieval using deep convolutional neural network , 2017, Neurocomputing.

[11]  Paul J. Besl,et al.  A Method for Registration of 3-D Shapes , 1992, IEEE Trans. Pattern Anal. Mach. Intell..

[12]  Yanxi Li,et al.  Truly 3D midsagittal plane extraction for robust neuroimage registration , 2006, 3rd IEEE International Symposium on Biomedical Imaging: Nano to Macro, 2006..

[13]  Christian Igel,et al.  Deep Feature Learning for Knee Cartilage Segmentation Using a Triplanar Convolutional Neural Network , 2013, MICCAI.

[14]  Xiangrong Zhou,et al.  Automatic segmentation and recognition of anatomical lung structures from high-resolution chest CT images , 2006, Comput. Medical Imaging Graph..

[15]  N. Japkowicz Why Question Machine Learning Evaluation Methods ? ( An illustrative review of the shortcomings of current methods ) , 2006 .

[16]  Tony F. Chan,et al.  Active contours without edges , 2001, IEEE Trans. Image Process..

[17]  Zellig S. Harris,et al.  Distributional Structure , 1954 .

[18]  Xiuying Wang,et al.  Learning Deep Spatial Lung Features by 3D Convolutional Neural Network for Early Cancer Detection , 2017, 2017 International Conference on Digital Image Computing: Techniques and Applications (DICTA).

[19]  Jorge S. Marques,et al.  Diagnosis of Alzheimer's disease using 3D local binary patterns , 2013, Comput. methods Biomech. Biomed. Eng. Imaging Vis..

[20]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[21]  Stephen M. Moore,et al.  The Cancer Imaging Archive (TCIA): Maintaining and Operating a Public Information Repository , 2013, Journal of Digital Imaging.

[22]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[23]  David Grant Colburn Hildebrand,et al.  Finding Mirror Symmetry via Registration and Optimal Symmetric Pairwise Assignment of Curves , 2017, 2017 IEEE International Conference on Computer Vision Workshops (ICCVW).

[24]  Jorge S. Marques,et al.  Alternative feature extraction methods in 3D brain image-based diagnosis of Alzheimer's Disease , 2012, 2012 19th IEEE International Conference on Image Processing.

[25]  Isabelle Bloch,et al.  Evaluation of the symmetry plane in 3D MR brain images , 2003, Pattern Recognit. Lett..

[26]  N. Otsu A threshold selection method from gray level histograms , 1979 .

[27]  Yanxi Liu,et al.  A classification based similarity metric for 3D image retrieval , 1998, Proceedings. 1998 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (Cat. No.98CB36231).

[28]  David L. Olson,et al.  Advanced Data Mining Techniques , 2008 .

[29]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[30]  Geoffrey E. Hinton,et al.  Rectified Linear Units Improve Restricted Boltzmann Machines , 2010, ICML.

[31]  David M. W. Powers,et al.  Evaluation: from precision, recall and F-measure to ROC, informedness, markedness and correlation , 2011, ArXiv.

[32]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[33]  Muhammet Üsame Öziç,et al.  T-test feature ranking based 3D MR classification with VBM mask , 2017, 2017 25th Signal Processing and Communications Applications Conference (SIU).

[34]  Guixia Kang,et al.  Multiview convolutional neural networks for lung nodule classification , 2017, Int. J. Imaging Syst. Technol..