arXiv:2606.29102cs.CVcs.AI2026-06

通过双向交互提升多任务医学图像分析的分割与分类精度

BTI-Net: Bidirectional Decoder-Level Task Interaction via Uncertainty-Aware Gating for Multi-Task Medical Image Analysis

论文配图:BTI-Net: Bidirectional Decoder-Level Task Interaction via Uncertainty-Aware Gating for Multi-Task Medical Image Analysis
图 1 · 摘自论文原文
  • 在解码器各层级构建双向通信路径,融合边界与语义信息
  • 分割mIoU提升2.36,分类准确率最高提升2.26个百分点
  • 无需额外标注或推理,单次前向传播即可判断任务失败

联合分割与分类医学图像需要跨任务协同,但共享编码器架构限制了解码器仅能使用任务私有表示,永久丢失彼此提供的边界线索与语义先验。本文提出BTI-Net,通过两个并行路径在每个解码器层级建立双向通信,引入任务交互模块(TIM)。空间边界上下文被门控注入分类分支,全局语义先验则乘性调制解码器,特征从粗粒度语义逐步传递至细粒度边界细节,覆盖四个解码器分辨率。由于跨任务交互对不同输入可靠性不同,采用不确定性代理注意力(UPA)在实例和层级层面动态门控TIM输出,利用跨任务对齐、场景复杂度与预测置信度三个信号,无需外部标注或额外推理。在超声、皮肤镜与脑MRI三个医学数据集上实验表明,该方法在分割mIoU与分类准确率上均优于共享编码器与解码器交互基线。消融实验确认自适应门控比固定双向交互提升+2.36 mIoU,分类准确率最高提升+2.26。UPA的不确定性代理可作为可靠单次前向传播的任务失败信号,无需随机采样开销。

原文摘要 · Abstract (English)

Jointly learning to segment and classify medical images demands cross-task synergy, yet encoder-sharing architectures limit decoder reconstruction to task-private representations, permanently discarding the boundary cues and semantic priors each branch could supply to the other. This work introduces BTI-Net, which establishes bidirectional communication at every decoder level through two parallel pathways via Task Interaction Modules (TIM). Spatial boundary context is gated into the classification branch, while global semantic priors multiplicatively modulate the decoder, with refined features propagating progressively from coarse semantics to fine boundary detail across all four decoder resolutions. Since cross-task interaction is not equally reliable for every input, Uncertainty Proxy Attention (UPA) gates each TIM output per instance and per level using three signals that capture cross-task alignment, scene complexity, and prediction confidence, without external annotations or additional inference passes. Experiments on three medical benchmarks spanning ultrasound, dermoscopy, and brain MRI demonstrate consistent improvements in segmentation IoU and classification accuracy over both encoder-sharing and decoder-interaction baselines. Ablation confirms adaptive gating contributes +2.36 IoU over fixed bidirectional interaction, and classification accuracy improves by up to +2.26 points over the strongest multi-task baseline. UPA's uncertainty proxies serve as reliable single-pass task-failure signals without the overhead of stochastic sampling. Code: https://github.com/C-loud-Nine/BTI-Net_MTL

多任务学习医学图像解码器交互不确定性感知

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。