LoRA-PT: Low-rank adapting UNETR for hippocampus segmentation using principal tensor singular values and vectors.
Authors
Affiliations (3)
Affiliations (3)
- Department of Mathematics, Hangzhou Dianzi University, Hangzhou 310018, China; School of Mathematics, Physics and Information, Shaoxing University, Shaoxing 312000, China.
- School of Mathematics, Physics and Information, Shaoxing University, Shaoxing 312000, China.
- Department of Mathematics, Hangzhou Dianzi University, Hangzhou 310018, China. Electronic address: [email protected].
Abstract
The hippocampus is an important brain structure involved in various psychiatric disorders, and its automatic and accurate segmentation is vital for studying these diseases. Recently, deep learning-based methods have made significant progress in hippocampus segmentation. However, training deep neural network models requires substantial computational resources, time, and a large amount of labeled training data, which is frequently scarce in medical image segmentation. To address these issues, we propose LoRA-PT, a novel parameter-efficient fine-tuning (PEFT) method that transfers the pre-trained UNETR model from the BraTS2021 dataset to the hippocampus segmentation task. Specifically, LoRA-PT divides the parameter matrix of the transformer structure into three distinct sizes, yielding three third-order tensors. These tensors are decomposed using tensor singular value decomposition to generate low-rank tensors consisting of the principal singular values and vectors, with the remaining singular values and vectors forming the residual tensor. During fine-tuning, only the low-rank tensors (i.e., the principal tensor singular values and vectors) are updated, while the residual tensors remain unchanged. We validated the proposed method on three public hippocampus datasets, and the experimental results show that LoRA-PT outperformed state-of-the-art PEFT methods in segmentation accuracy while significantly reducing the number of parameter updates. Our source code is available at https://github.com/WangangCheng/LoRA-PT/tree/LoRA-PT.