arXiv:2412.09445eess.IVcs.CV2024-12被引 3

用预训练模型的嵌入表示做医学图像分类,免训练也能高精度。

Embeddings are all you need! Achieving High Performance Medical Image Classification through Training-Free Embedding Analysis

  • 用预训练模型生成图像嵌入,再用简单线性分类器判断类别。
  • 在多种医学影像上,最高提升87%的AUC-ROC指标。
  • 无需训练,计算量小,适合临床快速部署。

开发医学影像的人工智能与机器学习模型通常需要在大规模数据集上进行大量训练和测试,消耗大量计算时间、能源和资源。本文探索了用嵌入方法替代传统训练流程的可行性,该方法利用预训练基础模型(如ResNet、CLIP)生成医学图像的紧凑且语义丰富的嵌入表示,再通过简单线性分类器完成多类分类任务。评估涵盖眼底图像、乳腺钼靶、皮肤镜图像和胸部X光片等不同模态。结果表明,嵌入模型在多分类任务中比基准模型的AUC-ROC最高提升87个百分点。其中,CLIP嵌入模型表现最优,既实现更高分类性能,又显著降低计算开销。研究证明,利用预训练模型的嵌入可有效替代传统耗时耗能的训练流程,在图像分类、分割与预测中提供更高效的解决方案,有望加速AI技术在临床中的应用。

原文摘要 · Abstract (English)

Developing artificial intelligence (AI) and machine learning (ML) models for medical imaging typically involves extensive training and testing on large datasets, consuming significant computational time, energy, and resources. There is a need for more efficient methods that can achieve comparable or superior diagnostic performance without the associated resource burden. We investigated the feasibility of replacing conventional training procedures with an embedding-based approach that leverages concise and semantically meaningful representations of medical images. Using pre-trained foundational models-specifically, convolutional neural networks (CNN) like ResNet and multimodal models like Contrastive Language-Image Pre-training (CLIP)-we generated image embeddings for multi-class classification tasks. Simple linear classifiers were then applied to these embeddings. The approach was evaluated across diverse medical imaging modalities, including retinal images, mammography, dermatoscopic images, and chest radiographs. Performance was compared to benchmark models trained and tested using traditional methods. The embedding-based models surpassed the benchmark area under the receiver operating characteristic curve (AUC-ROC) scores by up to 87 percentage in multi-class classification tasks across the various medical imaging modalities. Notably, CLIP embedding models achieved the highest AUC-ROC scores, demonstrating superior classification performance while significantly reducing computational demands. Our study indicates that leveraging embeddings from pre-trained foundational models can effectively replace conventional, resource-intensive training and testing procedures in medical image analysis. This embedding-based approach offers a more efficient alternative for image segmentation, classification, and prediction, potentially accelerating AI technology integration into clinical practice.

医学影像嵌入表示免训练CLIP

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