Saved in:
Bibliographic Details
Main Authors: Watson, Warin, Cherry, Cash, Lang, Rachelle
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.02806
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916876551979008
author Watson, Warin
Cherry, Cash
Lang, Rachelle
author_facet Watson, Warin
Cherry, Cash
Lang, Rachelle
contents We demonstrate that automatic differentiation (AD), which has become commonly available in machine learning frameworks, is an efficient way to explore ideas that lead to algorithmic improvement in multi-scale affine image registration and affine super-resolution problems. In our first experiment on multi-scale registration, we implement an ODE predictor-corrector method involving a derivative with respect to the scale parameter and the Hessian of an image registration objective function, both of which would be difficult to compute without AD. Our findings indicate that exact Hessians are necessary for the method to provide any benefits over a traditional multi-scale method; a Gauss-Newton Hessian approximation fails to provide such benefits. In our second experiment, we implement a variable projected Gauss-Newton method for super-resolution and use AD to differentiate through the iteratively computed projection, a method previously unaddressed in the literature. We show that Jacobians obtained without differentiating through the projection are poor approximations to the true Jacobians of the variable projected forward map and explore the performance of other approximations in the problem of super-resolution. By addressing these problems, this work contributes to the application of AD in image registration and sets a precedent for further use of machine learning tools in this field.
format Preprint
id arxiv_https___arxiv_org_abs_2411_02806
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Applications of Automatic Differentiation in Image Registration
Watson, Warin
Cherry, Cash
Lang, Rachelle
Optimization and Control
We demonstrate that automatic differentiation (AD), which has become commonly available in machine learning frameworks, is an efficient way to explore ideas that lead to algorithmic improvement in multi-scale affine image registration and affine super-resolution problems. In our first experiment on multi-scale registration, we implement an ODE predictor-corrector method involving a derivative with respect to the scale parameter and the Hessian of an image registration objective function, both of which would be difficult to compute without AD. Our findings indicate that exact Hessians are necessary for the method to provide any benefits over a traditional multi-scale method; a Gauss-Newton Hessian approximation fails to provide such benefits. In our second experiment, we implement a variable projected Gauss-Newton method for super-resolution and use AD to differentiate through the iteratively computed projection, a method previously unaddressed in the literature. We show that Jacobians obtained without differentiating through the projection are poor approximations to the true Jacobians of the variable projected forward map and explore the performance of other approximations in the problem of super-resolution. By addressing these problems, this work contributes to the application of AD in image registration and sets a precedent for further use of machine learning tools in this field.
title Applications of Automatic Differentiation in Image Registration
topic Optimization and Control
url https://arxiv.org/abs/2411.02806