How much data is needed to train a medical image deep learning system to achieve necessary high accuracy?
Open accessAbstract
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.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
The neural network method critically depends on relatively larger training and test datasets compared to image processing methods.
“Hence, the neural network method critically depends on relatively larger training and test datasets compared to image processing methods [52-54].”
From Automated Defect Detection and Decision-Support in Gas Turbine Blade Inspection (2021) · cites this paper as [53]
The Results and Discussions (p. 6) report that larger training sets improved this CNN’s classification accuracy, supporting the training-data portion. The paper does not establish that neural networks require larger training or test datasets than image-processing methods; it reports using 6,000 test images but makes no comparative claim about test-set requirements. (LLM, conf=0.86)
Experimental — accuracy not yet measuredNeural networks require significant amounts of data to accurately perform a classification, especially with increasing number of classes.
“However, they require significant amounts of data to accurately perform a classification, especially with increasing number of classes [52-54].”
From Automated Defect Detection and Decision-Support in Gas Turbine Blade Inspection (2021) · cites this paper as [53]
The Learning Curve section (p. 4) says classification accuracy depends heavily on training-set size, and Results and Discussions (p. 6) estimates that this classifier needs 4,092 images per class to reach its target accuracy. The study uses six fixed anatomical classes; it does not test whether increasing the number of classes increases the amount of data needed. (LLM, conf=0.86)
Experimental — accuracy not yet measured
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Improving Computer-Aided Detection UsingConvolutional Neural Networks and Random View Aggregation
Roth, H. R., Lu, L., Liu, J., Yao, J., Seff, A., Cherry, K., Kim, L., & Summers, R. M. (2016) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2015.2482920 · Cited by 608 · cited 4× · see citing works
“…medical image deep learning publications (Anavi et al. (2015), Havaei et al. (2015),Bar et al. (2015), Roth et al. (2015a), Yan et al. (2015), Roth et al. (2015c), Shin et al. (2015), Wang et al. (2015), Xu et al. (2015), Srivastava et al. (2015), Srivastava et al. (2015), Zhang et al.…”
BackgroundMethodInfluential
Flagged first, then by citations
Research works printed but not linked: 22
- Anavi, Yaron, Kogan, Ilya, Gelbart, Elad, Geva, Ofer, and Greenspan, Hayit. A comparative study for chest radiograph image retrieval using binary texture and deep learning classification. In Engineering in Medicine and Biology Society (EMBC), 2015 37th Annual International Conference of the IEEE , pp. 2940-2943. IEEE, 2015.
- Bar, Yaniv, Diamant, Idit, Wolf, Lior, and Greenspan, Hayit. Deep learning with non-medical training used for chest pathology identification. In SPIE Medical Imaging , pp. 94140V-94140V. International Society for Optics and Photonics, 2015.
- Dennis Jr, John E, Gay, David M, and Walsh, Roy E. An adaptive nonlinear least-squares algorithm. ACM Transactions on Mathematical Software (TOMS) , 7(3):348-368, 1981.
- Dobbin, Kevin K and Simon, Richard M. Sample size planning for developing classifiers using high-dimensional dna microarray data. Biostatistics , 8(1):101-117, 2007.
- Dobbin, Kevin K, Zhao, Yingdong, and Simon, Richard M. How large a training set is needed to develop a classifier for microarray data? Clinical Cancer Research , 14(1):108-114, 2008.
- Figueroa, Rosa L, Zeng-Treitler, Qing, Kandula, Sasikiran, and Ngo, Long H. Predicting sample size required for classification performance. BMC medical informatics and decision making , 12 (1):8, 2012.
- Havaei, Mohammad, Davy, Axel, Warde-Farley, David, Biard, Antoine, Courville, Aaron, Bengio, Yoshua, Pal, Chris, Jodoin, Pierre-Marc, and Larochelle, Hugo. Brain tumor segmentation with deep neural networks. arXiv preprint arXiv:1505.03540 , 2015. · arXiv
- Jia, Yangqing, Shelhamer, Evan, Donahue, Jeff, Karayev, Sergey, Long, Jonathan, Girshick, Ross, Guadarrama, Sergio, and Darrell, Trevor. Caffe: Convolutional architecture for fast feature embedding. In Proceedings of the ACM International Conference on Multimedia , pp. 675-678. ACM, 2014.
- Krizhevsky, Alex, Sutskever, Ilya, and Hinton, Geoffrey E. Imagenet classification with deep convolutional neural networks. In Advances in neural information processing systems , pp. 1097-1105, 2012.
- Margeta, Jan, Criminisi, Antonio, Cabrera Lozoya, R, Lee, Daniel C, and Ayache, Nicholas. Finetuned convolutional neural nets for cardiac mri acquisition plane recognition. Computer Methods in Biomechanics and Biomedical Engineering: Imaging & Visualization , pp. 1-11, 2015.
- Roth, Holger R, Lee, Christopher T, Shin, Hoo-Chang, Seff, Ari, Kim, Lauren, Yao, Jianhua, Lu, Le, and Summers, Ronald M. Anatomy-specific classification of medical images using deep convolutional nets. arXiv preprint arXiv:1504.04003 , 2015a. · arXiv
- Roth, Holger R, Yao, Jianhua, Lu, Le, Stieger, James, Burns, Joseph E, and Summers, Ronald M. Detection of sclerotic spine metastases via random aggregation of deep convolutional neural network classifications. In Recent Advances in Computational Methods and Clinical Applications for Spine Imaging , pp. 3-12. Springer, 2015c.
- Shin, Hoo-Chang, Lu, Le, Kim, Lauren, Seff, Ari, Yao, Jianhua, and Summers, Ronald M. Interleaved text/image deep mining on a large-scale radiology database for automated image interpretation. arXiv preprint arXiv:1505.00670 , 2015. · arXiv
- Shiraishi, Junji, Li, Qiang, Appelbaum, Daniel, and Doi, Kunio. Computer-aided diagnosis and artificial intelligence in clinical imaging. In Seminars in nuclear medicine , volume 41, pp. 449462. Elsevier, 2011.
- Simonyan, Karen and Zisserman, Andrew. Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556 , 2014. · arXiv
- Srivastava, Ruchir, Cheng, Jun, Wong, Damon WK, and Liu, Jiang. Using deep learning for robustness to parapapillary atrophy in optic disc segmentation. In Biomedical Imaging (ISBI), 2015 IEEE 12th International Symposium on , pp. 768-771. IEEE, 2015.
- Szegedy, Christian, Liu, Wei, Jia, Yangqing, Sermanet, Pierre, Reed, Scott, Anguelov, Dragomir, Erhan, Dumitru, Vanhoucke, Vincent, and Rabinovich, Andrew. Going deeper with convolutions. arXiv preprint arXiv:1409.4842 , 2014. · arXiv
- Wang, Jiazhuo, MacKenzie, John D, Ramachandran, Rageshree, and Chen, Danny Z. Neutrophils identification by deep learning and voronoi diagram of clusters. In Medical Image Computing and Computer-Assisted Intervention?MICCAI 2015 , pp. 226-233. Springer, 2015.
- Xu, Yan, Jia, Zhipeng, Ai, Yuqing, Zhang, Fang, Lai, Maode, Chang, Eric I, et al. Deep convolutional activation features for large scale brain tumor histopathology image classification and segmentation. In Acoustics, Speech and Signal Processing (ICASSP), 2015 IEEE International Conference on , pp. 947-951. IEEE, 2015.
- Yan, Zhennan, Zhan, Yiqiang, Peng, Zhigang, Liao, Shu, Shinagawa, Yoshihisa, Metaxas, Dimitris N, and Zhou, Xiang Sean. Bodypart recognition using multi-stage deep learning. In Information Processing in Medical Imaging , pp. 449-461. Springer, 2015.
- Ypsilantis, Petros-Pavlos, Siddique, Musib, Sohn, Hyon-Mok, Davies, Andrew, Cook, Gary, Goh, Vicky, and Montana, Giovanni. Predicting response to neoadjuvant chemotherapy with pet imaging using convolutional neural networks. PloS one , 10(9):e0137036, 2015.
- Zeiler, Matthew D and Fergus, Rob. Visualizing and understanding convolutional networks. In Computer Vision-ECCV 2014 , pp. 818-833. Springer, 2014.
Cited by
Sorted by citations
Deep learning for brain MRI segmentation: state of the art and future directions
Akkus, Z., Galimzianova, A., Hoogi, A., Rubin, D. L., & Erickson, B. J. (2017) · Journal of Digital Imaging · doi:10.1007/s10278-017-9983-4 · Cited by 1,129 · see citing works
Rasmy, L., Xiang, Y., Xie, Z., Tao, C., & Zhi, D. (2021) · npj Digital Medicine · doi:10.1038/s41746-021-00455-y · Cited by 886 · see citing works
Thrall, J. H., Li, X., Li, Q., Cruz, C., Do, S., Dreyer, K., & Brink, J. (2018) · Journal of the American College of Radiology · doi:10.1016/j.jacr.2017.12.026 · Cited by 710 · see citing works
Self-citationAlwosheel, A., van Cranenburgh, S., & Chorus, C. G. (2018) · Journal of Choice Modelling · doi:10.1016/j.jocm.2018.07.002 · Cited by 524 · see citing works
A scoping review of transfer learning research on medical image analysis using ImageNet
Morid, M. A., Borjali, A., & Del Fiol, G. (2021) · Computers in Biology and Medicine · doi:10.1016/j.compbiomed.2020.104115 · Cited by 508 · see citing works
Balki, I., Amirabadi, A., Levman, J., Martel, A. L., Emersic, Z., Meden, B., Garcia-Pedrero, A., Ramirez, S. C., Kong, D., Moody, A. R., et al. (2019) · Canadian Association of Radiologists Journal · doi:10.1016/j.carj.2019.06.002 · Cited by 299 · see citing works
Cheng, P. M., & Malhi, H. S. (2017) · Journal of Digital Imaging · doi:10.1007/s10278-016-9929-2 · Cited by 253 · see citing works
Shahinfar, S., Meek, P., & Falzon, G. (2020) · Ecological Informatics · doi:10.1016/j.ecoinf.2020.101085 · Cited by 229 · see citing works
Joo, S., Ko, E. S., Kwon, S., Jeon, E., Jung, H., Kim, J.-Y., Chung, M. J., & Im, Y.-H. (2021) · Scientific Reports · doi:10.1038/s41598-021-98408-8 · Cited by 137 · see citing works
Automated Defect Detection and Decision-Support in Gas Turbine Blade Inspection
Aust, J., Shankland, S., Pons, D., Mukundan, R., & Mitrovic, A. (2021) · Aerospace · doi:10.3390/aerospace8020030 · Cited by 66 · see citing works
Machine learning modeling practices to support the principles of AI and ethics in nutrition research
Thomas, D. M., Kleinberg, S., Brown, A. W., Crow, M., Bastian, N. D., Reisweber, N., Lasater, R., Kendall, T., Shafto, P., Blaine, R., et al. (2022) · Nutrition & Diabetes · doi:10.1038/s41387-022-00226-y · Cited by 48 · see citing works
- Authors
- Cho, J., Lee, K., Shin, E., Choy, G., & Do, S.
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2015
- ISSN
- 2331-8422 (ISSN-L)
- arXiv
- 1511.06348
- OpenAlex
- W2175082661
- Cited by
- 268
- References
- 1
- Citations by year
- 2026: 1 · 2025: 7 · 2024: 39 · 2023: 37 · 2022: 32 · 2021: 48 · 2020: 50 · 2019: 28 · 2018: 15 · 2017: 7 · 2016: 4
- Type
- posted content
- Language
- en
- Open access
- green
- arxiv.org
- Locations
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Topics
- Health Sciences › Medicine › Health Informatics › Artificial Intelligence in Healthcare and Education
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
- Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
- Keywords
- training data size, medical image classification, convolutional neural networks, learning curve, anatomical classification
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review