Deformable Medical Image Registration with KAN-Based Implicit Neural Representations
Nikita A. Drozdov, Marat O. Zinovev, Dmitry V. SorokinDeformable image registration (DIR) is central to medical image analysis, supporting spatial alignment for longitudinal studies and multi-modal fusion. Learning-based methods such as CNNs and transformers provide rapid inference but often require large training datasets and can underperform classical iterative methods for specific anatomies or modalities. Implicit neural representations (INRs) offer a data-efficient alternative by modeling deformation fields as continuous coordinate-to-displacement mappings, yet their per-pair optimization makes runtime efficiency and robustness to initialization essential. We introduce KAN-IDIR and RandKAN-IDIR, the first Kolmogorov–Arnold network (KAN)-based INR framework for pairwise-optimized, resolution-independent DIR, designed to improve seed stability and resource efficiency without requiring a large training dataset. KANs use learnable activation functions that are well suited to continuous, physically structured deformation fields. RandKAN-IDIR further reduces cost through randomized basis sampling, preserving registration quality with fewer basis functions. We evaluate the methods on lung CT, brain MRI, and cardiac MRI datasets against pairwise-optimized neural approaches, dataset-trained deep models, and classical baselines. KAN-IDIR and RandKAN-IDIR provide the strongest overall performance among pairwise-optimized neural registration methods across all three datasets, with low computational overhead and superior stability across random initializations. On ACDC, KAN-IDIR also achieves the highest DSC and best deformation regularity among all compared methods. RandKAN-IDIR slightly outperforms adaptive basis selection variants while avoiding their additional training-time complexity. This makes the approach practical for reproducible clinical research use. Source code is publicly available.