Abstract:We describe an adversarial learning approach to constrain convolutional neural network training for image registration, replacing heuristic smoothness measures of displacement fields often used in these tasks. Using minimally-invasive prostate cancer intervention as an example application, we demonstrate the feasibility of utilizing biomechanical simulations to regularize a weakly-supervised anatomical-label-driven registration network for aligning pre-procedural magnetic resonance (MR) and 3D intra-procedural transrectal ultrasound (TRUS) images. A discriminator network is optimized to distinguish the registration-predicted displacement fields from the motion data simulated by finite element analysis. During training, the registration network simultaneously aims to maximize similarity between anatomical labels that drives image alignment and to minimize an adversarial generator loss that measures divergence between the predicted- and simulated deformation. The end-to-end trained network enables efficient and fully-automated registration that only requires an MR and TRUS image pair as input, without anatomical labels or simulated data during inference. 108 pairs of labelled MR and TRUS images from 76 prostate cancer patients and 71,500 nonlinear finite-element simulations from 143 different patients were used for this study. We show that, with only gland segmentation as training labels, the proposed method can help predict physically plausible deformation without any other smoothness penalty. Based on cross-validation experiments using 834 pairs of independent validation landmarks, the proposed adversarial-regularized registration achieved a target registration error of 6.3 mm that is significantly lower than those from several other regularization methods.

A Brain-Inspired Regularizer for Adversarial Robustness

Learning From Brains How to Regularize Machines

A Neural-Network-Based Convex Regularizer for Inverse Problems

Strengthening Robustness Under Adversarial Attacks Using Brain Visual Codes

A regularization method to improve adversarial robustness of neural networks for ECG signal classification

A Regularization Perspective Based Theoretical Analysis for Adversarial Robustness of Deep Spiking Neural Networks.

Selective Output Smoothing Regularization: Regularize Neural Networks by Softening Output Distributions

On Connections Between Regularizations for Improving DNN Robustness

Analytical Moment Regularizer for Gaussian Robust Networks

Improving the Adversarial Robustness and Interpretability of Deep Neural Networks by Regularizing Their Input Gradients

Robust Sparse Regularization: Defending Adversarial Attacks Via Regularized Sparse Network

Adversarial Deformation Regularization for Training Image Registration Neural Networks

Improving Interpretability via Regularization of Neural Activation Sensitivity

Brain inspired Robust Vision using Convolutional Neural Networks with Feedback

Robust Sparse Regularization: Simultaneously Optimizing Neural Network Robustness and Compactness

Regularization for Adversarial Robust Learning

Dissociable Neural Representations of Adversarially Perturbed Images in Convolutional Neural Networks and the Human Brain

Avoiding Overfitting: A Survey on Regularization Methods for Convolutional Neural Networks

Group-wise Inhibition based Feature Regularization for Robust Classification

Improve Generalization and Robustness of Neural Networks via Weight Scale Shifting Invariant Regularizations

Towards a High Robust Neural Network Via Feature Matching