用对抗性先验保护关键脑部影像特征,提升分割精度与鲁棒性。
Keep the Core: Adversarial Priors for Significance-Preserving Brain MRI Segmentation
- 通过对抗优化识别对分割影响最大的像素块,生成重要性地图。
- 高重要性区域增强后恢复原值,低重要性区域选择性掩码重建。
- 无需额外推理开销,适用于多种模型,特别适合标注稀疏场景。
医学图像分割受限于病理标注稀缺。现有增强方法(如常规变换或随机掩码)缺乏特征感知,常破坏关键诊断语义或忽略核心特征。本文提出「保持核心」的数据驱动范式,利用对抗性先验指导增强与掩码过程,实现语义保留。引入SAGE(稀疏对抗门控估计器)作为离线模块,通过求解对抗优化问题并施加ℓ₁稀疏惩罚,识别出微小扰动即可改变分割边界的最小令牌集合,生成令牌重要性图W。在线的KEEP(关键区域增强与保护)模块基于W实施双路径策略:(1) 语义保持增强:高重要性令牌被增强但原始像素严格还原;(2) 引导掩码增强:低重要性令牌选择性掩码,以MAE形式重建,迫使模型从保留的关键特征中学习稳健表示。「保持核心」与主干网络无关且无推理开销。大量实验表明,SAGE的结构化先验与KEEP的区域选择机制高度互补,在二维医学数据集上实现最先进的分割鲁棒性与泛化能力。
原文摘要 · Abstract (English)
Medical image segmentation is constrained by sparse pathological annotations. Existing augmentation strategies, from conventional transforms to random masking for self-supervision, are feature-agnostic: they often corrupt critical diagnostic semantics or fail to prioritize essential features. We introduce "Keep the Core," a novel data-centric paradigm that uses adversarial priors to guide both augmentation and masking in a significance-preserving manner. Our approach uses SAGE (Sparse Adversarial Gated Estimator), an offline module identifying minimal tokens whose micro-perturbation flips segmentation boundaries. SAGE forges the Token Importance Map $W$ by solving an adversarial optimization problem to maximally degrade performance, while an $\ell_1$ sparsity penalty encourages a compact set of sensitive tokens. The online KEEP (Key-region Enhancement \& Preservation) module uses $W$ for a two-pronged augmentation strategy: (1) Semantic-Preserving Augmentation: High-importance tokens are augmented, but their original pixel values are strictly restored. (2) Guided-Masking Augmentation: Low-importance tokens are selectively masked for an $\text{MAE}$-style reconstruction, forcing the model to learn robust representations from preserved critical features. "Keep the Core" is backbone-agnostic with no inference overhead. Extensive experiments show SAGE's structured priors and KEEP's region-selective mechanism are highly complementary, achieving state-of-the-art segmentation robustness and generalization on 2D medical datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。