Improving PET-CT Image Segmentation via Deep Multi-modality Data Augmentation

Positron emission tomography (PET) - computed tomography (CT) is a widely-accepted imaging modality for staging, diagnosis and treatment response monitoring of cancers. Deep learning based computer aided diagnosis systems have achieved high accuracy on tumor segmentation on PET-CT images in recent years. PET images can be used to detect functional structures such as tumors, whilst CT images provide complementary anatomical information. As for tumor detection using deep learning methods, multi-modality segmentation was verified to be effective. In this work, we propose a generative adversarial network (GAN) based augmentation method to synthesized multi-modality data pairs on PET and CT to improve the training of multi-modality segmentation method. Our novelty lies in creating a semantic label augmentation method to provide latent information that is suitable for the multi-modality synthesis. In addition, we set out a ‘Split U’ structure which can generate both PET-CT modalities from a latent input. Our experimental results demonstrated that the synthesized images generated by our method can be used to augment the training data for PET-CT segmentation.

[1]  Jun Tang,et al.  A color image segmentation algorithm based on region growing , 2010, 2010 2nd International Conference on Computer Engineering and Technology.

[2]  Hu,et al.  Survey of Recent Volumetric Medical Image Segmentation Techniques , 2009 .

[3]  Michał Grochowski,et al.  Data augmentation for improving deep learning in image classification problem , 2018, 2018 International Interdisciplinary PhD Workshop (IIPhDW).

[4]  Djemel Ziou,et al.  Image Quality Metrics: PSNR vs. SSIM , 2010, 2010 20th International Conference on Pattern Recognition.

[5]  Brijesh Verma,et al.  A computer-aided diagnosis system for digital mammograms based on fuzzy-neural and feature extraction techniques , 2001, IEEE Transactions on Information Technology in Biomedicine.

[6]  Luis Perez,et al.  The Effectiveness of Data Augmentation in Image Classification using Deep Learning , 2017, ArXiv.

[7]  Yoshua Bengio,et al.  Generative Adversarial Nets , 2014, NIPS.

[8]  Zhao Feng,et al.  Two-Dimensional Otsu's Curve Thresholding Segmentation Method for Gray-Level Images , 2007 .

[9]  Bradley M. Hemminger,et al.  Contrast Limited Adaptive Histogram Equalization image processing to improve the detection of simulated spiculations in dense mammograms , 1998, Journal of Digital Imaging.

[10]  Keith E. Muller,et al.  Contrast-limited adaptive histogram equalization: speed and effectiveness , 1990, [1990] Proceedings of the First Conference on Visualization in Biomedical Computing.

[11]  Hore,et al.  [IEEE 2010 20th International Conference on Pattern Recognition (ICPR) - Istanbul, Turkey (2010.08.23-2010.08.26)] 2010 20th International Conference on Pattern Recognition - Image Quality Metrics: PSNR vs. SSIM , 2010 .

[12]  Dagan Feng,et al.  Deep multi-modality collaborative learning for distant metastases predication in PET-CT soft-tissue sarcoma studies , 2019, 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[13]  Ali M. Reza,et al.  Realization of the Contrast Limited Adaptive Histogram Equalization (CLAHE) for Real-Time Image Enhancement , 2004, J. VLSI Signal Process..

[14]  I. El Naqa,et al.  A radiomics model from joint FDG-PET and MRI texture features for the prediction of lung metastases in soft-tissue sarcomas of the extremities , 2015, Physics in medicine and biology.

[15]  David Dagan Feng,et al.  Co-Learning Feature Fusion Maps From PET-CT Images of Lung Cancer , 2018, IEEE Transactions on Medical Imaging.

[16]  Lipo Wang,et al.  Deep Learning Applications in Medical Image Analysis , 2018, IEEE Access.

[17]  David Dagan Feng,et al.  Synthesis of Positron Emission Tomography (PET) Images via Multi-channel Generative Adversarial Networks (GANs) , 2017, CMMI/RAMBO/SWITCH@MICCAI.

[18]  Dana Kulic,et al.  Data augmentation of wearable sensor data for parkinson’s disease monitoring using convolutional neural networks , 2017, ICMI.

[19]  Arindam Mandal,et al.  Multi-Task Learning and Weighted Cross-Entropy for DNN-Based Keyword Spotting , 2016, INTERSPEECH.

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

[21]  C. Kratochwil,et al.  Einsatz der PET/CT zur Diagnostik und Therapiestratifizierung des Bronchialkarzinoms , 2010, Der Radiologe.

[22]  Alexei A. Efros,et al.  Image-to-Image Translation with Conditional Adversarial Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[23]  Geoff S. Nitschke,et al.  Improving Deep Learning with Generic Data Augmentation , 2018, 2018 IEEE Symposium Series on Computational Intelligence (SSCI).