An Automated Method for Generating Training Sets for Deep Learning based Image Registration

Masato Ito, Fumihiko Ino

2018

Abstract

In this paper, we propose an automated method for generating training sets required for realizing deep learning based image registration. The proposed method minimizes effort for supervised learning by automatically generating thousands of training sets from a small number of seed sets, i.e., tens of deformation vector fields obtained with a conventional registration method. To automate this procedure, we solve an inverse problem instead of a direct problem; we produce a floating image by applying a deformation vector fieldFto a reference image and let the inverse vector of F be the ground truth for these images. In experiments, the proposed method took 33 minutes to produce 169,890 training sets from approximately 670,000 2-D magnetic resonance (MR) images and 30 seed sets. We further trained GoogLeNet with these training sets and performed holdout validation to compare the proposed method with the conventional registration method in terms of recall and precision. As a result, the proposed method increased recall and precision from 50% to 80%, demonstrating the impact of deep learning for image registration problems.

Paper Citation

in Harvard Style

Ito M. and Ino F. (2018). An Automated Method for Generating Training Sets for Deep Learning based Image Registration. In Proceedings of the 11th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2018) - Volume 2: BIOIMAGING; ISBN 978-989-758-278-3, SciTePress, pages 140-147. DOI: 10.5220/0006634501400147

in Bibtex Style

@conference{bioimaging18,
author={Masato Ito and Fumihiko Ino},
title={An Automated Method for Generating Training Sets for Deep Learning based Image Registration},
booktitle={Proceedings of the 11th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2018) - Volume 2: BIOIMAGING},
year={2018},
pages={140-147},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0006634501400147},
isbn={978-989-758-278-3},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 11th International Joint Conference on Biomedical Engineering Systems and Technologies (BIOSTEC 2018) - Volume 2: BIOIMAGING
TI - An Automated Method for Generating Training Sets for Deep Learning based Image Registration
SN - 978-989-758-278-3
AU - Ito M.
AU - Ino F.
PY - 2018
SP - 140
EP - 147
DO - 10.5220/0006634501400147
PB - SciTePress