MAGEFormer将体素间距嵌入位置编码与注意力,在BTCV与FLARE 22各向异性腹部CT分割上取得最低HD95
核心概要
作者提出MAGEFormer,把物理间距约束嵌入表示学习:用MASE按体素间距校准位置频率、用GCA抑制物理上不合理的跨尺度特征关联、用GVV在推理时做确定性亚体素视图聚合;在BTCV与FLARE 22统一5折协议下,其边界精度在对比方法中最优(HD95分别为10.58 mm与3.40 mm),Dice也一致提升,且各向异性越强优势越大。
Figure 1: MAGEFormer: Architecture Overview and Geometric Motivation
arXiv深度剖析
MAGEFormer在BTCV与FLARE 22上取得所比较方法中最低的HD95,即最强边界精度,同时Dice一致提升:FLARE 22为93.91% Dice / 3.40 mm HD95,BTCV为84.11% / 10.58 mm。 此前ViT类体积分割方法多依赖体素索引参数化的位置编码与相对位置偏置,或仅做厚度感知的算子设计,未把度量一致性作为表示层面的约束;该工作把物理间距约束同时施加于位置编码、跨尺度特征交互与推理聚合三个环节。 在统一固定5折交叉验证、共享折划分、官方模型与参数配置、统一指标流程下,与nnU-Net、TransUNet、UNETR、Swin-UNETR、U-Mamba对比;报告了跨折标准差,FLARE 22上Dice标准差0.20、HD95标准差0.48。
消融显示三个组件各自有效且可叠加:以Swin-UNETR为基线,单独加入MASE、GCA、GVV分别带来约+4.91%、+5.06%、+3.84%的平均Dice,MASE+GCA为+8.08%,三者全开为+9.24%(0.7487→0.8411),HD95从75.21 mm降至10.58 mm。 该工作把度量一致性拆解为编码、特征交互、推理三个正交层面并分别验证,说明几何校准不是单点修补,而是可在多个几何信息入口处同时施加的设计原则。 BTCV上的逐组件消融表,基线为Swin-UNETR;同时报告了HD95的相应变化,其中MASE与GCA单独使用即带来明显的HD95下降。
增益集中在易受各向异性采样影响的小器官与拓扑敏感结构:BTCV上肾上腺平均Dice达72.76%,胰腺提升至82.59%,为所比较方法中最好。 这些结构对层间分辨率与部分容积效应高度敏感,结果支持度量一致建模改善细粒度解剖勾画,而不仅是整体重叠指标的提升。 逐器官Dice对比表,覆盖13个腹部器官;作者指出肾上腺与胰腺尾部等低对比结构边界质量对采样几何高度敏感。
优势随各向异性加剧而扩大:按自然各向异性比分层后,相对基线的增益随比值增大而变宽;在仅测试时施加的合成z轴退化下,MAGEFormer比对比基线退化更平缓。GVV在精度与标准TTA相当的同时推理时间显著更低。 这直接检验了核心假设——内部几何校准优于仅依赖常规各向同性预处理,并给出推理阶段确定性聚合相对随机TTA的效率优势。 合并两个基准测试病例按各向异性比分层分析,以及受控合成z轴降采样再插值回原网格的测试时扰动实验,阴影区域为跨折标准差;速度—精度对比来自GVV与标准TTA的比较。
启示与展望
该结果面向腹部多器官CT分割场景,适用于层间间距明显粗于层内间距的采集条件,并在BTCV(30例)与FLARE 22(50例标注扫描)两个基准上以统一5折协议验证。方法设计为与标准U形Transformer骨干兼容,因此可直接嵌入既有体积分割流程;作者指出未来工作将把这一几何感知范式扩展到多参数MRI等其他固有各向异性模态。对读者而言,这意味着当数据采集本身存在明显各向异性、且关注小器官或薄结构边界时,把物理间距写入位置编码、跳跃交互与推理聚合是一条可复用的工程路径。
读者仍需关注:各向异性分层与合成z轴退化实验的具体分层阈值与曲线数值在正文中以图呈现,正文未给出完整数值表;GVV的视图数量与每视图位移量在正文中以符号表示,具体取值需查原文;训练超参数中初始学习率与输入裁剪尺寸在正文中未显示数值;此外,结论基于两个腹部CT基准,向其他解剖部位、其他各向异性模态以及不同采集协议的推广仍需独立验证。
