摘要
arXiv:2503.19152v1 交叉公告类型
摘要:医学图像分割,特别是针对脑肿瘤分析,需要精确且计算高效的模型,因为多模态MRI数据集和多样的肿瘤形态非常复杂。本研究引入了PSO-UNet,该模型结合了粒子群优化(PSO)和U-Net架构,用于动态超参数优化。与传统的手动调参或替代优化方法不同,PSO能够有效导航复杂的超参数搜索空间,显式地优化滤波器的数量、卷积核的尺寸和学习率。PSO-UNet显著增强了分割性能,在BraTS 2021和Figshare数据集上分别实现了Dice相似系数(DSC)为0.9578和0.9523,以及交并比(IoU)分别为0.9194和0.9097。此外,该方法显著降低了计算复杂度,仅使用了780万参数并在大约906秒内完成执行,明显快于基于U-Net的其他框架。这些结果表明,PSO-UNet具有强大的跨不同MRI模态和肿瘤分类的一般化能力,突显了其在临床中的潜力和相对于传统超参数调优方法的明显优势。未来的研究将探索混合优化策略,并与其他启发式算法进行比较验证,以增强其稳健性和可扩展性。