Wang Wenying, Li Junyuan, Tivnan Matthew, Stayman J Webster, Gang Grace J
Department of Biomedical Engineering, Johns Hopkins University, Baltimore, Maryland, USA.
Proc SPIE Int Soc Opt Eng. 2022 Feb-Mar;12031. doi: 10.1117/12.2612569. Epub 2022 Apr 4.
Recent years have seen the increasing application of deep learning methods in medical imaging formation, processing, and analysis. These methods take advantage of the flexibility of nonlinear neural network models to encode information and features in ways that can outperform conventional approaches. However, because of the nonlinear nature of this processing, images formed by neural networks have properties that are highly data-dependent and difficult to analyze. In particular, the generalizability and robustness of these approaches can be difficult to ascertain. In this work, we analyze a class of neural networks that use only piece-wise linear activation functions. This class of networks can be represented by locally linear systems where the linear properties are highly data-dependent - allowing, for example, estimation of noise in image output via standard propagation methods. We propose a nonlinearity index metric that quantifies the fidelity of a local linear approximation of trained models based on specific input data. We applied this analysis to three example CT denoising CNNs to analytically predict the noise properties in the output images. We found that the proposed nonlinearity metric highly correlates with the accuracy of noise predictions. The analysis proposed in this work provides theoretical understanding of the nonlinear behavior of neural networks and enables performance prediction and quantitation under certain conditions.
近年来,深度学习方法在医学成像的形成、处理和分析中的应用日益增加。这些方法利用非线性神经网络模型的灵活性,以优于传统方法的方式对信息和特征进行编码。然而,由于这种处理的非线性性质,神经网络形成的图像具有高度依赖数据且难以分析的特性。特别是,这些方法的通用性和鲁棒性可能难以确定。在这项工作中,我们分析了一类仅使用分段线性激活函数的神经网络。这类网络可以由局部线性系统表示,其中线性特性高度依赖数据——例如,允许通过标准传播方法估计图像输出中的噪声。我们提出了一种非线性指数度量,它基于特定输入数据量化训练模型的局部线性近似的保真度。我们将此分析应用于三个示例CT去噪卷积神经网络,以分析预测输出图像中的噪声特性。我们发现,所提出的非线性度量与噪声预测的准确性高度相关。这项工作中提出的分析提供了对神经网络非线性行为的理论理解,并能够在某些条件下进行性能预测和量化。