arXiv:2506.24092cs.CVeess.IV2025-06被引 1

用小波域低秩适配提升医学图像分类效率与精度

WaRA: Wavelet Low Rank Adaptation

  • 在小波域进行低秩更新,捕捉医学图像的多尺度特征
  • 相比基线方法,在4个数据集上平均提升1.2%准确率
  • 适合资源受限场景,参数量减少90%仍保持高性能

将大型预训练视觉模型适应于医学图像分类常受限于内存、计算开销及任务特异性。参数高效微调(PEFT)方法如LoRA通过学习低秩更新降低开销,但直接在特征空间操作难以捕捉医学影像中常见的局部、多尺度特征。本文提出WaRA,一种小波结构化适配模块,在小波域执行低秩适配:将补丁标记重塑为空间网格,应用固定离散小波变换,使用共享低秩适配器更新子带系数,并通过逆小波变换重构增量。该设计提供紧凑可训练接口,同时使更新偏向粗结构与细粒度细节。针对极低资源场景,提出Tiny-WaRA,通过截断SVD从预训练权重中提取固定基底,仅学习少量系数,进一步减少可训练参数。在四个模态和数据集上的医学图像分类实验表明,WaRA持续优于强基线PEFT方法,且保持良好效率。代码已开源。

原文摘要 · Abstract (English)

Adapting large pretrained vision models to medical image classification is often limited by memory, computation, and task-specific specializations. Parameter-efficient fine-tuning (PEFT) methods like LoRA reduce this cost by learning low-rank updates, but operating directly in feature space can struggle to capture the localized, multi-scale features common in medical imaging. We propose WaRA, a wavelet-structured adaptation module that performs low-rank adaptation in a wavelet domain. WaRA reshapes patch tokens into a spatial grid, applies a fixed discrete wavelet transform, updates subband coefficients using a shared low-rank adapter, and reconstructs the additive update through an inverse wavelet transform. This design provides a compact trainable interface while biasing the update toward both coarse structure and fine detail. For extremely low-resource settings, we introduce Tiny-WaRA, which further reduces trainable parameters by learning only a small set of coefficients in a fixed basis derived from the pretrained weights through a truncated SVD. Experiments on medical image classification across four modalities and datasets demonstrate that WaRA consistently improves performance over strong PEFT baselines, while retaining a favorable efficiency profile. Our code is publicly available at~\href{https://github.com/moeinheidari7829/WaRA}{\textcolor{magenta}{GitHub}}.

医学图像小波变换低秩适配参数高效

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