paper-with-me

홈 › Papers

Robust Multimodal Segmentation with Representation Regularization and Hybrid Prototype Distillation

2025-05-19 · Jiaqi Tan, Xu Zheng, Yang Liu

Multi-modal semantic segmentation (MMSS) faces significant challenges in real-world scenarios due to dynamic environments, sensor failures, and noise interference, creating a gap between theoretical models and practical performance. To address this, we propose a two-stage framework called RobustSeg, which enhances multi-modal robustness through two key components: the Hybrid Prototype Distillation Module (HPDM) and the Representation Regularization Module (RRM). In the first stage, RobustSeg pre-trains a multi-modal teacher model using complete modalities. In the second stage, a student model is trained with random modality dropout while learning from the teacher via HPDM and RRM. HPDM transforms features into compact prototypes, enabling cross-modal hybrid knowledge distillation and mitigating bias from missing modalities. RRM reduces representation discrepancies between the teacher and student by optimizing functional entropy through the log-Sobolev inequality. Extensive experiments on three public benchmarks demonstrate that RobustSeg outperforms previous state-of-the-art methods, achieving improvements of +2.76%, +4.56%, and +0.98%, respectively. Code is available at: https://github.com/RobustSeg/RobustSeg.

📄 PDF Abstract BibTeX arXiv:2505.12861

Code (1)

robustseg/robustseg 공식 구현 pytorch

Tasks

Knowledge DistillationSemantic Segmentation

Methods 이 논문이 사용한 방법론

Knowledge Distillation A very simple way to improve the performance of almost any machine learning algorithm is to train many different models on the same data and then to average their predictions.…
Dropout Dropout is a regularization technique for neural networks that drops a unit (along with connections) at training time with a specified probability $p$ (a common value is…

Similar Papers 제목 키워드 기반

Point Cloud Quantization through Multimodal Prompting for 3D Understanding

2025-11-15 · Hongxuan Li, Wencheng Zhu, Huiying Xu, Xinzhong Zhu 외 arxiv

Vector quantization has emerged as a powerful tool in large-scale multimodal models, unifying heterogeneous representations through discrete token encoding. However, its effectiveness hinges on robust codebook design. Cu…

Diversified Multi-prototype Representation for Semi-supervised Segmentation

2021-11-16 · Jizong Peng, Christian Desrosiers, Marco Pedersoli

This work considers semi-supervised segmentation as a dense prediction problem based on prototype vector correlation and proposes a simple way to represent each segmentation class with multiple prototypes. To avoid degen…

Segmentation

Multimodal Prototype Alignment for Semi-supervised Pathology Image Segmentation

2025-08-27 · Mingxi Fu, Fanglei Fu, Xitong Ling, Huaitian Yuan 외 arxiv

Pathological image segmentation faces numerous challenges, particularly due to ambiguous semantic boundaries and the high cost of pixel-level annotations. Although recent semi-supervised methods based on consistency regu…

Contrastive LearningImage Segmentation

Self-Regularized Prototypical Network for Few-Shot Semantic Segmentation

2022-10-30 · Henghui Ding, HUI ZHANG, Xudong Jiang

The deep CNNs in image semantic segmentation typically require a large number of densely-annotated images for training and have difficulties in generalizing to unseen object categories. Therefore, few-shot segmentation h…

Few-Shot Semantic SegmentationSegmentationSemantic Segmentation

PANet: Few-Shot Image Semantic Segmentation with Prototype Alignment

2019-08-18 · ICCV 2019 10 · Kaixin Wang, Jun Hao Liew, Yingtian Zou, Daquan Zhou 외

Despite the great progress made by deep CNNs in image semantic segmentation, they typically require a large number of densely-annotated images for training and are difficult to generalize to unseen object categories. Few…

Few-Shot Semantic SegmentationMetric LearningSegmentationSemantic Segmentation