用双曲空间建模解剖结构层次,提升少样本医学图像分割精度
H2AL: Hyperbolic Hierarchy-aware Aggregative Learning for Registration-based Few-shot Medical Image Segmentation

- 在双曲空间学习解剖层次表示,通过门控注入融合到欧式空间
- 联合优化注册与分割任务,梯度聚合提升模型协同能力
- 适用于需精准分割复杂解剖结构的少样本医学影像场景
基于配准的少样本医学图像分割(RFMIS)通过配准将标注图像形变至未标注图像以生成伪标签。现有方法多在欧氏空间进行像素级优化与推理,将解剖结构视为平坦且分离,忽视其固有层级关系,导致伪标签质量下降,模糊区域区分度弱,限制分割性能。为此,我们提出双曲层次感知聚合学习框架H2AL,同时提升形变合理性与解剖结构区分度。具体地,设计双曲层次感知注入(H2I)模块,利用双曲空间的层次建模能力,通过变换引导的监督双曲对比学习,获得精确的层次感知表示,并通过门控注入块将其注入欧式空间,保留语义丰富性;进一步提出端到端联合优化算法,通过梯度聚合,整合注册与分割解码器的梯度,携带语义与层次线索,更新共享编码器,促进任务间协同学习。在两个解剖区域、五种实验设置下进行的大量实验表明,该方法在注册与分割任务上均具有效性和高效性。代码已公开于https://github.com/JiamingCai469/H2AL。
原文摘要 · Abstract (English)
Registration-based Few-shot medical image segmentation (RFMIS) aims to generate pseudo-labels for unlabeled images by warping a labeled image through registration. However, existing methods primarily perform pixel-level optimization and inference in Euclidean space, treating anatomical structures as flat and disjoint. This neglect of inherent hierarchies degrades pseudo-label quality and weakens the discrimination of ambiguous regions, limiting the segmentation performance. To overcome this challenge, we propose a Hyperbolic Hierarchy-aware Aggregative Learning framework for RFMIS, termed H2AL, that enhances both deformation plausibility and anatomical discrimination for dual-task learning. Specifically, we introduce a Hyperbolic Hierarchy-aware Infusion (H2I) module, which leverages the hierarchical modeling capability of hyperbolic space to learn precise hierarchy-aware representations via transformation-guided supervised hyperbolic contrastive learning, and injects such hierarchical priors into Euclidean space through a gated infusion block while preserving semantic richness. Furthermore, we propose an end-to-end joint optimization algorithm by gradient aggregation, where the gradients from the registration and segmentation decoders, embedding semantic and hierarchical cues, are aggregated to update the shared encoder to promote collaborative learning across tasks. Extensive experiments on two anatomical regions, with five experimental settings, demonstrate the effectiveness and efficiency of our method in both registration and segmentation. The code is publicly available at https://github.com/JiamingCai469/H2AL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。