English

TotalRegistrator: Towards a Lightweight Foundation Model for CT Image Registration

Image and Video Processing 2025-08-07 v1 Computer Vision and Pattern Recognition

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.

Keywords

Cite

@article{arxiv.2508.04450,
  title  = {TotalRegistrator: Towards a Lightweight Foundation Model for CT Image Registration},
  author = {Xuan Loc Pham and Gwendolyn Vuurberg and Marjan Doppen and Joey Roosen and Tip Stille and Thi Quynh Ha and Thuy Duong Quach and Quoc Vu Dang and Manh Ha Luu and Ewoud J. Smit and Hong Son Mai and Mattias Heinrich and Bram van Ginneken and Mathias Prokop and Alessa Hering},
  journal= {arXiv preprint arXiv:2508.04450},
  year   = {2025}
}
R2 v1 2026-07-01T04:37:25.172Z