arXiv:2507.18481cs.CV2025-07被引 3

用预训练视觉模型构建医学异常检测新框架,无需微调即可达到顶尖性能。

Q-Former Autoencoder: A Modern Framework for Medical Anomaly Detection

  • 用冻结的DINO等模型作特征提取器,直接获取多层级高级特征。
  • 在BraTS2021、RESC、RSNA三个数据集上达当前最优,最高提升3.2%。
  • 引入Q-Former瓶颈与感知损失,适合无标注数据的医学图像异常检测。

医学图像中的异常检测因异常类型多样且难以获取全面标注数据而极具挑战。本文提出一种现代化的基于自编码器的无监督医学异常检测框架——Q-Former Autoencoder,利用DINO、DINOv2和掩码自编码器等先进预训练视觉基础模型作为特征提取器,不进行从零训练,直接冻结使用,从而获得丰富、多阶段、高层级的特征表示,无需领域特定微调。我们采用Q-Former架构作为瓶颈,可控制重建序列长度,并高效聚合多尺度特征。同时,引入基于预训练掩码自编码器特征计算的感知损失,引导重建生成语义合理的结构。该框架在四个不同的医学异常检测基准上进行了评估,在BraTS2021、RESC和RSNA上均取得当前最佳结果。实验表明,仅在自然图像上预训练的视觉基础模型,无需进一步微调即可有效泛化至医学图像分析任务。代码与模型已开源:https://github.com/emirhanbayar/QFAE。

原文摘要 · Abstract (English)

Anomaly detection in medical images is an important yet challenging task due to the diversity of possible anomalies and the practical impossibility of collecting comprehensively annotated data sets. In this work, we tackle unsupervised medical anomaly detection proposing a modernized autoencoder-based framework, the Q-Former Autoencoder, that leverages state-of-the-art pretrained vision foundation models, such as DINO, DINOv2 and Masked Autoencoder. Instead of training encoders from scratch, we directly utilize frozen vision foundation models as feature extractors, enabling rich, multi-stage, high-level representations without domain-specific fine-tuning. We propose the usage of the Q-Former architecture as the bottleneck, which enables the control of the length of the reconstruction sequence, while efficiently aggregating multiscale features. Additionally, we incorporate a perceptual loss computed using features from a pretrained Masked Autoencoder, guiding the reconstruction towards semantically meaningful structures. Our framework is evaluated on four diverse medical anomaly detection benchmarks, achieving state-of-the-art results on BraTS2021, RESC, and RSNA. Our results highlight the potential of vision foundation model encoders, pretrained on natural images, to generalize effectively to medical image analysis tasks without further fine-tuning. We release the code and models at https://github.com/emirhanbayar/QFAE.

医学图像异常检测自编码器视觉基础模型

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