Search papers, labs, and topics across Lattice.
This paper introduces a novel semi-supervised semantic segmentation method for remote sensing data, termed Unified Flow with Feature Memory Bank (UFFM), which addresses the imbalance in training between labeled and unlabeled data. By integrating a visual foundation model with a remote sensing domain teacher, UFFM generates less biased pseudo-labels and optimizes both labeled and pseudo-labeled data under a unified objective. Extensive experiments demonstrate that UFFM significantly outperforms state-of-the-art methods, effectively bridging the optimization and feature representation gap between labeled and unlabeled datasets.
UFFM achieves superior pseudo-label quality by harmonizing labeled and unlabeled data training, challenging the dominance of manual labeling in semi-supervised learning.
Although semi-supervised semantic segmentation ($\text{S}^4$) utilizes abundant unlabeled data to reduce manual labeling burdens, independent training of labeled and unlabeled data causes the former to dominate, which severely degrades pseudo-label quality. To address this challenges, we propose a novel remote sensing (RS) $\text{S}^4$ method via unified flow with feature memory bank (UFFM). Specifically, UFFM comprises two key innovations: unified flow (UF) and feature memory bank (FMB). The UF is a new training flow that generates less biased pseudo-labels by combining an external visual foundation model (VFM) with an RS domain teacher, and jointly optimizes labeled and pseudo-labeled data under a unified training objective. The FMB is a novel memory module for $\text{S}^4$ that dynamically updates class-specific features during training and reduces the feature discrepancy between labeled and unlabeled data through class-feature alignment. To verify the effectiveness of our model, we conduct extensive experiments on RS datasets. The experimental results show the superiority of our method over SOTA $\text{S}^4$ methods. Moreover, the results demonstrate the effectiveness of our contributions in bridging the optimization and feature representation gap between labeled and unlabeled data. Our code is released at \href{https://github.com/wangshanwen001/RS-UFFM}{https://github.com/wangshanwen001/RS-UFFM}.
