arXiv:2603.14644eess.IVcs.CV2026-03中稿 · CVPR被引 1

构建多厂商乳腺钼靶数据集并提出能量归一化方法,提升AI诊断可靠性

LUMINA: A Multi-Vendor Mammography Benchmark with Energy Harmonization Protocol

  • 基于6种设备、高低能成像,标注病变与密度信息
  • 能量归一化使模型性能提升,热力图更聚焦病灶
  • 适合医学影像算法研发与临床部署研究者

公开的全视野数字乳腺钼靶(FFDM)数据集在规模、临床标注和厂商多样性方面仍显不足,制约了鲁棒模型的发展。本文提出LUMINA,一个精心构建的多厂商FFDM数据集,明确记录采集能量与厂商元数据,以捕捉现有基准中常被忽略的临床相关外观差异。该数据集包含468名患者共1824张图像(960个良性,864个恶性),具备病理确诊标签、BI-RADS评估及乳腺密度标注。数据覆盖六种成像系统,包含高能与低能两种成像模式,支持对厂商与能量引起的域偏移进行系统分析。为此,我们提出一种仅对前景像素进行空间对齐的‘能量归一化’方法,将图像映射至低能参考标准,同时保持病灶形态。我们在三个临床任务上评估了CNN与Transformer模型:诊断(良性vs恶性)、BI-RADS分类、密度估计。双视图模型始终优于单视图模型。EfficientNet-B0在诊断任务上达到93.54% AUC,Swin-T在密度预测上取得89.43%宏AUC。能量归一化提升了各架构表现,并生成更局部化的Grad-CAM响应。总体而言,LUMINA提供了(1)多样厂商基准,(2)可适配各类模型的能量归一化框架,推动可靠且可部署的乳腺钼靶AI发展。

原文摘要 · Abstract (English)

Publicly available full-field digital mammography (FFDM) datasets remain limited in size, clinical annotations, and vendor diversity, hindering the development of robust models. We introduce LUMINA, a curated, multi-vendor FFDM dataset that explicitly encodes acquisition energy and vendor metadata to capture clinically relevant appearance variations often overlooked in existing benchmarks. This dataset contains 1824 images from 468 patients (960 benign, 864 malignant), with pathology-confirmed labels, BI-RADS assessments, and breast-density annotations. LUMINA spans six acquisition systems and includes both high- and low-energy imaging styles, enabling systematic analysis of vendor- and energy-induced domain shifts. To address these variations, we propose a foreground-only pixel-space alignment method (''energy harmonization'') that maps images to a low-energy reference while preserving lesion morphology. We benchmark CNN and transformer models on three clinically relevant tasks: diagnosis (benign vs. malignant), BI-RADS classification, and density estimation. Two-view models consistently outperform single-view models. EfficientNet-B0 achieves an AUC of 93.54% for diagnosis, while Swin-T achieves the best macro-AUC of 89.43% for density prediction. Harmonization improves performance across architectures and produces more localized Grad-CAM responses. Overall, LUMINA provides (1) a vendor-diverse benchmark and (2) a model-agnostic harmonization framework for reliable and deployable mammography AI.

乳腺钼靶多厂商数据能量归一化医学AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。