On-the-Fly Guidance Training for Medical Image Registration

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xin, Yuelin, Chen, Yicheng, Ji, Shengxiang, Han, Kun, Xie, Xiaohui
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917719874469888
author Xin, Yuelin
Chen, Yicheng
Ji, Shengxiang
Han, Kun
Xie, Xiaohui
author_facet Xin, Yuelin
Chen, Yicheng
Ji, Shengxiang
Han, Kun
Xie, Xiaohui
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2308_15216
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle On-the-Fly Guidance Training for Medical Image Registration
Xin, Yuelin
Chen, Yicheng
Ji, Shengxiang
Han, Kun
Xie, Xiaohui
Computer Vision and Pattern Recognition
I.4.9
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
title On-the-Fly Guidance Training for Medical Image Registration
topic Computer Vision and Pattern Recognition
I.4.9
url https://arxiv.org/abs/2308.15216