arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.09445eess.IVcs.CV
收藏

Embeddings are all you need! 通过免训练嵌入分析实现高性能医学图像分类

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

Raj Hansini Khoiwal, Alan B. McMillan

AI总结:

该研究提出一种免训练的医学图像分类方法,利用ResNet、CLIP等预训练基础模型生成图像嵌入并结合线性分类器,在多种医学成像模态的多分类任务中AUC-ROC最高超基准模型87个百分点,大幅降低计算成本,为医学图像分析提供高效新方案。

摘要:

开发用于医学成像的人工智能(AI)和机器学习(ML)模型通常需要在大型数据集上进行大量训练和测试,会消耗大量计算时间、能源和资源。目前亟需更高效的方法,既能达到相当甚至更优的诊断性能,又不会带来相应的资源负担。我们研究了用基于嵌入的方法替代传统训练流程的可行性,该方法利用医学图像的简洁且具有语义意义的表征。我们使用预训练基础模型——具体而言是ResNet这类卷积神经网络(CNN)和对比语言-图像预训练(Contrastive Language-Image Pre-training, CLIP)这类多模态模型——为多分类任务生成图像嵌入,随后将简单线性分类器应用于这些嵌入。我们在多种医学成像模态(包括视网膜图像、乳腺X线摄影、皮肤镜图像和胸部X光片)上评估了该方法,并将其性能与使用传统方法训练和测试的基准模型进行了比较。在各类医学成像模态的多分类任务中,基于嵌入的模型的受试者工作特征曲线下面积(AUC-ROC)得分比基准模型最高高出87个百分点。值得注意的是,CLIP嵌入模型取得了最高的AUC-ROC得分,在展现出更优分类性能的同时,显著降低了计算需求。我们的研究表明,利用预训练基础模型的嵌入可以有效替代医学图像分析中传统的、资源密集型的训练和测试流程。这种基于嵌入的方法为图像分割、分类和预测提供了一种更高效的替代方案,有望加速AI技术融入临床实践的进程。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑