详细信息
文献类型:会议论文
英文题名:Enhancing Hippocampus Segmentation: SwinUNETR Model Optimization with CPS
作者:Cheng, Wangang[1];He, Guanghua[1];Zhu, Hancan[1,2]
机构:[1]Shaoxing Univ, Sch Math Phys & Informat, Shaoxing 312000, Zhejiang, Peoples R China;[2]Shaoxing Univ, Inst Artificial Intelligence, Shaoxing 312000, Zhejiang, Peoples R China
会议论文集:7th Chinese Conference on Pattern Recognition and Computer Vision
会议日期:OCT 18-20, 2024
会议地点:Urumqi, PEOPLES R CHINA
语种:英文
外文关键词:Deep learning; Hippocampus segmentation; Parameter-efficient transfer learning; SwinUNETR model
外文摘要:Deep learning techniques have made remarkable strides in medical image segmentation, overcoming many challenges associated with traditional methods. Despite their success, these techniques typically rely on large amounts of manually annotated data, which is both costly and requires expert knowledge for accurate annotations. Additionally, the need for substantial computational power, especially when processing three-dimensional images, further complicates their application. To address these challenges, this paper presents a novel optimization method called Combining Parallel and Sequential Strategy (CPS). This method leverages an efficient parameter transfer learning strategy that integrates the strengths of LoRA and Adapter. CPS can retain the original knowledge structure of the pre-trained model while updating only a minimal number of parameters, thereby reducing the risk of overfitting. We employ CPS to enhance the state-of-the-art SwinUNETR model for medical image segmentation, initially pre-trained on the BraTs2021 dataset, this enhanced model is subsequently applied to three hippocampal datasets. The results reveal that CPS significantly outperforms existing methods, increasing the Dice coefficient by an average of 1.14% and decreasing the HD95 by an average of 0.767, compared to the LoRA method. These findings highlight the effectiveness of our fine-tuning method in leveraging limited data resources, marking a significant advancement in the field of hippocampus segmentation.
参考文献:
正在载入数据...