TotalRegistrator: Towards a Lightweight Foundation Model for CT Image Registration
2508.04450v1
eess.IV, cs.CV
2025-08-09
Авторы:
Xuan Loc Pham, Gwendolyn Vuurberg, Marjan Doppen, Joey Roosen, Tip Stille, Thi Quynh Ha, Thuy Duong Quach, Quoc Vu Dang, Manh Ha Luu, Ewoud J. Smit, Hong Son Mai, Mattias Heinrich, Bram van Ginneken, Mathias Prokop, Alessa Hering
Резюме на русском
**Резюме**
Регистрация изображений является ключевым элементом анализа мультифазных и многоорганичных медицинских сканов, однако большинство существующих методов ориентированы на регистрацию отдельных органов, что ограничивает их применение. Мы представляем TotalRegistrator — новую систему регистрации, способную совместно выравнивать несколько органов в рамках одной модели. Она основывается на стандартной UNet-архитектуре с использованием оригинальной стратегии разделения полей, что позволяет значительно сократить требования к GPU-памяти — до 11 ГБ при обучении.
Мы обучили TotalRegistrator на большом лабиринте данных со сканами зоны грудной клетки, живота и таза, а также проверили его на трех внешних наборах данных. Результаты показали, что наше решение выполняет регистрацию многоорганичных снимков более эффективно, чем классические и новые генерик-модели, при этом масштабируемость и универсальность системы доказаны на различных внешних наборах. Таким образом, TotalRegistrator представляет собой прорыв в области многоорганичной регистрации медицинских изображений, обладая высокой обобщающей способностью.
Abstract
Image registration is a fundamental technique in the analysis of longitudinal
and multi-phase CT images within clinical practice. However, most existing
methods are tailored for single-organ applications, limiting their
generalizability to other anatomical regions. This work presents
TotalRegistrator, an image registration framework capable of aligning multiple
anatomical regions simultaneously using a standard UNet architecture and a
novel field decomposition strategy. The model is lightweight, requiring only
11GB of GPU memory for training. To train and evaluate our method, we
constructed a large-scale longitudinal dataset comprising 695 whole-body
(thorax-abdomen-pelvic) paired CT scans from individual patients acquired at
different time points. We benchmarked TotalRegistrator against a generic
classical iterative algorithm and a recent foundation model for image
registration. To further assess robustness and generalizability, we evaluated
our model on three external datasets: the public thoracic and abdominal
datasets from the Learn2Reg challenge, and a private multiphase abdominal
dataset from a collaborating hospital. Experimental results on the in-house
dataset show that the proposed approach generally surpasses baseline methods in
multi-organ abdominal registration, with a slight drop in lung alignment
performance. On out-of-distribution datasets, it achieved competitive results
compared to leading single-organ models, despite not being fine-tuned for those
tasks, demonstrating strong generalizability. The source code will be publicly
available at: https://github.com/DIAGNijmegen/oncology_image_registration.git.
Ссылки и действия
Дополнительные ресурсы: