Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Huayu Mai

dblp:354/1017 · DBLP profile ↗
← Back
17ranked-venue papers
4as first author
17since 2021 · last 2025
0000-0001-8131-6944ORCID · corroborated

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 14 · 4 first-author · 14 since 2021Graphics, computer vision, multimedia, augmented reality and games · 11 · 4 first-author · 11 since 2021Applied, interdisciplinary, general and emerging computing · 3 · 3 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
13 papers
Segmentation and scene understanding · 49% Transfer learning and domain adaptation · 10% Learning paradigms · 10%
Computer graphics and multimedia
1 paper
Multimedia analysis and retrieval · 100%

Topics — the 30 heaviest of 36, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Computer vision › Segmentation and scene understanding
semantic segmentation
4.962025
Beyond Confidence: Exploiting Homogeneous Pattern for Semi-Supervised Semantic Segmentation · ICML 2025
Towards Unbiased Learning in Semi-Supervised Semantic Segmentation · ICLR 2025
Two Losses, One Goal: Balancing Conflict Gradients for Semi-Supervised Semantic Segmentation · ICCV 2025
Computer vision › Segmentation and scene understanding › annotation-efficient segmentation
semi-supervised semantic segmentation
4.962025
Beyond Confidence: Exploiting Homogeneous Pattern for Semi-Supervised Semantic Segmentation · ICML 2025
Towards Unbiased Learning in Semi-Supervised Semantic Segmentation · ICLR 2025
Two Losses, One Goal: Balancing Conflict Gradients for Semi-Supervised Semantic Segmentation · ICCV 2025
Computer vision › Segmentation and scene understanding › biomedical image segmentation
mitochondria segmentation
2.332025
Alleviate and Mining: Rethinking Unsupervised Domain Adaptation for Mitochondria Segmentation from Pseudo-Label Perspective · AAAI 2025
RankMatch: Exploring the Better Consistency Regularization for Semi-Supervised Semantic Segmentation · CVPR 2024
DualRel: Semi-Supervised Mitochondria Segmentation from A Prototype Perspective · CVPR 2023
Machine learning › Transfer learning and domain adaptation › domain adaptation
unsupervised domain adaptation
1.722025
BeyondMix: Leveraging Structural Priors and Long-Range Dependencies for Domain-Invariant LiDAR Segmentation · NeurIPS 2025
Alleviate and Mining: Rethinking Unsupervised Domain Adaptation for Mitochondria Segmentation from Pseudo-Label Perspective · AAAI 2025
Machine learning › Trustworthy machine learning
robustness
1.122025
Relaxed Class-consensus Consistency for Semi-supervised Semantic Segmentation · AAAI 2025
Pay Attention to Target: Relation-Aware Temporal Consistency for Domain Adaptive Video Semantic Segmentation · AAAI 2024
Computer vision › Segmentation and scene understanding
biomedical image segmentation
0.912025
Alleviate and Mining: Rethinking Unsupervised Domain Adaptation for Mitochondria Segmentation from Pseudo-Label Perspective · AAAI 2025
Machine learning › Learning paradigms › semi-supervised learning
consistency regularization
0.912025
Relaxed Class-consensus Consistency for Semi-supervised Semantic Segmentation · AAAI 2025
Machine learning › Representation and self-supervised learning
contrastive learning
0.912025
Rethinking Noisy Video-Text Retrieval via Relation-aware Alignment · CVPR 2025
Machine learning › Generative modeling
diffusion model
0.912025
Towards Unbiased Learning in Semi-Supervised Semantic Segmentation · ICLR 2025
Machine learning › Representation and self-supervised learning › representation learning › invariant representation learning
domain-invariant representation
0.912025
BeyondMix: Leveraging Structural Priors and Long-Range Dependencies for Domain-Invariant LiDAR Segmentation · NeurIPS 2025
Computer vision › 3D vision › point cloud segmentation
point cloud semantic segmentation
0.912025
BeyondMix: Leveraging Structural Priors and Long-Range Dependencies for Domain-Invariant LiDAR Segmentation · NeurIPS 2025
Machine learning › Learning paradigms › weakly supervised learning
pseudo-label reliability
0.912025
Alleviate and Mining: Rethinking Unsupervised Domain Adaptation for Mitochondria Segmentation from Pseudo-Label Perspective · AAAI 2025
Machine learning › Learning paradigms › semi-supervised learning › pseudo-labeling
pseudo-label selection
0.912025
Beyond Confidence: Exploiting Homogeneous Pattern for Semi-Supervised Semantic Segmentation · ICML 2025
Multimedia analysis and retrieval › cross-modal retrieval
noisy correspondence
0.912025
Rethinking Noisy Video-Text Retrieval via Relation-aware Alignment · CVPR 2025
Multimedia analysis and retrieval › cross-modal retrieval
text-video retrieval
0.912025
Rethinking Noisy Video-Text Retrieval via Relation-aware Alignment · CVPR 2025
Machine learning › Transfer learning and domain adaptation
domain adaptation
0.812024
Pay Attention to Target: Relation-Aware Temporal Consistency for Domain Adaptive Video Semantic Segmentation · AAAI 2024
Machine learning › Transfer learning and domain adaptation › domain adaptation › visual domain adaptation
video domain adaptation
0.812024
Pay Attention to Target: Relation-Aware Temporal Consistency for Domain Adaptive Video Semantic Segmentation · AAAI 2024
Computer vision › Segmentation and scene understanding › video segmentation
video semantic segmentation
0.812024
Pay Attention to Target: Relation-Aware Temporal Consistency for Domain Adaptive Video Semantic Segmentation · AAAI 2024
Machine learning › Graph learning
affinity learning
0.712023
Appearance Prompt Vision Transformer for Connectome Reconstruction · IJCAI 2023
Computer vision › Segmentation and scene understanding
instance segmentation
0.712023
Appearance Prompt Vision Transformer for Connectome Reconstruction · IJCAI 2023
Computer vision › Video understanding and tracking › video object segmentation
memory-based video object segmentation
0.712023
Alignment Before Aggregation: Trajectory Memory Retrieval Network for Video Object Segmentation · ICCV 2023
Computer vision › Segmentation and scene understanding › medical image segmentation
semi-supervised segmentation
0.712023
DualRel: Semi-Supervised Mitochondria Segmentation from A Prototype Perspective · CVPR 2023
Machine learning › Graph learning
spatial correlation modeling
0.712023
Appearance Prompt Vision Transformer for Connectome Reconstruction · IJCAI 2023
Computer vision › Video understanding and tracking
video object segmentation
0.712023
Alignment Before Aggregation: Trajectory Memory Retrieval Network for Video Object Segmentation · ICCV 2023
Computer vision › 3D vision
volumetric image analysis
0.712023
Appearance Prompt Vision Transformer for Connectome Reconstruction · IJCAI 2023
Bioinformatics and computational biology › bioimage informatics › cell segmentation
neuron segmentation
0.712023
Appearance Prompt Vision Transformer for Connectome Reconstruction · IJCAI 2023
Machine learning › Learning paradigms
class imbalance
0.312025
Towards Unbiased Learning in Semi-Supervised Semantic Segmentation · ICLR 2025
Machine learning › Representation and self-supervised learning › representation analysis
embedding space analysis
0.312025
Beyond Confidence: Exploiting Homogeneous Pattern for Semi-Supervised Semantic Segmentation · ICML 2025
Machine learning › Trustworthy machine learning
fairness
0.312025
Towards Unbiased Learning in Semi-Supervised Semantic Segmentation · ICLR 2025
Machine learning › Learning paradigms › multi-task learning
gradient conflict
0.312025
Two Losses, One Goal: Balancing Conflict Gradients for Semi-Supervised Semantic Segmentation · ICCV 2025

Methods — techniques the papers use, named apart from their topics

consistency regularization · 2.3representative agents · 1.7ranking distribution alignment · 1.7ranking alignment · 1.7cooperative game theory · 1.7class-consensus consistency · 1.7rank-aware regularization · 0.9prototype pseudo-labeling · 0.9loss balancing · 0.9gradient conflict resolution · 0.9vision transformer · 0.7metric learning · 0.7attention · 0.7affinity learning · 0.7
YearPublicationVenuePosition
2025 Alleviate and Mining: Rethinking Unsupervised Domain Adaptation for Mitochondria Segmentation from Pseudo-Label Perspective
abstract
Mitochondria segmentation from electron microscopy (EM) images plays a crucial role in biological and medical research. However, models trained on source domains often suffer from performance degradation when applied to target domains due to domain shift. Unsupervised domain adaptation (UDA) methods have been proposed to address this issue, but they often overlook the reliability of pseudo-labels and the effectiveness of supervision signals. In this paper, we propose R4MITO, a novel UDA framework for robust mitochondria segmentation. First, we introduce Reliable Prototype Pseudo-labels to mitigate the inconsistency of class-level features between across domains by leveraging source prototypes to model target prototypes. Second, we devise Correlation-wise Consistency Regularization to exploit inter-pixel correlations, aligning agent-level correlations under various perturbations. Third, we propose Rank-aware Relationship Consistency Regularization to fully utilize the rich information encoded in inter-agent relationships by imposing rank-aware constraints on agent-ranking probability distributions. Extensive experiments on multiple EM datasets demonstrate the superiority of our R4MITO over existing state-of-the-art UDA methods for mitochondria segmentation.
Rui Sun 0006, Wangkai Li, Huayu Mai, Naisong Luo, Yuwen Pan, Tianzhu Zhang 0001
AAAI4
2025 Relaxed Class-consensus Consistency for Semi-supervised Semantic Segmentation
abstract
The key to semi-supervised semantic segmentation lies in how to fully exploit a large amount of unlabeled data to improve the model’s generalization performance. Most methods are lured into the trap of taking each class independently (i.e., class-independent consistency) and neglecting the fact that there exist semantic dependencies among classes. In this paper, we analyze the bottlenecks of class-independent consistency inherent in previous methods and offer a fresh perspective of cooperative game theory to explicitly encourage class-consensus alignment (i.e., class-consensus consistency between the teacher (weak augmented view) and student network (strong augmented view). We formulate classes as players in an cooperative game to model their interpretable consensus and shed light on the possibility of closer collaboration between consensus themselves and consistency regularization, yielding more comprehensive and effective supervision signals. To this end, we carefully design the class-consensus consistency without introducing any external knowledge to model class structure information which renders better interpretability, and further, prepend relaxed class-consensus consistency (RCC) to unlock the potential of modeling class consensus by relaxing the strict alignment of direct class consensus values to ranking alignment. Extensive experimental results on multiple benchmarks demonstrate that RCC performs favorably against state-of-the-art methods. Particularly in the low-data regimes, RCC achieves significant improvements.
Huayu Mai, Rui Sun 0006, Feng Wu 0001
AAAI1
2025 Rethinking Noisy Video-Text Retrieval via Relation-aware Alignment
abstract
Video-Text Retrieval (VTR) is a core task in multi-modal understanding, drawing growing attention from both academia and industry in recent years. While numerous VTR methods have achieved success, most of them assume accurate visual-text correspondences during training, which is difficult to ensure in practice due to ubiquitous noise, known as noisy correspondences (NC). In this paper, we rethink how to mitigate the NC from the perspective of representative reference features (termed agents), and propose a novel relation-aware purified consistency (RPC) network to amend direct pairwise correlation, including representative agents construction and relation-aware ranking distribution alignment. The proposed RPC enjoys several merits. First, to learn the agents well without any correspondence supervision, we customize the agents construction according to the three characteristics of reliability, representativeness, and resilience. Second, the ranking distribution-based alignment process leverages the structural information inherent in inter-pair relationships, making it more robust compared to individual comparisons. Extensive experiments on five datasets under different settings demonstrate the efficacy and robustness of our method.
Huakai Lai, Guoxin Xiong, Huayu Mai, Xiang Liu 0020, Tianzhu Zhang 0001
CVPR3
2025 Two Losses, One Goal: Balancing Conflict Gradients for Semi-Supervised Semantic Segmentation
Rui Sun 0006, Huayu Mai, Wangkai Li, Yuan Wang 0064
ICCV2
2025 Towards Unbiased Learning in Semi-Supervised Semantic Segmentation
abstract
Semi-supervised semantic segmentation aims to learn from a limited amount of labeled data and a large volume of unlabeled data, which has witnessed impressive progress with the recent advancement of deep neural networks. However, existing methods tend to neglect the fact of class imbalance issues, leading to the Matthew effect, that is, the poorly calibrated model’s predictions can be biased towards the majority classes and away from minority classes with fewer samples. In this work, we analyze the Matthew effect present in previous methods that hinder model learning from a discriminative perspective. In light of this background, we integrate generative models into semi-supervised learning, taking advantage of their better class-imbalance tolerance. To this end, we propose DiffMatch to formulate the semi-supervised semantic segmentation task as a conditional discrete data generation problem to alleviate the Matthew effect of discriminative solutions from a generative perspective. Plus, to further reduce the risk of overfitting to the head classes and to increase coverage of the tail class distribution, we mathematically derive a debiased adjustment to adjust the conditional reverse probability towards unbiased predictions during each sampling step. Extensive experimental results across multiple benchmarks, especially in the most limited label scenarios with the most serious class imbalance issues, demonstrate that DiffMatch performs favorably against state-of-the-art methods.
Rui Sun 0006, Huayu Mai, Wangkai Li, Tianzhu Zhang 0001
ICLR2
2025 Beyond Confidence: Exploiting Homogeneous Pattern for Semi-Supervised Semantic Segmentation
abstract
The critical challenge of semi-supervised semantic segmentation lies in how to fully exploit a large volume of unlabeled data to improve the model’s generalization performance for robust segmentation. Existing methods mainly rely on confidence-based scoring functions in the prediction space to filter pseudo labels, which suffer from the inherent trade-off between true and false positive rates. In this paper, we carefully design an agent construction strategy to build clean sets of correct (positive) and incorrect (negative) pseudo labels, and propose the Agent Score function (AgScore) to measure the consensus between candidate pixels and these sets. In this way, AgScore takes a step further to capture homogeneous patterns in the embedding space, conditioned on clean positive/negative agents stemming from the prediction space, without sacrificing the merits of confidence score, yielding better trad-off. We provide theoretical analysis to understand the mechanism of AgScore, and demonstrate its effectiveness by integrating it into three semi-supervised segmentation frameworks on Pascal VOC, Cityscapes, and COCO datasets, showing consistent improvements across all data partitions.
Rui Sun 0006, Huayu Mai, Wangkai Li, Naisong Luo, Yuan Wang 0064, Tianzhu Zhang 0001
ICML2
2025 MDPG: Multi-domain Diffusion Prior Guidance for MRI Reconstruction
Lingtong Zhang, Mengdie Song, Xiaohan Hao, Huayu Mai, Bensheng Qiu
MICCAI (2)4
2025 BeyondMix: Leveraging Structural Priors and Long-Range Dependencies for Domain-Invariant LiDAR Segmentation
abstract
Domain adaptation for LiDAR semantic segmentation remains challenging due to the complex structural properties of point cloud data. While mix-based paradigms have shown promise, they often fail to fully leverage the rich structural priors inherent in 3D LiDAR point clouds. In this paper, we identify three critical yet underexploited structural priors: permutation invariance, local consistency, and geometric consistency. We introduce BeyondMix, a novel framework that harnesses the capabilities of State Space Models (specifically Mamba) to construct and exploit these structural priors while modeling long-range dependencies that transcend the limited receptive fields of conventional voxel-based approaches. By employing space-filling curves to impose sequential ordering on point cloud data and implementing strategic spatial partitioning schemes, BeyondMix effectively captures domain-invariant representations. Extensive experiments on challenging LiDAR semantic segmentation benchmarks demonstrate that our approach consistently outperforms existing state-of-the-art methods, establishing a new paradigm for unsupervised domain adaptation in 3D point cloud understanding.
Rui Sun 0006, Wangkai Li, Huayu Mai, Zhixin Cheng, Tianzhu Zhang 0001
NeurIPS4
2025 Towards Unsupervised Domain Bridging via Image Degradation in Semantic Segmentation
abstract
Semantic segmentation suffers from significant performance degradation when the trained network is applied to a different domain. To address this issue, unsupervised domain adaptation (UDA) has been extensively studied. Despite the effectiveness of selftraining techniques in UDA, they still overlook the explicit modeling of domain-shared feature extraction. In this paper, we propose DiDA, an unsupervised domain bridging approach for semantic segmentation. DiDA consists of two key modules: (1) Degradation-based Intermediate Domain Construction, which creates continuous intermediate domains through simple image degradation operations to encourage learning domain-invariant features as domain differences gradually diminish; (2) Semantic Shift Compensation, which leverages a diffusion encoder to disentangle and compensate for semantic shift information with degraded time-steps, preserving discriminative representations in the intermediate domains. As a plug-and-play solution, DiDA supports various degradation operations and seamlessly integrates with existing UDA methods. Extensive experiments on multiple domain adaptive semantic segmentation benchmarks demonstrate that DiDA consistently achieves significant performance improvements across all settings. Code is available at https://github.com/Woof6/DiDA.
Wangkai Li, Rui Sun 0006, Huayu Mai, Tianzhu Zhang 0001
NeurIPS3
2025 OCELOT 2023: Cell detection from cell-tissue interaction challenge
Jaewoong Shin, Jeongun Ryu, Aaron Valero Puche, Biagio Brattoli, Wonkyung Jung, Soo Ick Cho, Kyunghyun Paeng, Chan-Young Ock, Donggeun Yoo, Wangkai Li, Huayu Mai, Joshua Millward, Zhen He 0002, Aiden Nibali, Lydia A. Schoenpflug, Viktor H. Koelzer, Shuoyu Xu, Ji Zheng, Yu-Wen Lo, Ching-Hui Yang, Sérgio Pereira
Medical Image Anal.13
2024 Pay Attention to Target: Relation-Aware Temporal Consistency for Domain Adaptive Video Semantic Segmentation
abstract
Video semantic segmentation has achieved conspicuous achievements attributed to the development of deep learning, but suffers from labor-intensive annotated training data gathering. To alleviate the data-hunger issue, domain adaptation approaches are developed in the hope of adapting the model trained on the labeled synthetic videos to the real videos in the absence of annotations. By analyzing the dominant paradigm consistency regularization in the domain adaptation task, we find that the bottlenecks exist in previous methods from the perspective of pseudo-labels. To take full advantage of the information contained in the pseudo-labels and empower more effective supervision signals, we propose a coherent PAT network including a target domain focalizer and relation-aware temporal consistency. The proposed PAT network enjoys several merits. First, the target domain focalizer is responsible for paying attention to the target domain, and increasing the accessibility of pseudo-labels in consistency training. Second, the relation-aware temporal consistency aims at modeling the inter-class consistent relationship across frames to equip the model with effective supervision signals. Extensive experimental results on two challenging benchmarks demonstrate that our method performs favorably against state-of-the-art domain adaptive video semantic segmentation methods.
Huayu Mai, Rui Sun 0006, Yuan Wang 0064, Tianzhu Zhang 0001, Feng Wu 0001
AAAI1
2024 RankMatch: Exploring the Better Consistency Regularization for Semi-Supervised Semantic Segmentation
abstract
The key lie in semi-supervised semantic segmentation is how to fully exploit substantial unlabeled data to im-prove the model's generalization performance by resorting to constructing effective supervision signals. Most methods tend to directly apply contrastive learning to seek additional supervision to complement independent regular pixel-wise consistency regularization. However, these methods tend not to be preferred ascribed to their complicated designs, heavy memory footprints and susceptibility to confirmation bias. In this paper, we analyze the bottlenecks exist in con-trastive learning-based methods and offer a fresh perspective on inter-pixel correlations to construct more safe and effective supervision signals, which is in line with the nature of semantic segmentation. To this end, we develop a coherent RankMatch network, including the construction of representative agents to model inter-pixel correlation beyond regular individual pixel-wise consistency, and fur-ther unlock the potential of agents by modeling inter-agent relationships in pursuit of rank-aware correlation consis-tency. Extensive experimental results on multiple bench-marks, including mitochondria segmentation, demonstrate that RankMatch performs favorably against state-of-the-art methods. Particularly in the low-data regimes, RankMatch achieves significant improvements.
Huayu Mai, Rui Sun 0006, Tianzhu Zhang 0001, Feng Wu 0001
CVPR1
2023 DualRel: Semi-Supervised Mitochondria Segmentation from A Prototype Perspective
abstract
Automatic mitochondria segmentation enjoys great popularity with the development of deep learning. However, existing methods rely heavily on the labor-intensive manual gathering by experienced domain experts. And naively applying semi-supervised segmentation methods in the natural image field to mitigate the labeling cost is undesirable. In this work, we analyze the gap between mitochondrial images and natural images and rethink how to achieve effective semi-supervised mitochondria segmentation, from the perspective of reliable prototype-level supervision. We propose a novel end-to-end dual-reliable (DualRel) network, including a reliable pixel aggregation module and a reliable prototype selection module. The proposed DualRel enjoys several merits. First, to learn the prototypes well without any explicit supervision, we carefully design the referential correlation to rectify the direct pairwise correlation. Second, the reliable prototype selection module is responsible for further evaluating the reliability of prototypes in constructing prototype-level consistency regularization. Extensive experimental results on three challenging benchmarks demonstrate that our method performs favorably against state-of-the-art semi-supervised segmentation methods. Importantly, with extremely few samples used for training, DualRel is also on par with current state-of-the-art fully supervised methods.
Huayu Mai, Rui Sun 0006, Tianzhu Zhang 0001, Zhiwei Xiong, Feng Wu 0001
CVPR1
2023 Alignment Before Aggregation: Trajectory Memory Retrieval Network for Video Object Segmentation
abstract
Memory-based methods in semi-supervised video object segmentation task achieve competitive performance by performing dense matching between query and memory frames. However, most of the existing methods neglect the fact that videos carry rich temporal information yet redundant spatial information. In this case, direct pixel-level global matching will lead to ambiguous correspondences. In this work, we reconcile the inherent tension of spatial and temporal information to retrieve memory frame information along the object trajectory, and propose a novel and coherent Trajectory Memory Retrieval Network (TMRN) to equip with the trajectory information, including a spatial alignment module and a temporal aggregation module. The proposed TMRN enjoys several merits. First, TMRN is empowered to characterize the temporal correspondence which is in line with the nature of video in a data-driven manner. Second, we elegantly customize the spatial alignment module by coupling SVD initialization with agent-level correlation for representative agent construction and rectifying false matches caused by direct pairwise pixel-level correlation, respectively. Extensive experimental results on challenging benchmarks including DAVIS 2017 validation / test and Youtube-VOS 2018/2019 demonstrate that our TMRN, as a general plugin module, achieves consistent improvements over several leading methods.
Rui Sun 0006, Yuan Wang 0064, Huayu Mai, Tianzhu Zhang 0001, Feng Wu 0001
ICCV3
2023 Appearance Prompt Vision Transformer for Connectome Reconstruction
abstract
Neural connectivity reconstruction aims to understand the function of biological reconstruction and promote basic scientific research. The intricate morphology and densely intertwined branches make it an extremely challenging task. Most previous best-performing methods adopt affinity learning or metric learning. Nevertheless, they either neglect to model explicit voxel semantics caused by implicit optimization or are hysteresis to spatial information. Furthermore, the inherent locality of 3D CNNs limits modeling long-range dependencies, leading to sub-optimal results. In this work, we propose a coherent and unified Appearance Prompt Vision Transformer (APViT) to integrate affinity and metric learning to exploit the complementarity by learning long-range spatial dependencies. The proposed APViT enjoys several merits. First, the extension continuity-aware attention module aims at constructing hierarchical attention customized for neuron extensibility and slice continuity to learn instance voxel semantic context from a global perspective and utilize continuity priors to enhance voxel spatial awareness. Second, the appearance prompt modulator is responsible for leveraging voxel-adaptive appearance knowledge conditioned on affinity rich in spatial information to instruct instance voxel semantics, exploiting the potential of affinity learning to complement metric learning. Extensive experimental results on multiple challenging benchmarks demonstrate that our APViT achieves consistent improvements with huge flexibility under the same post-processing strategy.
Rui Sun 0006, Naisong Luo, Yuwen Pan, Huayu Mai, Tianzhu Zhang 0001, Zhiwei Xiong, Feng Wu 0001
IJCAI4
2023 Structure-Decoupled Adaptive Part Alignment Network for Domain Adaptive Mitochondria Segmentation
Rui Sun 0006, Huayu Mai, Naisong Luo, Tianzhu Zhang 0001, Zhiwei Xiong, Feng Wu 0001
MICCAI (4)2
2023 DAW: Exploring the Better Weighting Function for Semi-supervised Semantic Segmentation
abstract
The critical challenge of semi-supervised semantic segmentation lies in how to fully exploit a large volume of unlabeled data to improve the model’s generalization performance for robust segmentation. Existing methods tend to employ certain criteria (weighting function) to select pixel-level pseudo labels. However, the trade-off exists between inaccurate yet utilized pseudo-labels, and correct yet discarded pseudo-labels in these methods when handling pseudo-labels without thoughtful consideration of the weighting function, hindering the generalization ability of the model. In this paper, we systematically analyze the trade-off in previous methods when dealing with pseudo-labels. We formally define the trade-off between inaccurate yet utilized pseudo-labels, and correct yet discarded pseudo-labels by explicitly modeling the confidence distribution of correct and inaccurate pseudo-labels, equipped with a unified weighting function. To this end, we propose Distribution-Aware Weighting (DAW) to strive to minimize the negative equivalence impact raised by the trade-off. We find an interesting fact that the optimal solution for the weighting function is a hard step function, with the jump point located at the intersection of the two confidence distributions. Besides, we devise distribution alignment to mitigate the issue of the discrepancy between the prediction distributions of labeled and unlabeled data. Extensive experimental results on multiple benchmarks including mitochondria segmentation demonstrate that DAW performs favorably against state-of-the-art methods.
Rui Sun 0006, Huayu Mai, Tianzhu Zhang 0001, Feng Wu 0001
NeurIPS2