arXiv:2507.15793cs.CV2025-07中稿 · MICCAI 2025

动态调整低秩适配秩,提升医学图像分割少样本学习效果

Regularized Low-Rank Adaptation for Few-Shot Organ Segmentation

  • 通过l_1正则化自动学习最优低秩结构
  • 在少样本设置下对基础器官和新器官均提升分割精度
  • 适合资源受限场景下的医疗图像模型微调

参数高效微调(PEFT)在医学影像领域日益受到关注,因其高效与计算节省。其中,低秩适配(LoRA)基于适应发生在低维子空间的假设,表现良好,但其秩固定不变,难以针对不同任务选择最优值。受自然图像处理进展启发,本文提出一种动态调整内在秩的新方法。将可训练权重矩阵的低秩表示视为奇异值分解,引入l_1稀疏正则项至损失函数,并使用近端优化器求解。该正则项可视为对分解秩的惩罚,其最小化过程可自动寻找任务适配的秩。我们在真实少样本微调场景下评估该方法,对比标准LoRA及其他几种PEFT方法,涵盖两个不同任务:基础器官与新器官分割。大量实验表明,本方法显著提升性能,展现出对初始秩设定不敏感的鲁棒性与高效性。代码已公开:https://github.com/ghassenbaklouti/ARENA

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) of pre-trained foundation models is increasingly attracting interest in medical imaging due to its effectiveness and computational efficiency. Among these methods, Low-Rank Adaptation (LoRA) is a notable approach based on the assumption that the adaptation inherently occurs in a low-dimensional subspace. While it has shown good performance, its implementation requires a fixed and unalterable rank, which might be challenging to select given the unique complexities and requirements of each medical imaging downstream task. Inspired by advancements in natural image processing, we introduce a novel approach for medical image segmentation that dynamically adjusts the intrinsic rank during adaptation. Viewing the low-rank representation of the trainable weight matrices as a singular value decomposition, we introduce an l_1 sparsity regularizer to the loss function, and tackle it with a proximal optimizer. The regularizer could be viewed as a penalty on the decomposition rank. Hence, its minimization enables to find task-adapted ranks automatically. Our method is evaluated in a realistic few-shot fine-tuning setting, where we compare it first to the standard LoRA and then to several other PEFT methods across two distinguishable tasks: base organs and novel organs. Our extensive experiments demonstrate the significant performance improvements driven by our method, highlighting its efficiency and robustness against suboptimal rank initialization. Our code is publicly available: https://github.com/ghassenbaklouti/ARENA

少样本学习医学图像低秩适配参数高效

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