paper-with-me

Papers

FireANTs: Adaptive Riemannian Optimization for Multi-Scale Diffeomorphic Matching

2024-04-01 · Rohit Jena, Pratik Chaudhari, James C. Gee

The paper proposes FireANTs, the first multi-scale Adaptive Riemannian Optimization algorithm for dense diffeomorphic image matching. One of the most critical and understudied aspects of diffeomorphic image matching algorithms are its highly ill-conditioned nature. We quantitatively capture the extent of ill-conditioning in a typical MRI matching task, motivating the need for an adaptive optimization algorithm for diffeomorphic matching. To this end, FireANTs generalizes the concept of momentum and adaptive estimates of the Hessian to mitigate this ill-conditioning in the non-Euclidean space of diffeomorphisms. Unlike common non-Euclidean manifolds, we also formalize considerations for multi-scale optimization of diffeomorphisms. Our rigorous mathematical results and operational contributions lead to a state-of-the-art dense matching algorithm that can be applied to generic image data with remarkable accuracy and robustness. We demonstrate consistent improvements in image matching performance across a spectrum of community-standard medical and biological correspondence matching challenges spanning a wide variety of image modalities, anatomies, resolutions, acquisition protocols, and preprocessing pipelines. This improvement is supplemented by 300x to 3200x speedup over existing CPU-based state-of-the-art algorithms. For the first time, we perform diffeomorphic matching of sub-micron mouse isocortex volumes at native resolution, and generate a 25{\mu}m mouse brain atlas in under 25 minutes. Our fast implementation also enables hyperparameter studies that were intractable with existing correspondence matching algorithms.

📄 PDF Abstract BibTeX arXiv:2404.01249

Code (1)

rohitrango/fireants 공식 구현 pytorch

Tasks

CPUImage RegistrationRiemannian optimization

Methods 이 논문이 사용한 방법론

Library 설명 없음

Similar Papers 제목 키워드 기반

Factored Levenberg-Marquardt for Diffeomorphic Image Registration: An efficient optimizer for FireANTs

2026-03-19 · Rohit Jena, Pratik Chaudhari, James C. Gee arxiv

FireANTs introduced a novel Eulerian descent method for plug-and-play behavior with arbitrary optimizers adapted for diffeomorphic image registration as a test-time optimization problem, with a GPU-accelerated implementa…

Image Registration

Large-Scale Riemannian Meta-Optimization via Subspace Adaptation

2025-01-25 · Peilin Yu, Yuwei Wu, Zhi Gao, Xiaomeng Fan 외

Riemannian meta-optimization provides a promising approach to solving non-linear constrained optimization problems, which trains neural networks as optimizers to perform optimization on Riemannian manifolds. However, exi…

Simultaneous Optimization of Geodesics and Fréchet Means

2025-11-06 · Frederik Möbius Rygaard, Søren Hauberg, Steen Markvorsen arxiv

A central part of geometric statistics is to compute the Fréchet mean. This is a well-known intrinsic mean on a Riemannian manifold that minimizes the sum of squared Riemannian distances from the mean point to all other …

Riemannian Adaptive Optimization Methods

2018-10-01 · ICLR 2019 5 · Gary Bécigneul, Octavian-Eugen Ganea

Several first order stochastic optimization methods commonly used in the Euclidean domain such as stochastic gradient descent (SGD), accelerated gradient descent or variance reduced methods have already been adapted to c…

Riemannian optimizationStochastic Optimization

Rieoptax: Riemannian Optimization in JAX

2022-10-10 · Saiteja Utpala, Andi Han, Pratik Jawanpuria, Bamdev Mishra

We present Rieoptax, an open source Python library for Riemannian optimization in JAX. We show that many differential geometric primitives, such as Riemannian exponential and logarithm maps, are usually faster in Rieopta…

CPUGPURiemannian optimizationStochastic Optimization