通过分步定位与互斥分类,提升脑部病变多类分割精度。
Mutually Exclusive Multiclass Lesion Segmentation in Neuroimaging: Binary-Guided Weak Supervision with Inter-Class Orthogonality

- 先定位病变区域,再精准分配类别,避免混淆
- 水肿边界误差仅29.56毫米,罕见类型表现最优
- 适合医学影像中复杂病变的弱监督分割任务
由于激活重叠、伪标签噪声及缺乏显式类别互斥约束,共现神经影像病灶的弱监督分割仍具挑战。本文提出BiMEx-MS框架,将多类分割分解为整体病灶定位与互斥类别分配:二值定位模块提供不依赖类别频率的结构先验,将多类预测限制在检测到的病灶区域内;多出口分类架构结合有监督对比预训练,生成多尺度类别判别激活图,并通过类别特定注意力网络聚合。通过包含类别内分离、类别间正交性和二值-多类空间一致性在内的三元损失,强制实现类别互斥性,并结合层级形态学伪标签精修。在脑肿瘤MRI(BraTS 2020、BraTS 2023 SSA)和颅内出血CT(RSNA-ICH至BHSD)数据集上评估,相较于16种弱监督基线,BiMEx-MS在水肿HD95上达29.56毫米(唯一低于40毫米的方法),硬膜下出血Dice为0.704,边界指标与稀有亚型上的增益最为显著。跨数据集泛化、六种骨干网络消融及不确定性量化均表明,结构引导是性能提升的关键,而非模型容量。代码已开源。
原文摘要 · Abstract (English)
Weakly supervised segmentation of co-occurring neuroimaging lesion subclasses remains challenging due to overlapping activations, noisy pseudo-labels, and the absence of explicit inter-class exclusivity constraints. We propose BiMEx-MS (Binary-guided Mutually Exclusive Multiclass Segmentation), a framework that decomposes multiclass segmentation into whole-lesion localization and exclusive class assignment: a binary localization module provides a class-frequency-agnostic structural prior confining multiclass predictions within the detected lesion domain, while a multi-exit classification architecture with supervised contrastive pretraining produces multi-scale class-discriminative activation maps aggregated via a class-specific attention network. Inter-class exclusivity is enforced through a tri-partite loss comprising per-class separation, inter-class orthogonality, and binary-multiclass spatial consensus, followed by hierarchical morphological pseudo-label refinement. Evaluated across brain tumor MRI (BraTS 2020, BraTS 2023 SSA) and intracranial hemorrhage CT (RSNA-ICH to BHSD) against sixteen weakly supervised baselines, BiMEx-MS achieves Edema HD95 of 29.56 mm (the only method below 40 mm) and subdural hemorrhage Dice of 0.704, with gains consistently largest on boundary metrics and rare subtypes. Cross-dataset generalization, backbone ablations across six architectures, and uncertainty quantification confirm that structural guidance rather than model capacity drives performance. Code: https://github.com/ashutoshkr45/BiMEx-MS-Neuro.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。