arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.04795cs.CVcs.AI
收藏

法律与秩序:医学扩散与分割的自适应空间加权

LAW & ORDER: Adaptive Spatial Weighting for Medical Diffusion and Segmentation

Elmore Family School of Electrical and Computer Engineering(埃尔莫尔电气与计算机工程学院);Purdue University(普渡大学);Sesame AI;Department of Computer Science(计算机科学系);Department of Agricultural and Biological Engineering (ABE)(农业与生物工程系);Department of Agricultural Sciences Education and Communication (ASEC)(农业科学教育与交流系)

Anugunj Naman, Ayushman Singh, Gaibo Zhang, Yaguang Zhang

AI总结:

研究医学图像分析中自适应空间加权,通过两个适配器实现。LAW为掩码条件扩散学习像素损失权重,ORDER通过选择性双向跳跃注意力改进分割,在多个数据集上取得良好效果,证明该方法对医学扩散和分割的有效性。

摘要:

医学图像分析依赖准确分割和可控合成,但面临空间不平衡问题。研究自适应空间加权作为任务级设计原则并在两个适配器中实例化。LAW通过调制比率先验与特征相关增量图学习掩码条件扩散的像素损失权重,ORDER通过添加选择性双向跳跃注意力和阶段置信门控改进轻量级分割,在多个数据集上取得显著性能提升。

英文摘要:

Medical image analysis depends on accurate segmentation and controllable synthesis, but both tasks face severe spatial imbalance: lesions occupy small regions against large backgrounds. We study adaptive spatial weighting as a task-level design principle and instantiate it in two adapters. LAW learns per-pixel loss weights for mask-conditioned diffusion by modulating a ratio prior with a feature-dependent delta map, with normalization, clamping, and Dice regularization for stability. ORDER improves lightweight segmentation by adding selective bidirectional skip attention with stage-wise confidence gating. On held-out diffusion test sets, LAW lowers FID from 158.13$\pm$0.15 to 108.43$\pm$0.71 on Polyps, from 144.13$\pm$0.31 to 89.51$\pm$0.96 on KiTS19, and from 139.22$\pm$0.38 to 112.58$\pm$0.68 on BRISC, while improving held-out mask-recovery Dice from 0.681$\pm$0.013 to 0.825$\pm$0.003 on Polyps. When the resulting images are added to nnUNet training, downstream Polyps mDice rises from 71.7$\pm$0.4 to 74.1$\pm$0.8. On the cleaned Polyps segmentation protocol, the reported ORDER configuration reaches 76.3$\pm$1.9 mDice and 67.2$\pm$2.0 mIoU at 42K parameters and 0.11 GFLOPs, versus 70.3$\pm$1.5 mDice and 59.9$\pm$1.7 mIoU for matched MK-UNet. On BRISC under the same training recipe, ORDER reaches 77.4$\pm$0.8 mDice and 68.1$\pm$0.7 mIoU. These results position adaptive spatial weighting as a practical design idea for both medical diffusion and efficient segmentation.

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

↑