Program now available
Registration open
>
16 - 20 February 2025
San Diego, California, US
Conference 13407 > Paper 13407-125
Paper 13407-125

Parameter-efficient fine-tuning and few-shot learning of multiscale vision transformers for liver tumour segmentation in abdominal CT scans

19 February 2025 • 5:30 PM - 7:00 PM PST | Golden State Ballroom

Abstract

Effective Parameter-Efficient Fine-Tuning (PEFT) of segmentation models for computed tomography (CT) scans is crucial for liver tumour treatment planning, including preoperative volumetrics and radioembolization dose calculations. This study integrates Low-Rank Adaptation (LoRA) into the Swin UNETR, the state-of-the-art (SOTA) abdominal segmentation model, to enhance fine-tuning efficiency with minimal CT data for liver and tumour segmentation. Evaluating various LoRA ranks and sample sizes, we found that a rank of 8 achieved optimal performance, improving the Dice score by 15% and reducing the 95th percentile Hausdorff distance (HD95) by 66% compared to traditional fine-tuning, which fine-tunes the full model's parameters. Our LoRA-enhanced model demonstrates high adaptability with limited data through a few-shot learning (FSL) approach, providing a robust solution for efficient fine-tuning of deep learning (DL) segmentation models in clinical settings.

Presenter

Ramtin Mojtahedi
Queen's Univ. (Canada)
Ramtin is a Ph.D. candidate at Simpson Lab, focusing on supervised, semi-supervised, and self-supervised deep learning methods, especially vision transformers for segmentation tasks. He earned his BSc and MSc in Electrical and Computer Engineering from the Iran University of Science and Technology (IUST) with summa cum laude distinction. Ramtin has received several awards, including the Iran’s National Elites Foundation (INEF) scholarship and Queen’s University academic excellence and leadership awards. His Ph.D. research aims to improve deep learning models in image segmentation, specifically targeting liver tumours, including primary and secondary cancer. Outside his research, Ramtin enjoys playing guitar and Santoor, swimming competitively, exploring nature, and learning about different cultures and finance.
Application tracks: AI/ML
Presenter/Author
Ramtin Mojtahedi
Queen's Univ. (Canada)
Author
Mohammad Hamghalam
Queen's Univ. (Canada)
Author
William R. Jarnagin
Memorial Sloan-Kettering Cancer Ctr. (United States)
Author
Richard K. G. Do
Memorial Sloan-Kettering Cancer Ctr. (United States)
Author
Queen's Univ. (Canada)