mlsanityData to 14 Oct 2026

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

arXiv 2606.29102 ↗First posted 27 Jun 2026cs.CV cs.AI cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

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

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 29 references that are in the map. First posted on arXiv 27 Jun 2026.

Builds on

20 of 29 references.

References · 20

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  2. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
  3. 2017Squeeze-and-Excitation Networkscited by 4,789
  4. 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,241
  5. 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
  6. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
  7. 2020Gradient Surgery for Multi-Task Learningcited by 1,256
  8. 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
  9. 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 834
  10. 2018End-to-End Multi-Task Learning with Attentioncited by 534
  11. 20183D MRI brain tumor segmentation using autoencoder regularizationcited by 328
  12. 2018PAD-Net: Multi-Tasks Guided Prediction-and-Distillation Network for Simultaneous Depth Estimation and Scene Parsingcited by 224
8 more
  1. 2018A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentationcited by 172
  2. 2020MTI-Net: Multi-Scale Task Interaction Networks for Multi-Task Learningcited by 157
  3. 2021MISSFormer: An Effective Medical Image Segmentation Transformercited by 103
  4. 2022InvPT: Inverted Pyramid Multi-task Transformer for Dense Scene Understandingcited by 89
  5. 2019A Mutual Bootstrapping Model for Automated Skin Lesion Segmentation and Classificationcited by 32
  6. 2022DenseMTL: Cross-task Attention Mechanism for Dense Multi-task Learningcited by 25
  7. 2025BRISC: Annotated Dataset for Brain Tumor Segmentation and Classificationcited by 15
  8. 2023DynaShare: Task and Instance Conditioned Parameter Sharing for Multi-Task Learningcited by 3

Built on by

Not cited yet.