用粒子群优化U-Net,精准分割脑肿瘤影像
PSO-UNet: Particle Swarm-Optimized U-Net Framework for Precise Multimodal Brain Tumor Segmentation
- 用粒子群算法自动调优U-Net的滤波器数、卷积核大小和学习率
- 在BraTS 2021和Figshare数据集上达到0.9578和0.9523的Dice系数
- 参数量仅780万,运行时间约906秒,速度快且泛化能力强
医学图像分割,尤其是脑肿瘤分析,需要在复杂多模态MRI数据集和多样肿瘤形态下实现高精度与高效计算。本文提出PSO-UNet,将粒子群优化(PSO)与U-Net架构结合,实现超参数的动态优化。相比传统人工调参或其它优化方法,PSO能有效探索复杂的超参数空间,显式优化滤波器数量、卷积核大小和学习率。PSO-UNet显著提升分割性能,在BraTS 2021和Figshare数据集上分别获得0.9578和0.9523的骰子相似系数(DSC),以及0.9194和0.9097的交并比(IoU)。此外,该方法大幅降低计算复杂度,仅需780万参数,运行时间约906秒,明显快于同类U-Net框架。结果表明,PSO-UNet在多种MRI模态和肿瘤类型间具备强泛化能力,凸显其临床应用潜力及对传统调参方法的优势。未来研究将探索混合优化策略,并与其他生物启发算法对比以增强鲁棒性与可扩展性。
原文摘要 · Abstract (English)
Medical image segmentation, particularly for brain tumor analysis, demands precise and computationally efficient models due to the complexity of multimodal MRI datasets and diverse tumor morphologies. This study introduces PSO-UNet, which integrates Particle Swarm Optimization (PSO) with the U-Net architecture for dynamic hyperparameter optimization. Unlike traditional manual tuning or alternative optimization approaches, PSO effectively navigates complex hyperparameter search spaces, explicitly optimizing the number of filters, kernel size, and learning rate. PSO-UNet substantially enhances segmentation performance, achieving Dice Similarity Coefficients (DSC) of 0.9578 and 0.9523 and Intersection over Union (IoU) scores of 0.9194 and 0.9097 on the BraTS 2021 and Figshare datasets, respectively. Moreover, the method reduces computational complexity significantly, utilizing only 7.8 million parameters and executing in approximately 906 seconds, markedly faster than comparable U-Net-based frameworks. These outcomes underscore PSO-UNet's robust generalization capabilities across diverse MRI modalities and tumor classifications, emphasizing its clinical potential and clear advantages over conventional hyperparameter tuning methods. Future research will explore hybrid optimization strategies and validate the framework against other bio-inspired algorithms to enhance its robustness and scalability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。