arXiv:2607.04353cs.CVcs.AI2026-07

让细胞图像自监督学习更关注细微形态,而非拍摄方式等粗粒度因素。

HASSL: Hierarchy-Aware Self-Supervised Learning Framework for Single Cell Microscopy

论文配图:HASSL: Hierarchy-Aware Self-Supervised Learning Framework for Single Cell Microscopy
图 1 · 摘自论文原文
  • 用分割教师指导蒸馏,增强潜空间对细胞形态的感知。
  • 基于HDBSCAN设计层次对比损失,提升不同层级亚型的区分能力。
  • 适合需要精准细胞分型的生物医学研究者使用。

生物细胞图像普遍存在层次结构,细粒度簇常融合为更粗粒度的语义组。当前自监督学习模型常忽略这一层次性,使成像模态等粗粒度因素掩盖了精细形态特征。本文提出一种层次感知的自监督学习框架,包含两部分:基于分割教师的蒸馏框架,提升潜空间对形态的敏感性;以及基于HDBSCAN的层次感知对比损失,优化不同层级相近亚型间的决策边界。该方法减少对粗粒度因素的过度强调,使嵌入向量更契合语义与形态线索,生成由细微形态驱动的生物学有意义亚簇。我们在20个显微镜数据集整合的230万张单细胞图像上训练与评估,涵盖208个细胞类别。相比基线和对比方法,本方法平均Top-K准确率提升2.8%,在层次最深的数据集上Top-9检索率提升6.3%,下游药物分类任务中生物相关性的F1分数提升7.8%。

原文摘要 · Abstract (English)

Hierarchical structure is common in image data, where fine-grained clusters often merge into larger, coarser semantic groups. In biological cell images, current self-supervised learning models often suppress this hierarchy, as coarse factors such as imaging modality can obscure finer morphological attributes in the latent space. We propose a hierarchy-aware self-supervised training framework to address this problem. Our method combines two components: a distillation framework with a segmentation teacher to improve morphological awareness in the latent space, and a hierarchy-aware contrastive loss based on HDBSCAN to improve decision boundaries between closely related subtypes at different hierarchical levels. Together, these components reduce the tendency of self-supervised learning to overemphasize coarse factors and instead align embeddings with semantic and morphological cues. This yields biologically meaningful sub-clusters driven by fine morphological detail. We train and evaluate our method on a curated corpus of 2.3 million single cells aggregated from 20 microscopy datasets, both labeled and unlabeled, covering 208 cell classes. Our method improves over baseline and counterpart methods, increasing average top-K accuracy by 2.8%, top-9 retrieval on the dataset with the deepest hierarchy by 6.3%, and downstream F1-score for biologically relevant drug classification from perturbed cell morphology by 7.8%.

自监督学习单细胞图像层次结构形态分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。