arXiv:2506.10813cs.CVeess.IV2025-06中稿 · publication at Inf…被引 4

提出SmoothProper模块,提升无监督图像配准在稀疏特征场景下的精度与鲁棒性。

Unsupervised Deformable Image Registration with Structural Nonparametric Smoothing

  • 引入可插拔的SmoothProper模块,在前向传播中实现平滑约束与信息传递。
  • 在2912x2912视网膜血管图像上实现1.88像素注册误差,首次有效应对大位移与孔径问题。
  • 无需调参正则化超参数,适配现有框架,适合医学图像配准研究者使用。

基于学习的可变形图像配准(DIR)通过神经网络摊销传统优化过程,加速图像对齐。标签监督进一步提升精度,实现对未见扫描图像的高效精准非线性对齐。然而,如视网膜血管这类特征稀疏且存在大面积平滑区域的图像,会带来孔径与大位移挑战,使无监督DIR方法难以应对。其根源在于神经网络单次前向传播预测形变场后,场本身无约束,导致正则化压力完全由网络权重承担。为此,本文提出SmoothProper——一种可插拔神经模块,通过结合基于对偶的优化层与定制交互项,在前向传播中实现平滑性约束与流信号跨空间传播,强化结构一致性。该模块具备模型无关性,可无缝集成至现有配准框架,参数开销极低,且无需正则化超参数调优。在视网膜血管数据集上的初步实验表明,本方法将注册误差降至1.88像素(针对2912x2912图像),成为首个成功应对孔径与大位移双重挑战的无监督DIR方法。代码将开源于https://github.com/tinymilky/SmoothProper。

原文摘要 · Abstract (English)

Learning-based deformable image registration (DIR) accelerates alignment by amortizing traditional optimization via neural networks. Label supervision further enhances accuracy, enabling efficient and precise nonlinear alignment of unseen scans. However, images with sparse features amid large smooth regions, such as retinal vessels, introduce aperture and large-displacement challenges that unsupervised DIR methods struggle to address. This limitation occurs because neural networks predict deformation fields in a single forward pass, leaving fields unconstrained post-training and shifting the regularization burden entirely to network weights. To address these issues, we introduce SmoothProper, a plug-and-play neural module enforcing smoothness and promoting message passing within the network's forward pass. By integrating a duality-based optimization layer with tailored interaction terms, SmoothProper efficiently propagates flow signals across spatial locations, enforces smoothness, and preserves structural consistency. It is model-agnostic, seamlessly integrates into existing registration frameworks with minimal parameter overhead, and eliminates regularizer hyperparameter tuning. Preliminary results on a retinal vessel dataset exhibiting aperture and large-displacement challenges demonstrate our method reduces registration error to 1.88 pixels on 2912x2912 images, marking the first unsupervised DIR approach to effectively address both challenges. The source code will be available at https://github.com/tinymilky/SmoothProper.

图像配准无监督学习医学影像神经网络

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