Longitudinal brain alignment is critical for disease monitoring and adaptive treatment planning in glioblastoma (GBM) patients. However, the current methods are either non-adaptive to pathological brains, or time and laborintensive. Here, we aim to develop a novel deep-learning-based framework for longitudinal postoperative brain GBM scan registration. The proposed pathology adaptive registration framework (PARF) adopts a double UNET architecture: a 2D 7-level UNET, NETseg, for pathology segmentation, and a 3D 5-level UNET, NETreg, for unsupervised image registration, connected through a spatial transformer and a volume combiner. NETseg was first trained separately and then combined with NETreg for pathology adaptive registration training. In aggregated registration testing of PARF, 36 registrations from 18 intra-subject pairs of post-operative follow-up MR scans were selected, and the results were compared to those from current state-of-the-art methods as well as non-adaptive NETreg alone. PARF is significantly faster and more accurate than comparison methods, in terms of sum-of-squared differences, segmentation alignment dice coefficients, and landmark mislignment errors. PARF may pave the path for various clinical and research applications that depend on the accurate registration of GBM longitudinal images.
|