LLM2D
TTT-Unet:通过测试时训练层增强U-Net以提升生物医学图像分割效果
TTT-Unet: Enhancing U-Net with Test-Time Training Layers for Biomedical Image Segmentation
发布日期: 9/20/2024
arXiv ID: oai:arXiv.org:2409.11299v2

摘要

生物医学图像分割对于准确诊断和分析各种疾病至关重要。然而,卷积神经网络(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获取。