arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.07576cs.CV
收藏

细粒度提示微调:一种用于高分辨率医学图像分类的参数与内存高效迁移学习方法

Fine-grained Prompt Tuning: A Parameter and Memory Efficient Transfer Learning Method for High-resolution Medical Image Classification

School of Biomedical Engineering, The University of British Columbia(不列颠哥伦比亚大学生物医学工程学院);Department of Electrical and Electronic Engineering, The University of Hong Kong(香港大学电机电子工程学系);Jiaxing Research Institute, Southern University of Science and Technology(南方科技大学嘉兴研究院);Department of Electronic and Electrical Engineering, Southern University of Science and Technology(南方科技大学电子与电气工程系)

Yijin Huang, Pujin Cheng, Roger Tam, Xiaoying Tang

AI总结:

本文提出细粒度提示微调FPT,通过冻结大模型并引入轻量侧网络与细粒度提示,在仅用1.8%可学习参数和13%内存成本下,实现高分辨率医学图像分类中与全量微调相当的性能。

摘要:

参数高效迁移学习(PETL)被提出作为一种经济高效的方式,用于将预训练模型迁移到下游任务,从而避免更新整个大规模预训练模型(LPM)所带来的高昂成本。在本工作中,我们提出了细粒度提示微调(FPT),这是一种用于医学图像分类的新型PETL方法。与其他PETL方法相比,FPT显著降低了内存消耗,尤其是在高分辨率输入场景下。为实现这一目标,我们首先冻结LPM的权重,并构建一个可学习的轻量级侧网络。冻结的LPM以高分辨率图像作为输入来提取细粒度特征,而侧网络则接收低分辨率图像以减少内存使用。为了使侧网络能够获取预训练知识,我们引入了细粒度提示,通过一个融合模块汇总来自LPM的信息。此外,还采用了重要令牌选择与预加载技术,以进一步降低训练成本和内存需求。我们在四个具有不同规模、模态和复杂度的医学数据集上对FPT进行了评估。实验结果表明,FPT在使用仅1.8%的可学习参数和13%的内存成本(相对于输入分辨率为512×512的编码器ViT-B模型)的情况下,取得了与微调整个LPM相当的性能。

英文摘要:

Parameter-efficient transfer learning (PETL) is proposed as a cost-effective way to transfer pre-trained models to downstream tasks, avoiding the high cost of updating entire large-scale pre-trained models (LPMs). In this work, we present Fine-grained Prompt Tuning (FPT), a novel PETL method for medical image classification. FPT significantly reduces memory consumption compared to other PETL methods, especially in high-resolution input contexts. To achieve this, we first freeze the weights of the LPM and construct a learnable lightweight side network. The frozen LPM takes high-resolution images as input to extract fine-grained features, while the side network is fed low-resolution images to reduce memory usage. To allow the side network to access pre-trained knowledge, we introduce fine-grained prompts that summarize information from the LPM through a fusion module. Important tokens selection and preloading techniques are employed to further reduce training cost and memory requirements. We evaluate FPT on four medical datasets with varying sizes, modalities, and complexities. Experimental results demonstrate that FPT achieves comparable performance to fine-tuning the entire LPM while using only 1.8% of the learnable parameters and 13% of the memory costs of an encoder ViT-B model with a 512 x 512 input resolution.

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

↑