摘要
生物医学图像分割对于准确诊断和分析各种疾病至关重要。然而,卷积神经网络(CNNs)和Transformer是目前用于此任务的最常用架构,由于CNNs的固有局部性和Transformer的计算复杂性,它们难以有效捕捉长程依赖关系。为了解决这一限制,我们提出了TTT-Unet,这是一种将测试时训练(TTT)层集成到传统U-Net架构中的新型框架,用于生物医学图像分割。TTT-Unet在测试期间动态调整模型参数,增强了模型捕捉局部和长程特征的能力。我们在多个医学影像数据集上评估了TTT-Unet,包括CT和MR图像中的3D腹部器官分割、内窥镜图像中的器械分割以及显微镜图像中的细胞分割。结果表明,TTT-Unet在所有任务中均一致优于最先进的基于CNN和Transformer的分割模型。代码可在https://github.com/rongzhou7/TTT-Unet获取。