Jan, 2024

注意力引导去除:提升下游乳房密度分类的一种新的增强方法

TL;DR该研究介绍了一种名为注意力引导擦除(AGE)的新型数据增强技术,旨在改善越南人群中根据 BI-RADS 推荐对乳腺 X 线摄影图像中四种不同乳腺密度类别的下游分类。该方法在迁移学习过程中整合了自我监督 DINO 方法训练的视觉变换器背骨网络生成的视觉注意力图,用于擦除乳腺 X 线摄影图像中的背景区域,只显示出潜在的密集乳腺组织区域。通过在迁移学习过程中采用 AGE 并调整不同的随机概率,我们始终优于没有 AGE 和传统随机擦除变换的分类性能。我们使用公开数据集 VinDr-Mammo 验证了我们的方法,其中平均 F1 分数为 0.5910,超过了没有 AGE 和带有随机擦除(RE)的情况下的 0.5594 和 0.5691 的值。t 检验进一步证明了我们方法的统计显著性,显示出 p 值小于 0.0001。