Anatomy-Structured Hierarchical MIL for Weakly-Supervised Thoracic Disease Detection in Chest X-rays
Organizations: Division of Artificial Intelligence & Software, Ewha Womans University · Ewha Medical Artificial Intelligence Research Institute, Ewha Womans University · Department of Nuclear Medicine, Ewha Womans University · REMEDI Inc. R&D Center · Ewha Womans University Seoul Hospital
Abstract
Weakly-supervised thoracic disease detection in chest X-rays (CXR) is challenging due to subtle appearances and complex anatomical overlap, motivating anatomy-aware modeling for improved localization. However, prior anatomy-aware methods typically rely on coarse region proxies or static spatial priors, which may restrict dynamic instance discovery and limit precise localization of small abnormalities. We propose Anatomy-Structured Hierarchical Multiple Instance Learning (ASH-MIL), a framework that introduces parallel anatomy-structured observation branches (cardiac, pulmonary, and agnostic) combined with hierarchical MIL aggregation. Anatomical priors are injected as soft spatial biases into decoder cross-attention, enabling anatomically grounded evidence maps without disease bounding-box supervision. Instance localization is derived directly from MIL-weighted cross-attention maps without bounding box supervision. Experiments on CXR8 and cross-domain MIMIC-CXR held-out sets demonstrate consistent improvements over prior weakly-supervised and anatomy-aware approaches, particularly under stricter localization criteria. Our code is available at https://github.com/jn-kim/ash-mil.
Figures & tables
| IoU@0.1 | IoU@0.3 | IoU@0.5 | IoU@0.1–0.5 | |||||
| Method | AP | CorLoc | AP | CorLoc | AP | CorLoc | mAP | CorLoc |
| CXR8 | ||||||||
| ASH-MIL (Ours) | 28.82 | 0.81 | 14.23 | 0.59 | 6.03 | 0.26 | 15.77 | 0.55 |
| Ablations | ||||||||
| 3 branch (w/o prior) | 25.02 | 0.68 | 11.48 | 0.44 | 2.05 | 0.18 | 13.07 | 0.43 |
| 1 branch (w/ prior) | 22.55 | 0.60 | 8.86 | 0.32 | 3.31 | 0.13 | 11.35 | 0.34 |