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.

Ссылки и действия