Registration is a Representation Problem: Solving Whole-Body PET/CT Registration
Organizations: A. A. Martinos Center for Biomedical Imaging, Massachusetts General Hospital · Department of Radiology, Harvard Medical School · Computer Science and AI Lab, Massachusetts Institute of Technology
Abstract
Raw image intensities are a poor surrogate for similarity across images in correspondence problems. Further, while auxiliary segmentation labels can provide semantic identification and improve matching, they can assign only coarse identities to fine-grained anatomy and often cover only a subset of structures, particularly in whole-body settings. We present an approach that instead establishes correspondence using complementary representations of anatomy and demonstrate its utility for longitudinal whole-body PET/CT registration in the Learn2Reg 2026 PSMAReg Challenge. Our 3D foundation model, anatomix, is used to derive fine-grained, per-voxel semantic representations that are consistent across volumes, and we further employ off-the-shelf segmentation models to obtain coarse anatomical identities. Together, these representations drive a GPU-accelerated iterative registration solver (FireANTs) in a hybrid multistage framework: the first stage estimates alignment from dense multi-channel features, after which a second refines the deformation using both anatomix- and segmentation-based losses. Lastly, because registration should not introduce non-biological tumor volume changes across time, we develop lesion-specific component-wise volume preservation schemes. Without any task-specific training, our system won first place in the PSMAReg Challenge leaderboard. Code and interactive visualizations are available at http://www.neeldey.com/reps-for-reg.
Figures & tables
| Team | CT Dice | CT HD95 | PET Dice | PET HD95 | MTV Error (%) | TLG Error (%) | NDV (%) | Rank |
|---|---|---|---|---|---|---|---|---|
| Ours | 0.830 | 4.6 | 0.610 | 19.4 | 1.78 | 1.39 | 0.0008 | 1 |
| heyufan1995 | 0.832 | 4.9 | 0.628 | 18.9 | 5.37 | 5.35 | 0.0032 | 2 |
| tinymilky | 0.832 | 4.6 | 0.598 | 20.0 | 3.41 | 3.24 | 0.0009 | 3 |
| Configuration | CT DSC | CT HD95 | %NDV | MTV% | TLG% | CT MAE |
|---|---|---|---|---|---|---|
| No registration | 0.114(0.011) | 41.72(2.98) | – | – | – | 302.1(12.0) |
| Stage 1, CT+PET, default FireANTs | 0.586(0.027) | 20.07(3.43) | 0.00025(0.00015) | 15.49(2.15) | 16.67(2.31) | 118.1(14.5) |
| Stage 1, CT+PET, tuned FireANTs | 0.686(0.012) | 11.09(0.94) | 0.00113(0.00086) | 15.56(2.08) | 18.47(2.39) | 57.3 (3.3) |
| + anatomix + MIND-SSC | 0.775(0.007) | 6.54(0.37) | 0.00125(0.00085) | 6.31(1.68) | 6.93(1.76) | 58.6(3.8) |
| + volume penalty | 0.773(0.007) | 6.57(0.36) | 0.00145(0.00090) | 4.95(1.28) | 5.36(1.48) | 60.0(3.8) |
| + volume projection | 0.772(0.007) | 6.57(0.36) | 0.00147(0.00092) | 3.41(1.18) | 4.14(1.49) | 60.5(3.8) |