Image space formalism of convolutional neural networks for k-space interpolation

TL;DR

This paper introduces an image space formalism for RAKI neural networks to analytically analyze noise propagation and the role of nonlinear activations in k-space interpolation, enhancing understanding of noise resilience and artifacts.

Contribution

The paper presents a novel image space formalism for RAKI, enabling analytical noise propagation analysis and visualization of nonlinear activation effects in k-space interpolation.

Findings

Analytical g-factor maps match Monte Carlo simulations.

Enhanced noise resilience leads to artifacts like blurring and contrast loss.

Adjusting nonlinearity controls noise resilience and artifacts.

Abstract

Purpose: Noise resilience in image reconstructions by scan-specific robust artificial neural networks for k-space interpolation (RAKI) is linked to nonlinear activations in k-space. To gain a deeper understanding of this relationship, an image space formalism of RAKI is introduced for analyzing noise propagation analytically, identifying and characterizing image reconstruction features and to describe the role of nonlinear activations in a human readable manner. Methods: The image space formalism for RAKI inference is employed by expressing nonlinear activations in k-space as element-wise multiplications with activation masks, which transform into convolutions in image space. Jacobians of the de-aliased, coil-combined image relative to the aliased coil images can be expressed algebraically, and thus, the noise amplification is quantified analytically (g-factor maps). We analyze the role of nonlinearity for noise resilience by controlling the degree of nonlinearity in the reconstruction model via the negative slope parameter in leaky ReLU. Results: The analytical g-factor maps correspond with those obtained from Monte Carlo simulations and from an auto differentiation approach for in vivo brain images. Apparent blurring and contrast loss artifacts are identified as implications of enhanced noise resilience. These residual artifacts can be traded against noise resilience by adjusting the degree of nonlinearity in the model (Tikhonov-like regularization) in case of limited training data. The inspection of image space activations reveals an autocorrelation pattern leading to a potential center artifact. Conclusion: The image space formalism of RAKI provides the means for analytical quantitative noisepropagation analysis and human-readable visualization of the effects of the nonlinear activation functions in k-space.