arXiv:2608.25819cs.CV2026-08中稿 · MICCAI 2026

通过引导采样位置而非修改卷积核,提升细长结构分割精度

Steer the Sampling, Not the Kernel Grid: Geometry-Guided Sampling Operator for Volumetric Segmentation

论文配图:Steer the Sampling, Not the Kernel Grid: Geometry-Guided Sampling Operator for Volumetric Segmentation
图 1 · 摘自论文原文
  • 基于几何方向预测,动态调整特征采样位置,保持结构连贯性
  • 在BraTS等数据集上边界指标显著提升,参数量减少65%
  • 可无缝集成到多种3D分割模型中,适合医学图像细结构分析

精确的3D分割对临床病灶评估与解剖映射至关重要。细长或微小结构(如血管)分割尤为困难:一像素的边界误差可能导致分支断开并改变关键拓扑关系。在编码器-解码器网络(如U-Net)中,反复下采样和固定网格卷积会模糊或混叠细结构,削弱方向信息,导致早期错误跨尺度传播。本文提出一种几何引导的局部算子,不修改卷积核而引导特征采样位置,在单一致的框架下同时支持特征精炼(步长1)与分辨率降低(步长>1)。每个体素预测局部方向与受限步长,沿该方向对称采样,并通过轻量混合生成紧凑的几何与边界线索;跨尺度一致性对齐编码器与解码器特征以减少几何错位。在3D U-Net中替换所有步长1与步长2算子,在BraTS、MSD Hepatic Vessel和TDSC-ABUS数据集上均取得稳定提升,边界指标显著改善(如BraTS Dice从86.1升至88.9,HD95从7.1降至6.2;TDSC-ABUS HD95从39.1降至27.8),参数量由2.3M降至0.8M。进一步验证该算子可无修改地集成至nnU-Net、Swin-UNETR和MedNeXt等主流骨干网络,带来一致性能增益。

原文摘要 · Abstract (English)

Accurate 3D segmentation is central to quantitative lesion assessment and anatomy mapping for clinical planning and follow-up. Thin, elongated, and fine anatomical/pathological structures (e.g., vessels) are a particularly challenging case: a one-voxel boundary error can disconnect a branch and change clinically relevant topology. In encoder-decoder networks (e.g., U-Net), repeated downsampling and fixed-grid convolution blur or alias fine structures and weaken orientation cues, so early mistakes propagate across scales. We propose a geometry-guided local operator that steers where features are sampled, rather than deforming convolutional kernels, under a single formulation for both feature refinement (stride 1) and resolution reduction (stride > 1). At each voxel, it predicts a local orientation and bounded step sizes, samples symmetrically along these directions, and transforms paired samples into compact geometric and boundary cues with lightweight mixing; a cross-scale consensus aligns encoder and decoder features at skip connections to reduce geometric mismatch. Replacing all stride 1 and stride 2 operators in a 3D U-Net yields consistent improvements on BraTS, MSD Hepatic Vessel, and TDSC-ABUS, with notably better boundary metrics (e.g., BraTS Dice 86.1 to 88.9, HD95 7.1 to 6.2; TDSC-ABUS HD95 39.1 to 27.8) while reducing parameters from 2.3M to 0.8M. We further demonstrate that the operator can be integrated into other backbones (e.g., nnU-Net, Swin-UNETR, and MedNeXt) without changing their macro-architectures while providing consistent performance gains.

3D分割医学影像几何引导细结构

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