arXiv:2502.16610cs.CVcs.AI2025-02被引 1

用对抗VAE检测X光片质量异常,识别设备差异导致的分布偏移。

AdverX-Ray: Ensuring X-Ray Integrity Through Frequency-Sensitive Adversarial VAEs

  • 设计轻量级对抗变分自编码器,利用生成器次优输出作为负样本训练判别器。
  • 仅用64张随机切片即达96.2%平均AUROC,显著优于现有方法。
  • 适合医疗影像实时质检,尤其适用于不同设备或参数下的图像一致性验证。

确保医学影像的质量与完整性对于基于深度学习的计算机辅助诊断与检测(CAD)系统至关重要。协变量偏移是由不同成像设备或设置引起的细微数据分布变化,会严重降低模型性能,类似对抗攻击的影响。因此,亟需一种轻量、快速的方法在使用CAD模型前评估图像质量。AdverX-Ray作为图像质量评估层,能有效检测协变量偏移。该对抗变分自编码器强调判别器的作用,利用生成器的次优输出作为负样本,微调判别器对高频伪影的识别能力。由于对抗网络生成的图像常含严重高频伪影,判别器因此被引导聚焦于这些成分,使其成为本方法的理想选择。在特定设备型号的X光图像块上训练后,AdverX-Ray可判断扫描图像是否符合训练分布,或同一设备在不同设置下采集的图像是否一致。与多种OOD检测方法的广泛对比显示,AdverX-Ray显著优于现有技术,在仅使用64张随机切片的情况下达到96.2%的平均AUROC。其轻量且快速的架构适合实时应用,提升了医学影像系统的可靠性。代码与预训练模型已公开。

原文摘要 · Abstract (English)

Ensuring the quality and integrity of medical images is crucial for maintaining diagnostic accuracy in deep learning-based Computer-Aided Diagnosis and Computer-Aided Detection (CAD) systems. Covariate shifts are subtle variations in the data distribution caused by different imaging devices or settings and can severely degrade model performance, similar to the effects of adversarial attacks. Therefore, it is vital to have a lightweight and fast method to assess the quality of these images prior to using CAD models. AdverX-Ray addresses this need by serving as an image-quality assessment layer, designed to detect covariate shifts effectively. This Adversarial Variational Autoencoder prioritizes the discriminator's role, using the suboptimal outputs of the generator as negative samples to fine-tune the discriminator's ability to identify high-frequency artifacts. Images generated by adversarial networks often exhibit severe high-frequency artifacts, guiding the discriminator to focus excessively on these components. This makes the discriminator ideal for this approach. Trained on patches from X-ray images of specific machine models, AdverX-Ray can evaluate whether a scan matches the training distribution, or if a scan from the same machine is captured under different settings. Extensive comparisons with various OOD detection methods show that AdverX-Ray significantly outperforms existing techniques, achieving a 96.2% average AUROC using only 64 random patches from an X-ray. Its lightweight and fast architecture makes it suitable for real-time applications, enhancing the reliability of medical imaging systems. The code and pretrained models are publicly available.

医学影像对抗生成质量评估异常检测

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