arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.01693q-bio.QMcs.CVcs.LGeess.IVstat.ML

A Comprehensive Study of Data Augmentation Strategies for Prostate Cancer Detection in Diffusion-weighted MRI using Convolutional Neural Networks

University of Electronic Science and Technology of China(电子科技大学);University of Toronto(多伦多大学);The Hospital for Sick Children (SickKids)(SickKids 儿童医院(病童医院));Sinai Health System(西奈卫生系统);University Health Network(大学健康网络);Lunenfeld-Tanenbaum Research Institute(卢嫩菲尔德-塔南鲍姆研究所);Sunnybrook Research Institute(桑尼布鲁克研究所)

Ruqian Hao, Khashayar Namdar, Lin Liu, Masoom A. Haider, Farzad Khalvati

英文摘要:

Data augmentation refers to a group of techniques whose goal is to battle limited amount of available data to improve model generalization and push sample distribution toward the true distribution. While different augmentation strategies and their combinations have been investigated for various computer vision tasks in the context of deep learning, a specific work in the domain of medical imaging is rare and to the best of our knowledge, there has been no dedicated work on exploring the effects of various augmentation methods on the performance of deep learning models in prostate cancer detection. In this work, we have statically applied five most frequently used augmentation techniques (random rotation, horizontal flip, vertical flip, random crop, and translation) to prostate Diffusion-weighted Magnetic Resonance Imaging training dataset of 217 patients separately and evaluated the effect of each method on the accuracy of prostate cancer detection. The augmentation algorithms were applied independently to each data channel and a shallow as well as a deep Convolutional Neural Network (CNN) were trained on the five augmented sets separately. We used Area Under Receiver Operating Characteristic (ROC) curve (AUC) to evaluate the performance of the trained CNNs on a separate test set of 95 patients, using a validation set of 102 patients for finetuning. The shallow network outperformed the deep network with the best 2D slice-based AUC of 0.85 obtained by the rotation method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑