On-the-Fly Guidance Training for Medical Image Registration
This study introduces a novel On-the-Fly Guidance (OFG) training framework for enhancing existing learning-based image registration models, addressing the limitations of weakly-supervised and unsupervised methods. Weakly-supervised methods struggle due to the scarcity of labeled data, and unsupervised methods directly depend on image similarity metrics for accuracy. Our method proposes a supervised fashion for training registration models, without the need for any labeled data. OFG generates pseudo-ground truth during training by refining deformation predictions with a differentiable optimizer, enabling direct supervised learning. OFG optimizes deformation predictions efficiently, improving the performance of registration models without sacrificing inference speed. Our method is tested across several benchmark datasets and leading models, it significantly enhanced performance, providing a plug-and-play solution for training learning-based registration models. Code available at: https://github.com/cilix-ai/on-the-fly-guidance
Code (2)
Tasks
Image RegistrationMedical Image RegistrationSimilar Papers 제목 키워드 기반
General Vision Encoder Features as Guidance in Medical Image Registration
General vision encoders like DINOv2 and SAM have recently transformed computer vision. Even though they are trained on natural images, such encoder models have excelled in medical imaging, e.g., in classification, segmen…
Image RegistrationMedical Image RegistrationMeta-Registration: Learning Test-Time Optimization for Single-Pair Image Registration
Neural networks have been proposed for medical image registration by learning, with a substantial amount of training data, the optimal transformations between image pairs. These trained networks can further be optimized …
Image RegistrationMedical Image RegistrationMeta-LearningBIRNet: Brain Image Registration Using Dual-Supervised Fully Convolutional Networks
In this paper, we propose a deep learning approach for image registration by predicting deformation from image appearance. Since obtaining ground-truth deformation fields for training can be challenging, we design a full…
Image RegistrationSAME++: A Self-supervised Anatomical eMbeddings Enhanced medical image registration framework using stable sampling and regularized transformation
Image registration is a fundamental medical image analysis task. Ideally, registration should focus on aligning semantically corresponding voxels, i.e., the same anatomical locations. However, existing methods often opti…
Image RegistrationMedical Image AnalysisMedical Image RegistrationDeep Learning for Medical Image Registration: A Comprehensive Review
Image registration is a critical component in the applications of various medical image analyses. In recent years, there has been a tremendous surge in the development of deep learning (DL)-based medical image registrati…
Deep LearningGenerative Adversarial NetworkImage RegistrationMedical Image Registration