解决医学图像小目标分割与生成中的空间不平衡问题
LAW & ORDER: Adaptive Spatial Weighting for Medical Diffusion and Segmentation
- 设计自适应空间加权机制,动态调整像素损失权重
- 在多个数据集上显著降低生成质量指标FID,提升分割Dice分数
- 适用于轻量级医学图像分割与可控图像生成任务
医学图像分析依赖精准分割与可控生成,但两者均面临严重空间不平衡:病灶区域远小于背景。本文将自适应空间加权作为任务级设计原则,并实现两个适配器。LAW通过特征依赖的偏移图调制比例先验,生成逐像素损失权重,结合归一化、裁剪和Dice正则化以保证稳定;ORDER通过阶段式置信度门控的双向跳跃注意力提升轻量化分割性能。在独立测试集上,LAW将Polyps的FID从158.13±0.15降至108.43±0.71,KiTS19从144.13±0.31降至89.51±0.96,BRISC从139.22±0.38降至112.58±0.68,同时提升掩码恢复Dice至0.825±0.003。将生成图像加入nnUNet训练后,Polyps的mDice从71.7±0.4升至74.1±0.8。在清理后的Polyps分割协议中,ORDER以42K参数和0.11 GFLOPs达76.3±1.9 mDice、67.2±2.0 mIoU,优于匹配的MK-UNet(70.3±1.5 mDice,59.9±1.7 mIoU);BRISC下同样设置达77.4±0.8 mDice、68.1±0.7 mIoU。结果表明,自适应空间加权是医学扩散与高效分割的有效设计范式。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。