mlsanityData to 7 Oct 2026

Uni-Encoder Meets Multi-Encoders: Representation Before Fusion for Brain Tumor Segmentation with Missing Modalities

arXiv 2604.22177 ↗First posted 24 Apr 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Multimodal MRI offers complementary information for brain tumor segmentation, but clinical scans often lack one or more modalities, which degrades segmentation performance. In this paper, we propose UniME (Uni-Encoder Meets Multi-Encoders), a two-stage heterogeneous method for brain tumor segmentation with missing modalities that reconciles the trade-offs among fine-grained structure capture, cross-modal complementarity modeling, and exploitation of available modalities. The idea is to decouple representation learning from segmentation via a two-stage heterogeneous architecture. Stage 1 pretrains a single ViT Uni-Encoder with masked image modeling to establish a unified representation robust to missing modalities. Stage 2 adds modality-specific CNN Multi-Encoders to extract high-resolution, multi-scale, fine-grained features. We fuse these features with the global representation to produce precise segmentations. Experiments on BraTS 2023 and BraTS 2024 show that UniME outperforms previous methods under incomplete multi-modal scenarios. The code is available at https://github.com/Hooorace-S/UniME

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 51 of its 73 references that are in the map. First posted on arXiv 24 Apr 2026.

Builds on

51 of 73 references.

References · 51

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  4. 2024The Llama 3 Herd of Modelscited by 14,192
  5. 2025Qwen3 Technical Reportcited by 8,195
  6. 2023Segment Anythingcited by 8,158
  7. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
  8. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  9. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
  10. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  11. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  12. 2024SAM 2: Segment Anything in Images and Videoscited by 2,784
38 more
  1. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  2. 2020GLU Variants Improve Transformercited by 1,724
  3. 2025DINOv3cited by 1,297
  4. 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
  5. 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,147
  6. 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,051
  7. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
  8. 2023Vision Transformers Need Registerscited by 844
  9. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 823
  10. 2016Language Modeling with Gated Convolutional Networkscited by 688
  11. 2021The Medical Segmentation Decathloncited by 620
  12. 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 573
  13. 2019ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networkscited by 572
  14. 2018Identifying the Best Machine Learning Algorithms for Brain Tumor Segmentation, Progression Assessment, and Overall Survival Prediction in the BRATS Challengecited by 508
  15. 2023EVA-02: A Visual Representation for Neon Genesiscited by 377
  16. 2022AMOS: A Large-Scale Abdominal Multi-Organ Benchmark for Versatile Medical Image Segmentationcited by 347
  17. 2020CHAOS Challenge -- Combined (CT-MR) Healthy Abdominal Organ Segmentationcited by 307
  18. 2023Multi-modal Learning with Missing Modality via Shared-Specific Feature Modellingcited by 159
  19. 2019Data augmentation using learned transformations for one-shot medical image segmentationcited by 138
  20. 2023Act3D: 3D Feature Field Transformers for Multi-Task Robotic Manipulationcited by 135
  21. 2022ISLES 2022: A multi-center magnetic resonance imaging stroke lesion segmentation datasetcited by 104
  22. 2016HeMIS: Hetero-Modal Image Segmentationcited by 102
  23. 2022mmFormer: Multimodal Medical Transformer for Incomplete Multimodal Learning of Brain Tumor Segmentationcited by 94
  24. 2020Hi-Net: Hybrid-fusion Network for Multi-modal MR Image Synthesiscited by 86
  25. 2019Hetero-Modal Variational Encoder-Decoder for Joint Modality Completion and Segmentationcited by 68
  26. 2024The 2024 Brain Tumor Segmentation (BraTS) Challenge: Glioma Segmentation on Post-treatment MRIcited by 65
  27. 2023Adaptive Latent Diffusion Model for 3D Medical Image to Image Translation: Multi-modal Magnetic Resonance Imaging Studycited by 64
  28. 2025Your ViT is Secretly an Image Segmentation Modelcited by 61
  29. 2023M3AE: Multimodal Representation Learning for Brain Tumor Segmentation with Missing Modalitiescited by 56
  30. 2023The Brain Tumor Segmentation (BraTS) Challenge 2023: Glioma Segmentation in Sub-Saharan Africa Patient Population (BraTS-Africa)cited by 51
  31. 2021ACN: Adversarial Co-training Network for Brain Tumor Segmentation with Missing Modalitiescited by 43
  32. 2023MMANet: Margin-aware Distillation and Modality-aware Regularization for Incomplete Multimodal Learningcited by 40
  33. 2022SMU-Net: Style matching U-Net for brain tumor segmentation with missing modalitiescited by 31
  34. 2020LT-Net: Label Transfer by Learning Reversible Voxel-wise Correspondence for One-shot Medical Image Segmentationcited by 28
  35. 2025Primus: Enforcing Attention Usage for 3D Medical Image Segmentationcited by 22
  36. 20232-D SSM: A General Spatial Layer for Visual Transformerscited by 18
  37. 2017Joint Sequence Learning and Cross-Modality Convolution for 3D Biomedical Segmentationcited by 15
  38. 2024Residual Connections Harm Generative Representation Learningcited by 10

50 most cited of 51 All 51 (JSON)

Built on by

Not cited yet.