Capturing Longitudinal Changes in Brain Morphology Using Temporally Parameterized Neural Displacement Fields.

Aisha L. Shuaibu, Kieran A. Gibb, Peter A. Wijeratne, Ivor J A Simpson

Keywords: Longitudinal image registration, spatio-temporal regularization, monotonic regularization, implicit neural representations.

TL;DR: Capturing longitudinal changes in brain morphology using temporally parameterized neural displacement fields, while introducing a novel longitudinal regularization term that enforces monotonic rate of change over time.

Abstract: Longitudinal image registration enables studying temporal changes in brain morphology which is useful in applications where monitoring the growth or atrophy of specific structures is important. However this task is challenging due to; noise/artifacts in the data and quantifying small anatomical changes between sequential scans. We propose a novel longitudinal registration method that models structural changes using temporally parameterized neural displacement fields. Specifically, we implement an implicit neural representation (INR) using a multi-layer perceptron that serves as a continuous coordinate-based approximation of the deformation field at any time point. In effect, for any $N$ scans of a particular subject, our model takes as input a 3D spatial coordinate location $x, y, z$ and a corresponding temporal representation $t$ and learns to describe the continuous morphology of structures for both observed and unobserved points in time. Furthermore, we leverage the analytic derivatives of the INR to derive a new regularization function that enforces monotonic rate of change in the trajectory of the voxels, which is shown to provide more biologically plausible patterns. We demonstrate the effectiveness of our method on 4D brain MR registration. Our code is publicly available here https://github.com/aisha-lawal/inrmorph

Primary Subject Area: Image Registration

Secondary Subject Area: Image Registration

Paper Type: Methodological Development

Registration Requirement: Yes

Reproducibility: https://github.com/aisha-lawal/inrmorph

Visa & Travel: Yes

Submission Number: 74