用多数据集清洁标签训练胸部X光模型,提升泛化能力。
HarMoE: Multi-Source Chest Radiograph Pretraining with Dataset-Disentangled Experts

- 设计分治专家网络,分离数据源差异与通用医学语义。
- 在87.3万张图像上训练,零样本分类准确率超基线。
- 适合需要跨数据集泛化的医疗AI研究者使用。
当前胸部X光视觉-语言模型主要依赖图像与报告对齐,高度依赖MIMIC-CXR数据集。然而,这种范式忽视了多标签分类数据集提供的更清晰、明确的疾病信号及更广的病理覆盖。面对异构数据源的标签体系、标注协议和成像流程差异,模型易将临床语义与数据集身份混淆,导致迁移性能差。本文提出HarMoE,一种面向多源数据的混合专家框架,通过共享专家学习跨数据集医学语义,将源特异性差异限制在深层解码器的轻量残差专家中。进一步采用统一疾病词表与掩码多数据集监督,实现互补标注的利用而不引入假负例。在大规模胸部X光基准测试中,HarMoE持续优于强基线,在零样本分类、分布外迁移和定位任务上表现更优。结果表明,构建鲁棒放射科视觉-语言模型需从单一来源对齐转向基于异构数据的结构化知识构建。代码与87.3万张增强数据集将开源。
原文摘要 · Abstract (English)
Recent vision-language models for chest X-ray understanding are largely built on image-report alignment and therefore rely heavily on MIMIC-CXR as the dominant pretraining source. While effective at scale, this paradigm underexplores an important alternative source of supervision: a range of existing multi-label classification datasets, which provide cleaner and more explicit disease signals than free-text reports, and can offer broader pathology coverage when combined across sources. However, learning from such heterogeneous datasets is nontrivial, as differences in label ontologies, annotation protocols, acquisition pipelines, and report styles can cause models to entangle clinical semantics with dataset identity, leading to poor transfer despite increased scale. In this work, we revisit radiology VLM construction from the perspective of harmonized multi-source learning. We propose HarMoE, a dataset-aware mixture-of-experts framework that learns shared cross-dataset medical semantics while confining source-specific variation to lightweight residual experts in deeper decoder layers. To further exploit clean supervision from labeled datasets, we train in a unified disease vocabulary with masked multi-dataset supervision, enabling the model to leverage complementary annotations without introducing false negatives. Experiments on large-scale chest X-ray benchmarks show that HarMoE consistently improves zero-shot classification, out-of-distribution transfer, and grounding over strong baselines. Our results suggest that building robust radiology VLMs requires moving beyond single-source image-report alignment toward structured knowledge construction from heterogeneous datasets with cleaner supervision and broader coverage. Code and the 873k harmonized dataset will be released at https://github.com/Roypic/harmoe.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。