arXiv:2501.02227eess.IVcs.CV2025-01中稿 · MICCAI 2025被引 2

用张量分解提升医学图像分割的高效微调

tCURLoRA: Tensor CUR Decomposition Based Low-Rank Parameter Adaptation and Its Application in Medical Image Segmentation

  • 将权重矩阵转为三阶张量,用CUR分解只更新低秩成分
  • 在多个医学图像分割数据集上优于现有高效微调方法
  • 适合资源受限场景下的医疗模型快速适配

迁移学习通过利用预训练模型的知识显著提升了目标任务的性能。然而,随着深度神经网络规模增大,全量微调在资源受限环境中带来巨大的计算与存储压力,限制了其广泛应用。为此,参数高效微调(PEFT)方法被提出,通过减少需更新的参数数量来降低复杂度。尽管基于矩阵分解的PEFT方法(如LoRA)表现良好,但仍难以充分捕捉模型权重的高维结构特征。相比之下,高维张量能更自然地表示神经网络权重,更完整地建模高阶特征与多维交互。本文提出tCURLoRA,一种基于张量CUR分解的新型微调方法。通过将预训练权重矩阵拼接为三维张量,并应用张量CUR分解,仅更新低秩张量成分,从而有效降低计算与存储开销。实验结果表明,tCURLoRA在医学图像分割任务中优于现有PEFT方法。

原文摘要 · Abstract (English)

Transfer learning, by leveraging knowledge from pre-trained models, has significantly enhanced the performance of target tasks. However, as deep neural networks scale up, full fine-tuning introduces substantial computational and storage challenges in resource-constrained environments, limiting its widespread adoption. To address this, parameter-efficient fine-tuning (PEFT) methods have been developed to reduce computational complexity and storage requirements by minimizing the number of updated parameters. While matrix decomposition-based PEFT methods, such as LoRA, show promise, they struggle to fully capture the high-dimensional structural characteristics of model weights. In contrast, high-dimensional tensors offer a more natural representation of neural network weights, allowing for a more comprehensive capture of higher-order features and multi-dimensional interactions. In this paper, we propose tCURLoRA, a novel fine-tuning method based on tensor CUR decomposition. By concatenating pre-trained weight matrices into a three-dimensional tensor and applying tensor CUR decomposition, we update only the lower-order tensor components during fine-tuning, effectively reducing computational and storage overhead. Experimental results demonstrate that tCURLoRA outperforms existing PEFT methods in medical image segmentation tasks.

医学图像高效微调张量分解

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