arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2511.17392cs.CV

MorphSeek: 用于可变形图像配准的细粒度潜在表示级策略优化

MorphSeek: Fine-grained Latent Representation-Level Policy Optimization for Deformable Image Registration

Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);Sun Yat-sen University(中山大学);Fudan University(复旦大学);Hebei Medical University(河北医科大学)

Runxun Zhang, Yizhou Liu, Li Dongrui, Bo XU, Jingwei Wei

AI总结:

本文提出MorphSeek,一种在潜在特征空间中进行细粒度策略优化的方法,用于解决可变形图像配准中的高维变形空间和体素级监督稀缺问题,通过引入随机高斯策略头和组相对策略优化,实现了高效探索和粗到细的优化,提升了配准的Dice系数和标签效率。

摘要:

可变形图像配准(DIR)仍然是医学图像分析中的基本但具有挑战性的问题,主要由于密集位移场的高维变形空间和体素级监督的稀缺性。现有的强化学习框架通常将此空间投影到低维表示,限制了其捕捉空间变异形变的能力。我们提出MorphSeek,一种细粒度的表示级策略优化范式,将DIR重新表述为潜在特征空间中的连续优化过程。MorphSeek在编码器上引入随机高斯策略头,以建模潜在特征的分布,从而实现高效的探索和粗到细的优化。该框架通过组相对策略优化整合了无监督预热和弱监督微调,其中多轨迹采样稳定了训练并提高了标签效率。在三个3D配准基准(OASIS脑MRI、LiTS肝脏CT和腹部MR-CT)上,MorphSeek在保持高标签效率的同时,通过最小的参数成本和低步骤级延迟开销,实现了优于竞争基线的Dice改进。除了优化器细节外,MorphSeek推进了一种表示级策略学习范式,实现了空间一致且数据高效的形变优化,提供了一种原理上可行、不依赖于特定主干网络和优化器的可扩展视觉对齐解决方案。

英文摘要:

Deformable image registration (DIR) remains a fundamental yet challenging problem in medical image analysis, largely due to the prohibitively high-dimensional deformation space of dense displacement fields and the scarcity of voxel-level supervision. Existing reinforcement learning frameworks often project this space into coarse, low-dimensional representations, limiting their ability to capture spatially variant deformations. We propose MorphSeek, a fine-grained representation-level policy optimization paradigm that reformulates DIR as a spatially continuous optimization process in the latent feature space. MorphSeek introduces a stochastic Gaussian policy head atop the encoder to model a distribution over latent features, facilitating efficient exploration and coarse-to-fine refinement. The framework integrates unsupervised warm-up with weakly supervised fine-tuning through Group Relative Policy Optimization, where multi-trajectory sampling stabilizes training and improves label efficiency. Across three 3D registration benchmarks (OASIS brain MRI, LiTS liver CT, and Abdomen MR-CT), MorphSeek achieves consistent Dice improvements over competitive baselines while maintaining high label efficiency with minimal parameter cost and low step-level latency overhead. Beyond optimizer specifics, MorphSeek advances a representation-level policy learning paradigm that achieves spatially coherent and data-efficient deformation optimization, offering a principled, backbone-agnostic, and optimizer-agnostic solution for scalable visual alignment in high-dimensional settings.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑