用合成图像和自监督预训练提升浮游生物实例分割精度
PlankFormer: Robust Plankton Instance Segmentation via MAE-Pretrained Vision Transformers and Pseudo Community Image Generation

- 通过合成浮游生物图像生成伪群落数据,缓解标注数据不足问题
- 基于MAE预训练的ViT模型在高杂物密度下仍保持高精度分割
- 适合生态监测、自动化显微分析等需要少标注的场景
浮游生物监测对评估水生生态系统至关重要,但依赖人工显微分析,耗时费力。自动分割拥挤图像中的浮游生物面临两大挑战:(i) 像素级标注数据稀缺;(ii) 传统基于CNN的方法难以区分浮游生物与碎片及重叠个体。为此,我们提出PlankFormer框架。首先,为克服数据短缺,提出一种方法:将单个浮游生物图像合成到多样背景中(包括生成模型创建的背景),生成带标签的伪群落图像(PCI)。其次,设计一个采用ViT主干和Mask2Former解码器的分割模型。为鲁棒捕捉浮游生物全局结构特征以应对遮挡和碎片干扰,使用掩码自编码器(MAE)对未标注的单个浮游生物图像进行自监督预训练。在真实世界数据集上的实验表明,该方法显著优于传统方法(如Mask R-CNN),尤其在高杂物密度环境中表现突出。结果证明,我们的合成训练策略与MAE架构结合,可在减少人工标注的前提下实现高精度分割。
原文摘要 · Abstract (English)
Plankton monitoring is essential for assessing aquatic ecosystems but is limited by the labor-intensive nature of manual microscopic analysis. Automating the segmentation of plankton from crowded images is crucial, however, it faces two major challenges: (i) the scarcity of pixel-level annotated datasets and (ii) the difficulty of distinguishing plankton from debris and overlapping individuals using conventional CNN-based methods. To address these issues, we propose PlankFormer, a novel framework for plankton instance segmentation. First, to overcome the data shortage, we introduce a method to generate labeled Pseudo Community Images (PCI) by synthesizing individual plankton images onto diverse backgrounds, including those created by generative models. Second, we propose a segmentation model utilizing a Vision Transformer (ViT) backbone with a Mask2Former decoder. To robustly capture the global structural features of plankton against occlusion and debris, we employ a Masked Autoencoder (MAE) for self-supervised pre-training on unlabeled individual images. Experimental results on real-world datasets demonstrate that our method significantly outperforms conventional methods, such as Mask R-CNN, particularly in challenging environments with high debris density. We demonstrate that our synthetic training strategy and MAE-based architecture enable high-precision segmentation with requiring less manual annotations for individual plankton images.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。