精读笔记

Problem Setting

[HASSL: Hierarchy-Aware Self-Supervised Learning Framework for Single Cell Microscopy]（arXiv preprint / 2026）

这篇论文实际处理的是单细胞显微图像表征中的“层级语义被 acquisition factor 淹没”的问题。标准 SSL 在这类数据上常会优先编码 modality、staining、batch、dataset source，因为这些因素比细胞形态更显著、更稳定、更容易优化；结果是 latent space 形成按成像条件划分的 superclusters，而不是按 biological morphology 组织的结构。

真正困难点不在于没有标签，而在于标签缺失时模型没有理由区分“成像差异”和“生物差异”。单细胞 morphology 的差异往往细微、局部、连续，并且不同层级同时存在：cell type、subtype、cell state、perturbation phenotype、imaging modality 交织在一起。已有方法卡在 flat semantic representation：DINO 类方法擅长形成粗语义簇，但不会主动维护 coarse-to-fine hierarchy；microscopy-specific SSL 多数解决 channel / assay heterogeneity，却没有显式处理层级边界。

Motivation

作者的出发点是：cell microscopy 的有效表征不应该只是“同类靠近、异类远离”，而应该允许相关类别在高层级靠近、在低层级分开。比如同一 cell type 跨 brightfield / fluorescence 应该对齐，但 OPC 与 neuron 在同一 modality 下又需要被分开。标准 contrastive / distillation objective 对这种约束表达不够细。

关键缺口是缺少 label-free hierarchy signal。监督标签太稀缺，且真实 ontology 往往不完整；纯 SSL 又会被最强 nuisance factor 牵引。HASSL 的动机就是在无显式 hierarchy annotation 的情况下，从两个弱信号里构造 inductive bias：segmentation mask 提供 morphology prior，HDBSCAN density tree 提供当前 embedding 的伪层级结构。

Core Idea

核心思想可以概括为：不要让 SSL 自己“自然发现”细胞层级，而是在训练过程中持续把 latent space 的局部密度层级反馈给模型。HASSL 把 embedding 当前的 HDBSCAN 树当成可迭代的 weak supervision：每个样本不仅属于一个 cluster，而是属于从 leaf 到 root 的一条路径；训练目标要求它贴近这条路径上的原型，同时远离每个分叉处的 sibling 原型。

这和 prior 的本质差异在于建模单位从 instance pair / flat prototype 变成 hierarchical prototype path。它引入的 inductive bias 是“细胞表型空间是嵌套的”：粗层级保留 modality 或大类结构，细层级拉开 morphology subtype。相比 HCSC 这类 hierarchy-aware contrastive 方法，HASSL 更依赖 density-based hierarchy 与 stability weighting，而不是 hard positive / negative mining；相比 CellPaint-DINO / scDINO，它新增的信息不是更多 microscopy pretraining，而是把 morphology geometry 和 latent hierarchy 显式接入训练目标。

Method

方法上最关键的不是 double teacher 和 HDBSCAN 两个模块本身，而是它们分别解决 representation collapse 到错误因素的两个阶段。

第一，segmentation-guided distillation 解决“模型一开始被 modality shortcut 吸走”的问题。mask teacher 提供的是 shape-biased target，使 image branch 的 embedding 不能只依赖 staining / illumination / channel statistics。这里 segmentation 不是下游任务，而是一个 weak morphology prior，用来改变 SSL 的信息瓶颈。

第二，HDBSCAN hierarchy-aware loss 解决“局部亚型边界不够 sharp”的问题。它在 batch embeddings 上构造层级原型，让每个 anchor 同时受到祖先 cluster 的吸引和 sibling cluster 的排斥。这个机制的核心变化是：负样本不再是全局同质的 negatives，而是在层级分叉处定义的 competing prototypes，因此不会把相关细胞类型过度推远。

第三，stability weighting 是为了避免伪层级噪声。无标签聚类一定会产生错误节点，uniform weighting 会把不稳定 cluster 当作真结构强化；HDBSCAN persistence 被用作置信度，可靠 positive 更强，不可靠 negative 更弱。这个设计比单纯 DBSCAN 或 unweighted hierarchy 更像是在做 curriculum：先信稳定结构，再逐步利用细粒度结构。

Key Insight / Why It Works

这篇最有价值的 insight 是：在 microscopy SSL 中，错误的 dominant factor 不是噪声，而是一个比目标信号更容易学习的稳定结构。因此仅靠更大模型或更多 crop augmentation 不会自动得到 morphology-aware representation；必须改变训练信号，让 morphology 和 hierarchy 在优化中有可见梯度。

最可能的核心贡献是 HDBSCAN path-level prototype loss，而不是 double teacher 本身。segmentation teacher 提供了一个有用的 morphology prior，但这类 mask-guided representation alignment 并不新；真正改变表征几何的是在每个层级分叉处约束 anchor 与 sibling prototypes 的关系。它让 latent space 从 flat nearest-neighbor retrieval 空间变成层级 margin space，这解释了为什么 multi-level datasets 和 perturbation classification 上收益更明显。

不过增益来源不应被过度解读。部分提升可能主要来自 scaling / data：2.3M single-cell crops、20 个数据源、多 modality overlap 本身就会显著改善泛化。HASSL 与 DINOv3 的比较相对公平，但与已有 microscopy foundation models 的比较仍受训练数据、预训练域、公开权重状态影响。HPA 上 CellPaint-DINO 更强也说明 domain exposure 仍然压过 objective novelty。

这里所谓 hierarchy 也不等价于 biological ontology。HDBSCAN 得到的是 density hierarchy，可能混合了 morphology、modality、batch、crop quality、segmentation error。HASSL 的成功更准确地说是：它把 density structure 变成更有约束的 retrieval geometry；它未证明模型学到了真实生物层级。

Relation To Prior Work

HASSL 属于 DINO-style self-distillation + prototypical contrastive learning + hierarchy-aware mining 的交叉谱系。它不是从零提出新的 SSL paradigm，而是把几个已有思想重新组合到单细胞显微图像的关键痛点上：DINO 负责稳定的大规模无标签训练，segmentation prior 负责 morphology bias，HDBSCAN prototypes 负责层级边界。

和 CellPaint-DINO / scDINO 的差异在于：后者主要是 domain adaptation 或 microscopy-specific pretraining，目标仍然偏通用 SSL；HASSL 显式告诉模型“哪些结构应该被保留、哪些边界应该被拉开”。和 ChadaViT / OpenPhenom 的差异在于：那些方法更多处理通道和规模异质性，HASSL 处理的是 latent organization。

和 HCSC 最接近。真正不同点是 HASSL 避免依赖语义 hard positive / negative，转而用 HDBSCAN 的 condensed tree 和 stability 来定义原型路径。实质创新在于把 density-based hierarchical clustering 作为 batch-level additive loss 接入 DINO，并用 segmentation teacher 降低 modality shortcut。看似新的部分，如 double teacher、prototype contrast、mask prior，单独看都不是全新；新意主要在组合后的 inductive bias 是否刚好匹配 single-cell morphology。

Dataset / Evaluation

数据集构建是这篇论文的重要资产。作者把多个 segmentation benchmarks 转成 single-cell crops，覆盖多 modality、多来源、部分有层级结构的数据，这比常见 MedMNIST/PathMNIST 更贴近 instance-level cell representation。这个 benchmark 确实更适合检验“跨模态 morphology representation”而不是普通 image classification。

评估上，kNN retrieval、NMI/AMI、frozen MLP classification、perturbation drug identification、HPA transfer、AICS cross-modal retrieval 基本覆盖了论文 claim 的主要侧面。尤其 multi-level subset 和 perturbation task 比整体 retrieval 更有说服力，因为它们更接近作者强调的细粒度 morphology hierarchy。

但 evaluation 仍有明显 limitation。第一，retrieval label 对未标注数据使用 dataset name pseudo-label，这会把 dataset identity 混入目标，可能与作者想削弱的 acquisition factor 有冲突。第二，HDBSCAN 学到的 hierarchy 没有和专家 ontology 或 perturbation pathway 做系统对齐验证。第三，cross-modal 和 unseen evaluation 的规模与分布细节有限，难以判断是真泛化还是 dataset overlap / morphology overlap。第四，整体指标提升不大，最强证据反而来自特定 downstream；这说明主 claim 更适合表述为“改善若干 morphology-sensitive retrieval/classification tasks”，而不是广义 microscopy foundation model breakthrough。

Limitation

最大限制是方法把一部分监督问题转移给了 segmentation 和 density clustering。segmentation mask 被当作 morphology prior，但如果 mask 质量差、细胞边界不能代表 phenotype、关键生物信号在 texture / organelle localization / intensity pattern 中，double teacher 可能会压制有用信息。文中未充分说明 segmentation error 如何影响 representation，也没有展示对不同 segmentation model 的敏感性。

第二，batch 内 HDBSCAN 的层级稳定性是隐含前提。batch size 128 下构造的 density tree 未必能代表全局 manifold；min_cluster_size 很小会增强局部结构，但也更容易形成偶然 microclusters。stability weighting 缓解了这个问题，但不能保证伪层级不会被反复自强化。

第三，泛化可能 heavily rely on data coverage。2.3M cells、20 datasets、8 modalities 是很强的数据先验；如果部署到真正新实验室、新染色方案、新细胞状态，HASSL 是否仍优于简单 DINOv3 + 更大数据并不清楚。HPA 上接近但未超过 CellPaint-DINO，说明 domain-specific exposure 依然关键。

第四，增益归因不完全清晰。double teacher 和 HDBSCAN 都有效，但二者之间可能存在耦合：segmentation teacher 先改变 embedding geometry，HDBSCAN 才能得到更合理树；反过来如果初始 embedding 被 modality dominate，HDBSCAN loss 也可能强化 modality hierarchy。论文没有充分拆解训练早期/中期/后期 hierarchy 的演化。

Takeaway

• 1. microscopy SSL 的核心问题不是缺 backbone，而是 latent space 被错误因素组织；未来方法应显式建模 acquisition factor 与 biological factor 的层级关系。
• 2. 把 clustering tree 作为训练时 weak supervision 是可迁移 insight：在标签稀缺但语义天然层级化的领域，path-level prototype loss 可能比 flat contrastive 更合适。
• 3. segmentation prior 的价值不在于提供精确 mask，而在于改变模型的 shortcut：让形态几何成为可优化目标。
• 这一思路可迁移到 medical imaging、phenotyping、材料显微等对象形态强相关任务。

一句话总结

HASSL 是一篇把 DINO 式单细胞 SSL 从 flat representation 推向 morphology-biased hierarchical representation 的工作，真正贡献在于用 segmentation prior 和 HDBSCAN path prototypes 重塑 latent geometry，而不是提出全新的自监督范式。