首页    期刊浏览 2025年07月23日 星期三
登录注册

文章基本信息

  • 标题:Hierarchical deep learning models using transfer learning for disease detection and classification based on small number of medical images
  • 本地全文:下载
  • 作者:Guangzhou An ; Masahiro Akiba ; Kazuko Omodaka
  • 期刊名称:Scientific Reports
  • 电子版ISSN:2045-2322
  • 出版年度:2021
  • 卷号:11
  • DOI:10.1038/s41598-021-83503-7
  • 语种:English
  • 出版社:Springer Nature
  • 摘要:Deep learning is being employed in disease detection and classification based on medical images for clinical decision making. It typically requires large amounts of labelled data; however, the sample size of such medical image datasets is generally small. This study proposes a novel training framework for building deep learning models of disease detection and classification with small datasets. Our approach is based on a hierarchical classification method where the healthy/disease information from the first model is effectively utilized to build subsequent models for classifying the disease into its sub-types via a transfer learning method. To improve accuracy, multiple input datasets were used, and a stacking ensembled method was employed for final classification. To demonstrate the method’s performance, a labelled dataset extracted from volumetric ophthalmic optical coherence tomography data for 156 healthy and 798 glaucoma eyes was used, in which glaucoma eyes were further labelled into four sub-types. The average weighted accuracy and Cohen’s kappa for three randomized test datasets were 0.839 and 0.809, respectively. Our approach outperformed the flat classification method by 9.7% using smaller training datasets. The results suggest that the framework can perform accurate classification with a small number of medical images.
国家哲学社会科学文献中心版权所有