arXiv:2609.05484cs.CVcs.LG2026-09

医学影像配准中,正则化损失是提升精度与稳定性的关键,几乎零成本。

Architectural and Regularization Components in Deep Learning Medical Image Registration: Systematic Ablation Study

  • 分离分析架构与正则化的作用,控制变量验证效果。
  • 正则化使最大形变降低99%,精度提升21.3%,推理时间几乎不变。
  • 适合关注临床可用性与模型稳定性的医学图像研究者。

深度学习医学图像配准常结合架构改进(如仿射预对齐)与训练目标改进(如正则化损失),但二者协同作用难以区分。本研究在OASIS脑部MRI数据集(394个训练、20个测试)上系统性地对比四种模型变体:基础3D U-Net、加入完整正则化套件的U-Net、仿射+可变形架构、以及仿射架构加完整正则化。评估指标包括配准精度(MSE、NCC、SSIM)、形变质量(雅可比行列式保持性、位移统计、解剖合理性评分)及计算开销。结果显示,仅正则化即带来21.3%的相对MSE提升(1.78%→2.16%,P<.001)、21.8%的NCC提升,最大形变从53.1降至0.51单位(减少99.0%),且推理时间仅增加0.06%。联合模型实现最大精度提升25.8%(1.78%→2.24%),解剖合理性从0.596升至0.930,推理时间增加9.8%。梯度相关性由0.742升至0.980。所有增强模型均达到亚像素精度,且在合理形变约束下表现优异。结论表明:正则化是核心驱动力,几乎免费提供精度与形变控制;仿射架构提供较小补充收益,代价可控。99%的形变抑制解决了临床部署的关键障碍。

原文摘要 · Abstract (English)

Deep learning registration methods routinely stack two kinds of enhancement on a base network: architectural additions such as affine pre-alignment stages, and training-objective additions such as regularization losses. Papers tend to adopt both at once, so it is unclear which is doing the work. I ran a controlled ablation to separate them. Using the OASIS brain MRI dataset (394 training subjects, 20 test subjects), I trained four variants of the same registration pipeline: a baseline 3D U-Net with basic similarity losses, the same U-Net with a full regularization suite, an affine-plus-deformable architecture with basic losses, and the affine architecture with the full suite. I evaluated registration accuracy (MSE, NCC, SSIM), deformation quality (Jacobian determinant preservation, displacement statistics, an anatomical plausibility score), and computational cost. Regularization alone accounted for most of the gain: a 21.3% relative gain on the MSE-improvement metric (1.78% to 2.16%, P<.001) and a 21.8% relative gain in NCC improvement, while cutting maximum deformation from 53.1 to 0.51 units, a 99.0% reduction, at essentially no computational cost (-0.06% inference time). The combined model produced the largest accuracy gain, 25.8% (1.78% to 2.24%), and raised anatomical plausibility from 0.596 to 0.930, at a moderate +9.8% inference-time cost. Gradient correlation rose from 0.742 at baseline to 0.980 for the fully enhanced model. All enhanced variants reached sub-voxel accuracy under plausible deformation constraints. Regularization losses are the primary driver in this setting, delivering the accuracy gains and almost all of the deformation control for free at inference time, while the affine architecture adds a smaller complementary benefit at acceptable cost. The 99% reduction in unrealistic deformations addresses a known barrier to clinical deployment.

医学图像图像配准正则化形变控制

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