arXiv:2603.16940eess.IVcs.AI2026-03

用稀疏控制点网格提升医学图像配准的效率与稳定性

On the Degrees of Freedom of Gridded Control Points in Learning-Based Medical Image Registration

  • 用稀疏网格控制点替代密集体素解码,降低参数量和内存占用
  • 在前列腺、盆腔器官等三组数据上实现更高精度配准,计算成本更低
  • 支持推理时自适应不同网格密度,无需重新训练

许多配准问题在均质或噪声区域中病态,而密集体素解码器维度过高。稀疏控制点参数化可提供紧凑平滑的形变表示,减少内存并提高稳定性。本文研究学习型配准网络所需的控制点数量,提出GridReg框架:将密集体素解码替换为在稀疏网格控制点上的位移预测。该设计显著降低参数量与内存消耗,同时保持配准精度。多尺度3D编码器特征图经位置编码后展平为1D令牌序列以保留空间上下文,模型通过交叉注意力模块预测稀疏网格形变场。进一步提出网格自适应训练,使模型在推理时可适配多种网格尺寸而无需重训。在前列腺腺体、盆腔器官及神经结构三组数据集上,结果表明使用网格控制位移场显著提升性能;相较现有基于密集形变场(DDF)或散点关键点采样的算法,本方法取得更优配准效果,且计算开销相当或更低。

原文摘要 · Abstract (English)

Many registration problems are ill-posed in homogeneous or noisy regions, and dense voxel-wise decoders can be unnecessarily high-dimensional. A sparse control-point parameterisation provides a compact, smooth deformation representation while reducing memory and improving stability. This work investigates the required control points for learning-based registration network development. We present GridReg, a learning-based registration framework that replaces dense voxel-wise decoding with displacement predictions at a sparse grid of control points. This design substantially cuts the parameter count and memory while retaining registration accuracy. Multiscale 3D encoder feature maps are flattened into a 1D token sequence with positional encoding to retain spatial context. The model then predicts a sparse gridded deformation field using a cross-attention module. We further introduce grid-adaptive training, enabling an adaptive model to operate at multiple grid sizes at inference without retraining. This work quantitatively demonstrates the benefits of using sparse grids. Using three data sets for registering prostate gland, pelvic organs and neurological structures, the results suggested a significant improvement with the usage of grid-controled displacement field. Alternatively, the superior registration performance was obtained using the proposed approach, with a similar or less computational cost, compared with existing algorithms that predict DDFs or displacements sampled on scattered key points.

医学图像配准稀疏控制点网格自适应轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。