Self-Distilled Hierarchical Network for Unsupervised Deformable Image Registration

Area of Science:

• Medical Imaging
• Computer Vision
• Machine Learning

Background:

• Unsupervised deformable image registration is crucial for medical image analysis.
• Existing progressive networks have limitations in handling multi-scale deformations and long-range dependencies.

Purpose of the Study:

• To develop a novel unsupervised learning approach for deformable image registration.
• To address the limitations of existing progressive networks by introducing hierarchical deformation fields and a self-distillation scheme.

Main Methods:

• The Self-Distilled Hierarchical Network (SDHNet) decomposes registration into iterations, generating hierarchical deformation fields (HDFs) simultaneously.
• HDFs are generated using parallel gated recurrent units and adaptively fused.
• A self-deformation distillation scheme constrains intermediate fields in deformation-value and gradient spaces.

Main Results:

• SDHNet demonstrates superior performance compared to state-of-the-art methods on five benchmark datasets (brain MRI, liver CT).
• The proposed method achieves faster inference speed and requires less GPU memory.
• Experiments validate the effectiveness of the hierarchical structure and self-distillation scheme.

Conclusions:

• SDHNet offers an effective and efficient solution for unsupervised deformable image registration.
• The novel approach advances the field by incorporating hierarchical representations and self-supervision.
• The method shows promise for various medical imaging applications.