HBFormer通过桥接机制提升微肿瘤分割的精度与上下文理解能力。
HBFormer: A Hybrid-Bridge Transformer for Microtumor and Miniature Organ Segmentation
- 采用混合编码器-解码器结构融合局部细节与全局信息
- 多尺度特征融合解码器显著提升微小病灶边界识别能力
- 适用于微肿瘤、微型器官等精细分割任务,性能领先
医学图像分割是现代临床诊断的核心。尽管基于移位窗口自注意力的视觉变换器已在该领域树立新基准,但其局部注意力机制难以有效融合局部细节与全局上下文,这对微肿瘤和微型器官分割等挑战性任务尤为不利。为此,我们提出HBFormer,一种新型混合桥接变压器架构。其'混合'设计将经典U型编码器-解码器框架与强大的Swin Transformer骨干网络结合,实现稳健的分层特征提取。核心创新在于'桥接'机制,由新型多尺度特征融合(MFF)解码器实现。不同于传统对称结构,MFF解码器通过一系列空洞卷积和深度可分离卷积构建通道与空间注意力模块,协同融合编码器多尺度特征与全局上下文信息,显式捕捉长程依赖并精炼物体边界。在多器官、肝肿瘤、膀胱肿瘤等挑战性数据集上的实验表明,HBFormer达到最先进的分割性能,充分展现其在微肿瘤与微型器官分割中的卓越能力。代码与模型可在 https://github.com/lzeeorno/HBFormer 获取。
原文摘要 · Abstract (English)
Medical image segmentation is a cornerstone of modern clinical diagnostics. While Vision Transformers that leverage shifted window-based self-attention have established new benchmarks in this field, they are often hampered by a critical limitation: their localized attention mechanism struggles to effectively fuse local details with global context. This deficiency is particularly detrimental to challenging tasks such as the segmentation of microtumors and miniature organs, where both fine-grained boundary definition and broad contextual understanding are paramount. To address this gap, we propose HBFormer, a novel Hybrid-Bridge Transformer architecture. The 'Hybrid' design of HBFormer synergizes a classic U-shaped encoder-decoder framework with a powerful Swin Transformer backbone for robust hierarchical feature extraction. The core innovation lies in its 'Bridge' mechanism, a sophisticated nexus for multi-scale feature integration. This bridge is architecturally embodied by our novel Multi-Scale Feature Fusion (MFF) decoder. Departing from conventional symmetric designs, the MFF decoder is engineered to fuse multi-scale features from the encoder with global contextual information. It achieves this through a synergistic combination of channel and spatial attention modules, which are constructed from a series of dilated and depth-wise convolutions. These components work in concert to create a powerful feature bridge that explicitly captures long-range dependencies and refines object boundaries with exceptional precision. Comprehensive experiments on challenging medical image segmentation datasets, including multi-organ, liver tumor, and bladder tumor benchmarks, demonstrate that HBFormer achieves state-of-the-art results, showcasing its outstanding capabilities in microtumor and miniature organ segmentation. Code and models are available at: https://github.com/lzeeorno/HBFormer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。