arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1511.06348cs.LGcs.CVcs.NE
收藏

How much data is needed to train a medical image deep learning system to achieve necessary high accuracy?

Massachusetts General Hospital(麻省总医院);Harvard Medical School(哈佛医学院)

Junghwan Cho, Kyewook Lee, Ellie Shin, Garry Choy, Synho Do

英文摘要:

The use of Convolutional Neural Networks (CNN) in natural image classification systems has produced very impressive results. Combined with the inherent nature of medical images that make them ideal for deep-learning, further application of such systems to medical image classification holds much promise. However, the usefulness and potential impact of such a system can be completely negated if it does not reach a target accuracy. In this paper, we present a study on determining the optimum size of the training data set necessary to achieve high classification accuracy with low variance in medical image classification systems. The CNN was applied to classify axial Computed Tomography (CT) images into six anatomical classes. We trained the CNN using six different sizes of training data set (5, 10, 20, 50, 100, and 200) and then tested the resulting system with a total of 6000 CT images. All images were acquired from the Massachusetts General Hospital (MGH) Picture Archiving and Communication System (PACS). Using this data, we employ the learning curve approach to predict classification accuracy at a given training sample size. Our research will present a general methodology for determining the training data set size necessary to achieve a certain target classification accuracy that can be easily applied to other problems within such systems.

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

↑