Unsupervised deep hashing by joint optimization for pulmonary nodule image retrieval

Abstract In recent years, hash-based image retrieval has attracted great attention due to the rapid growth of medical images. In the paper, an end-to-end unsupervised deep hashing is proposed, where feature extraction and binary optimization are carried out by joint optimization. Our method consists of five components: a shared deep convolution neural network for learning image representations, a deconvolution module for reconstructing the original images, a classification module for leveraging semantic supervision by pseudo labels, a binary code learning module for encoding images features into binary codes, and a joint loss function for deep hash function learning. In addition, the real-valued features balanced in different dimensions by a rotation matrix are quantized directly into discrete binary codes in an alternating optimization approach to minimize the quantization loss. Experiments have been performed on the pulmonary nodule images dataset and the results demonstrate the proposed method can yield better retrieval performance by comparing with the state-of-the-art unsupervised hashing methods.

[1]  Yan Gao,et al.  A new method of content based medical image retrieval and its applications to CT imaging sign retrieval , 2017, J. Biomed. Informatics.

[2]  Geoffrey E. Hinton,et al.  Semantic hashing , 2009, Int. J. Approx. Reason..

[3]  Atsuto Maki,et al.  A systematic study of the class imbalance problem in convolutional neural networks , 2017, Neural Networks.

[4]  Lijun Xie,et al.  A regularized ensemble framework of deep learning for cancer detection from multi-class, imbalanced training data , 2018, Pattern Recognit..

[5]  Mohamed Abouelenien,et al.  A multi-class boosting method for learning from imbalanced data , 2016, Int. J. Granul. Comput. Rough Sets Intell. Syst..

[6]  Cordelia Schmid,et al.  Product Quantization for Nearest Neighbor Search , 2011, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[7]  Wei Qian,et al.  Content-based image retrieval for Lung Nodule Classification Using Texture Features and Learned Distance Metric , 2017, Journal of Medical Systems.

[8]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[9]  Richard C. Pais,et al.  The Lung Image Database Consortium (LIDC) and Image Database Resource Initiative (IDRI): a completed reference database of lung nodules on CT scans. , 2011, Medical physics.

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

[11]  Kaare Brandt Petersen,et al.  The Matrix Cookbook , 2006 .

[12]  Alexandr Andoni,et al.  Near-Optimal Hashing Algorithms for Approximate Nearest Neighbor in High Dimensions , 2006, 2006 47th Annual IEEE Symposium on Foundations of Computer Science (FOCS'06).

[13]  Gjorgji Madjarov,et al.  Multimodal medical image retrieval system , 2016, Multimedia Tools and Applications.