修剪
计算机科学
医学影像学
领域(数学)
人工神经网络
人工智能
点(几何)
深层神经网络
图像(数学)
残余物
相(物质)
模式识别(心理学)
机器学习
算法
数学
化学
几何学
有机化学
纯数学
农学
生物
作者
Francisco Erivaldo Fernandes,Gary G. Yen
出处
期刊:IEEE transactions on neural networks and learning systems
[Institute of Electrical and Electronics Engineers]
日期:2021-12-01
卷期号:32 (12): 5664-5674
被引量:36
标识
DOI:10.1109/tnnls.2020.3027308
摘要
The field of medical imaging diagnostic makes use of a modality of imaging tests, e.g., X-rays, ultrasounds, computed tomographies, and magnetic resonance imaging, to assist physicians with the diagnostic of patients' illnesses. Due to their state-of-the-art results in many challenging image classification tasks, deep neural networks (DNNs) are suitable tools for use by physicians to provide diagnostic support when dealing with medical images. To further advance the field, the present work proposes a two-phase algorithm capable of automatically generating compact DNN architectures given a database, called here DNNDeepeningPruning. In the first phase, also called the deepening phase, the algorithm grows a DNN by adding blocks of residual layers one after another until the model overfits the given data. In the second phase, called the pruning phase, the algorithm prunes the created DNN model from the first phase to produce a DNN with a small amount of floating-point operations guided by some preference given by the user. The proposed algorithm unifies the two separate fields of DNN architecture searching and pruning under a single framework, and it is tested in two medical imaging data sets with satisfactory results.
科研通智能强力驱动
Strongly Powered by AbleSci AI