用极少标注实现显微图像语义分割,突破标签稀缺瓶颈。
ε-Seg: Sparsely Supervised Semantic Segmentation of Microscopy Data
- 基于分层变分自编码器与中心区域掩码,学习鲁棒特征表示。
- 仅需0.05%标签即达可比肩全监督性能,复杂结构分割准确率高。
- 无需聚类直接预测标签,适合生物医学图像少样本分割任务。
电子显微镜(EM)图像的语义分割在生命科学中仍具挑战性,因其能捕捉复杂生物结构细节,甚至令人类观察者难以应对。本文提出ε-Seg方法,基于分层变分自编码器(HVAE),结合中心区域掩码、稀疏标签对比学习(CL)、高斯混合模型(GMM)先验及无聚类标签预测。中心区域掩码与修复损失促使模型学习鲁棒且具有代表性的嵌入,即使训练标签稀疏(≤0.05%总图像数据)。通过对比学习与GMM先验优化潜空间,使编码后的图像块自然聚类于目标语义类别。最终采用MLP分类头直接从潜空间预测类别标签,避免传统聚类步骤。在两个密集生物组织的EM数据集上验证,ε-Seg在极低标签比例下仍表现优异,并成功拓展至荧光显微图像。结果表明,该方法在复杂生物图像中实现了有竞争力的稀疏监督分割性能。
原文摘要 · Abstract (English)
Semantic segmentation of electron microscopy (EM) images of biological samples remains a challenge in the life sciences. EM data captures details of biological structures, sometimes with such complexity that even human observers can find it overwhelming. We introduce ε-Seg, a method based on hierarchical variational autoencoders (HVAEs), employing center-region masking, sparse label contrastive learning (CL), a Gaussian mixture model (GMM) prior, and clustering-free label prediction. Center-region masking and the inpainting loss encourage the model to learn robust and representative embeddings to distinguish the desired classes, even if training labels are sparse (0.05% of the total image data or less). For optimal performance, we employ CL and a GMM prior to shape the latent space of the HVAE such that encoded input patches tend to cluster wrt. the semantic classes we wish to distinguish. Finally, instead of clustering latent embeddings for semantic segmentation, we propose a MLP semantic segmentation head to directly predict class labels from latent embeddings. We show empirical results of ε-Seg and baseline methods on 2 dense EM datasets of biological tissues and demonstrate the applicability of our method also on fluorescence microscopy data. Our results show that ε-Seg is capable of achieving competitive sparsely-supervised segmentation results on complex biological image data, even if only limited amounts of training labels are available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。