Organizations: USTC, Anhui, China · Anhui Province Key Laboratory of Biomedical Imaging and Intelligent Processing, Institute of Artificial Intelligence, Hefei Comprehensive National Science Center, Anhui, China
Whole-slide images may contain coexisting lesions, yet weak supervision typically provides only the most severe diagnosis. We formulate this setting as dominant-label multiple instance learning, where the observed label is the maximum-severity element of a latent lesion set. Standard multiclass MIL treats it as a complete one-hot description and therefore suppresses lower-severity evidence whenever it co-occurs with a dominant lesion. We propose DomSet-MIL, a label-aware framework that aligns the annotation mechanism across bag and instance levels. Dominant-Label Censored Likelihood models the reported diagnosis without declaring unreported lesions absent. Severity-Ordered Residual Routing assigns patch evidence from high to low severity and preserves residual support for secondary findings, while top-q pooling links local routing scores to bag prediction. Experiments across natural and controlled settings show that DomSet-MIL achieves state-of-the-art performance in dominant-label prediction while more effectively recovering latent coexisting-lesion evidence. These findings demonstrate that modeling the dominant-label mechanism can improve diagnostic prediction without sacrificing clinically meaningful secondary findings.
Figures & tables
Figure 1: Motivation for lesion-set learning under dominant-label supervision. Coexisting lesions are summarized by a single dominant label in (a); single-label MIL produces one pooled winner in (b), whereas lesion-set learning retains evidence for each lesion in (c).
Figure 2: Overview of DomSet-MIL: SORR routes patch evidence in severity order, and DCL predicts the dominant label.
BRACS
EBRAINS-Mix
NSCLC-Mix
RCC-Mix
Method
M-AUC
BAcc
UER ↓
M-AUC
BAcc
M-AUC
BAcc
M-AUC
BAcc
Published MIL methods
ABMIL
0.881
0.643
0.118
0.897
0.804
0.936
0.867
0.892
0.781
CLAM-SB
0.887
0.652
0.110
0.903
0.814
0.942
0.876
0.899
0.793
DSMIL
0.894
0.661
0.101
0.914
0.827
0.950
0.887
0.908
0.806
TransMIL
0.901
0.668
0.096
0.909
0.820
0.947
0.883
0.906
0.804
Table 1: Dominant-label classification on BRACS and controlled mixed-bag tasks. Best results are bold.
Bag level
ROI level
EBRAINS-Mix
NSCLC-Mix
RCC-Mix
BRACS
Method
Set-mAP
Hidden-R
Set-mAP
Hidden-R
Set-mAP
Hidden-R
ROI-mAP
Sec-ROI-R
ABMIL
0.618
0.286
0.661
0.318
0.592
0.251
–
–
DSMIL
0.653
0.349
0.702
0.386
0.636
0.317
0.556
0.392
DTFD-MIL (AFS)
0.671
0.371
0.718
0.401
0.654
0.337
–
–
SemaMIL
0.664
0.362
0.712
0.396
0.649
0.331
–
–
Table 2: Latent-evidence recovery on controlled mixed bags and BRACS ROIs. Dashes mark unavailable ROI metrics; best results are bold.
Variant
M-AUC
BAcc
UER ↓
ROI-mAP
Sec-ROI-R
DomSet-MIL
0.916
0.690
0.071
0.625
0.500
DCL → CE
0.906
0.677
0.091
0.600
0.433
SORR → independent gates
0.911
0.684
0.077
0.610
0.462
Top- q→ mean
0.903
0.670
0.089
0.596
0.442
Top- q→ max
0.909
0.680
0.081
0.606
0.455
Top- q→ noisy-OR
0.905
0.674
0.086
0.599
0.446
Table 3: Ablation study on BRACS. Best results are bold.
Multiple instance learning (MIL) is the standard approach for whole-slide image (WSI) classification and survival prediction, where attention-based models ag gregate patch features into slide-level predictions. These models treat attention weights as evidence for their predictions, but attention is optimized for classi fication, not for identifying which patches actually support the diagnosis. This conflation leads to three failures: selected patches are insufficient (keeping them alone drops Macro-F1 by 0.078), unnecessary (removing them barely changes the prediction), and unrecoverable (continuous attention scores disagree with discrete patch subsets used at inference). The central premise is that evidence quality should be optimized directly through explicit criteria- Sufficiency, Necessity, and Recov erability (S/N/R)- rather than inherited as a byproduct of classification. GCE-MIL is a backbone-agnostic wrapper implemented through three injection modes and three evidence components: a grounding mechanism that aligns selection with domain-specific concepts, noisy-OR coverage that acts as a differentiable proxy for interventional evidence search, and threshold-plus-repair recovery that converts continuous selectors into discrete subsets through marginal-guided repair. Across 9 backbones and 9 datasets (81 configurations), GCE-MIL improves average Macro-F1 by 0.024 and C-index by 0.014, reduces the continuous-discrete gap by 4-7, and increases complement degradation by 2-4. With optional tile prefiltering after discrete recovery, inference runs up to 5 faster while retaining 0.989 full-bag utility.
Xiangyu Li, Ran Su
College of Intelligence and Computing Tianjin University
Multiple Instance Learning (MIL) is a standard paradigm for Whole-Slide Image (WSI) analysis and has achieved strong results in computational pathology. However, most MIL pipelines assume a single "gold" label per slide, which conflicts with clinical practice where substantial inter-pathologist variability is common. Existing multi-annotator learning and label-refinement methods typically estimate global annotator reliability or rely on single-instance assumptions, making them poorly suited to MIL and to localized diagnostic contexts where experts disagree. We propose RaLMPH (Reliability-aware Learning for Multi-Pathologist Harmonization), a MIL-based label reconciliation framework for WSIs annotated by multiple pathologists. RaLMPH introduces a reliability field that jointly models (i) local neighborhood structure in WSI feature space and (ii) expert uncertainty (entropy), enabling per-sample identification of trustworthy reference neighborhoods. Leveraging this field, RaLMPH performs sample-wise local annotator ranking to select reliable opinions per slide and applies an adaptive gating mechanism to fuse labels conditioned on local reliability. Experiments on a clinical WSI dataset with labels from six pathologists, as well as controlled simulated benchmarks, show that RaLMPH consistently outperforms existing approaches. Further analyses clarify how our reliability-aware mechanism improves label reconciliation and downstream MIL performance.
Sungrae Hong, Jiwon Jeong, Soeun Cheon +5
Korea Advanced Institute of Science and Technology, Daejeon, South Korea · Seegene Medical Foundation, Seoul, South Korea
Multiple Instance Learning (MIL) has emerged as a promising paradigm for Whole Slide Image (WSI) diagnosis, offering effective learning with limited annotations. However, existing MIL frameworks overlook diagnostic priorities and fail to differentiate the severity of misclassifications in multiclass, leaving clinically critical errors unaddressed. We propose a mistake-severity-aware training strategy that organizes diagnostic classes into a hierarchical structure, with each level optimized using a severity-weighted cross-entropy loss that penalizes high-severity misclassifications more strongly. Additionally, hierarchical consistency is enforced through probabilistic alignment, a semantic feature remix applied to the instance bag to robustly train class priority and accommodate clinical cases involving multiple symptoms. An asymmetric Mikel's Wheel-based metric is also introduced to quantify the severity of errors specific to medical fields. Experiments on challenging public and real-world in-house datasets demonstrate that our approach significantly mitigates critical errors in MIL diagnosis compared to existing methods. We present additional experimental results on natural domain data to demonstrate the generalizability of our proposed method beyond medical contexts.
Sungrae Hong, Jiwon Jeong, Jisu Shin +4
Korea Advanced Institute of Science and Technology, Daejeon, South Korea · Seegene Medical Foundation, Seoul, South Korea