Medical image augmentation for lesion detection using a texture-constrained multichannel progressive GAN

被引:98
作者
Guan, Qiu [1 ]
Chen, Yizhou [1 ]
Wei, Zihan [1 ]
Heidari, Ali Asghar [2 ]
Hu, Haigen [1 ]
Yang, Xu-Hua [1 ]
Zheng, Jianwei [1 ]
Zhou, Qianwei [1 ]
Chen, Huiling [3 ]
Chen, Feng [4 ]
机构
[1] Zhejiang Univ Technol, Coll Comp Sci & Technol, Hangzhou, Peoples R China
[2] Univ Tehran, Coll Engn, Sch Surveying & Geospatial Engn, Tehran, Iran
[3] Wenzhou Univ, Coll Comp Sci & Artificial Intelligence, Wenzhou 325035, Zhejiang, Peoples R China
[4] Zhejiang Univ, Affiliated Hosp 1, Sch Med, Hangzhou, Peoples R China
基金
中国国家自然科学基金;
关键词
Medical image augmentation; Lesion detection; Texture feature; Generative adversarial network; ARTIFICIAL-INTELLIGENCE;
D O I
10.1016/j.compbiomed.2022.105444
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
Lesion detectors based on deep learning can assist doctors in diagnosing diseases. However, the performance of current detectors is likely to be unsatisfactory due to the scarcity of training samples. Therefore, it is beneficial to use image generation to augment the training set of a detector. However, when the imaging texture of the medical image is relatively delicate, the synthesized image generated by an existing method may be too poor in quality to meet the training requirements of the detectors. In this regard, a medical image augmentation method, namely, a texture-constrained multichannel progressive generative adversarial network (TMP-GAN), is proposed in this work. TMP-GAN uses joint training of multiple channels to effectively avoid the typical shortcomings of the current generation methods. It also uses an adversarial learning-based texture discrimination loss to further improve the fidelity of the synthesized images. In addition, TMP-GAN employs a progressive generation mechanism to steadily improve the accuracy of the medical image synthesizer. Experiments on the publicly available dataset CBIS-DDMS and our pancreatic tumor dataset show that the precision/recall/F1-score of the detector trained on the TMP-GAN augmented dataset improves by 2.59%/2.70%/2.77% and 2.44%/2.06%/ 2.36%, respectively, compared to the optimal results of other data augmentation methods. The FROC curve of the detector is also better than the curve from the contrast-augmented trained dataset. Therefore, we believe the proposed TMP-GAN is a practical technique to efficiently implement lesion detection case studies.
引用
收藏
页数:14
相关论文
共 36 条
[1]   Convolutional neural networks for breast cancer detection in mammography: A survey [J].
Abdelrahman, Leila ;
Al Ghamdi, Manal ;
Collado-Mesa, Fernando ;
Abdel-Mottaleb, Mohamed .
COMPUTERS IN BIOLOGY AND MEDICINE, 2021, 131
[2]   Applying deep learning in digital breast tomosynthesis for automatic breast cancer detection: A review [J].
Bai, Jun ;
Posner, Russell ;
Wang, Tianyu ;
Yang, Clifford ;
Nabavi, Sheida .
MEDICAL IMAGE ANALYSIS, 2021, 71
[3]   Domain adversarial networks and intensity-based data augmentation for male pelvic organ segmentation in cone beam CT [J].
Brion, Eliott ;
Leger, Jean ;
Barragan-Montero, A. M. ;
Meert, Nicolas ;
Lee, John A. ;
Macq, Benoit .
COMPUTERS IN BIOLOGY AND MEDICINE, 2021, 131
[4]   Semi-supervised task-driven data augmentation for medical image segmentation [J].
Chaitanya, Krishna ;
Karani, Neerav ;
Baumgartner, Christian F. ;
Erdil, Ertunc ;
Becker, Anton ;
Donati, Olivio ;
Konukoglu, Ender .
MEDICAL IMAGE ANALYSIS, 2021, 68
[5]   Diverse data augmentation for learning image segmentation with cross-modality annotations [J].
Chen, Xu ;
Lian, Chunfeng ;
Wang, Li ;
Deng, Hannah ;
Kuang, Tianshu ;
Fung, Steve H. ;
Gateno, Jaime ;
Shen, Dinggang ;
Xia, James J. ;
Yap, Pew-Thian .
MEDICAL IMAGE ANALYSIS, 2021, 71
[6]   Challenges and opportunities for artificial intelligence in oncological imaging [J].
Cheung, H. M. C. ;
Rubin, D. .
CLINICAL RADIOLOGY, 2021, 76 (10) :728-736
[7]   Assisting Barrett's esophagus identification using endoscopic data augmentation based on Generative Adversarial Networks [J].
de Souza Jr, Luis A. ;
Passos, Leandro A. ;
Mendel, Robert ;
Ebigbo, Alanna ;
Probst, Andreas ;
Messmann, Helmut ;
Palm, Christoph ;
Papa, Joao P. .
COMPUTERS IN BIOLOGY AND MEDICINE, 2020, 126
[8]   Artificial intelligence in oncology: From bench to clinic [J].
Elkhader, Jamal ;
Elemento, Olivier .
SEMINARS IN CANCER BIOLOGY, 2022, 84 :113-128
[9]   Machine learning applications in radiation oncology [J].
Field, Matthew ;
Hardcastle, Nicholas ;
Jameson, Michael ;
Aherne, Noel ;
Holloway, Lois .
PHYSICS & IMAGING IN RADIATION ONCOLOGY, 2021, 19 :13-24
[10]   Enabling Data Diversity: Efficient Automatic Augmentation via Regularized Adversarial Training [J].
Gao, Yunhe ;
Tang, Zhiqiang ;
Zhou, Mu ;
Metaxas, Dimitris .
INFORMATION PROCESSING IN MEDICAL IMAGING, IPMI 2021, 2021, 12729 :85-97