精读笔记
Problem Setting
这篇论文不是在泛泛做 WSI 回归,而是在解决 PD-L1 TPS 弱监督估计中最棘手的边界状态:非表达/近零表达样本。TPS 的临床阈值使得 0 或 <1% 不是普通连续变量端点,而是诊断决策中的特殊状态。真正困难在于 slide-level 标签只能告诉模型整张片子的比例,不能告诉哪些 tumor cells 阳性、哪些是炎症细胞阳性、哪些 patch 是伪影或无诊断意义区域。以前的 MIL/回归路线通常把所有东西压成一个连续分数,导致低表达区域被平均化,0 类被当成回归误差的一部分处理,模型很难区分“确实无表达”和“看不准但预测很低”。关键矛盾是:临床需要稳定识别离散阈值状态,而训练信号却是粗粒度、噪声大、比例型的 slide-level 标签。
Motivation
已有路线不够的原因不是 attention 不够复杂,而是目标变量建模错位。TPS 天然是比例变量,且在 0 处有额外质量;用 MSE 直接回归会默认误差结构近似连续、对称、同方差,这和 PD-L1 scoring 的实际分布不匹配。作者的核心观察是:非表达样本经常是自动方法的薄弱点,而这个薄弱点可以被看作分布建模问题,而不是单纯的 backbone 或 MIL aggregator 问题。关键缺口是现有模型缺少内生的 confidence/uncertainty 表达,也缺少对 TPS=0 这一特殊状态的显式参数化。因此他们把问题从“预测一个 TPS 数值”改成“预测一个带零膨胀的 TPS 概率分布”。
Core Idea
论文真正核心的思想是:把 TPS 估计从点回归变成 zero-inflated beta 分布参数估计。phi 捕捉是否为真正零表达,mu 捕捉正表达部分的比例,nu 捕捉分布集中程度。这个建模把两个本来混在一起的问题拆开:一是 slide 是否属于非表达状态,二是在非零表达条件下表达比例是多少。相比普通 MIL 回归,这个设计显式改变了输出空间的几何结构,让模型知道 0 是一个离散事件,而不是 [0,1] 区间上的普通数值。
直觉上它可能有效,因为 TPS 同时具有比例约束、边界效应、零膨胀和异方差。Beta 分布比 Gaussian/MSE 更适合表达比例变量的不确定性,zero-inflation 则让模型有能力把“完全没有表达”作为独立状态建模。它和 prior 的本质区别不在于用了 gated attention,而在于引入了任务层面的统计 inductive bias:让模型输出诊断变量的分布结构,而不是只在 embedding 空间里学一个回归器。
Method
方法中值得保留的机制很少。第一,外部 MIHIC embedding 用来提供 tumor/background/tissue 类型相关的视觉表征,解决小规模 TPS 数据不足以从头学习组织语义的问题;它本质上是 representation reuse,而不是新 MIL 机制。第二,attention-based MIL 用于从 4096 个 patch 中压制背景、边缘、伪影和低信息 patch,解决 slide-level 标签下有效实例稀疏的问题;它把 bag aggregation 从均匀平均改成可解释的信息选择。第三,patch-wise TPS prediction 加 attention weighted sum 让局部表达估计先形成,再汇总到 slide 级比例,符合 TPS 作为细胞/区域比例的结构。第四,ZIBeta head 和 NLL loss 让零表达与正表达走不同似然路径,避免所有样本都被同一个 MSE 目标拉扯。第五,patch entropy regularization 试图避免模型在所有 patch 上输出平滑中间值,强迫 patch-level 预测更接近局部阳性/阴性状态;这个假设合理,但也可能过强,因为真实 patch 内可能存在混合细胞群。
Key Insight / Why It Works
最可能的核心贡献是对标签分布的重参数化,而不是网络结构。TPS 的困难来自弱监督比例标签和特殊零点;ZIBeta 正好把这两个结构性问题写进了输出层。phi 让模型不必用一个很小的 mu 去模拟零表达,从而减轻低 TPS 和真正 0 TPS 的混淆;nu 则提供了一个可被解释为 concentration 的自由度,使模型在“看得准”和“只是给出均值”之间有表达空间。这属于 better inductive bias,而不是 scaling、retrieval 或 test-time compute。
attention 的作用更像必要的噪声过滤器。WSI 中大量 patch 与 TPS 无关,简单 mean pooling 会把正表达证据稀释掉,特别是在 tissue edge、背景、artifact 和炎症细胞干扰下。gated attention 能把模型容量集中到信息密集区域,因此零类和中高表达类都可能受益。但 attention 不是充分解释:没有 cell-level ground truth 时,attention 高不等于模型真的看到了正确的 tumor-cell membrane staining,只能算 sanity check。
外部 embedding 的贡献可能被低估,也可能是主要工程来源之一。MIHIC 上训练的 MobileNetV3 已经把组织类型和 tumor patch 识别做好,后续 MIL 只是在较好的 patch representation 上学习比例映射。由于目标数据只有 162 例,性能很可能 heavily rely on representation alignment 和数据覆盖,而不是 ZIBeta 本身单独带来的能力。增益来源不清:ZIBeta 相比 MIL_att 的提升很小,且部分指标不占优,因此不能把全部效果归因于分布式建模。
nu 作为置信度的 claim 需要谨慎。它与误差有负相关,但关系噪声大,且不同 TPS 类保留率差异明显:高 concentration 更多保留零类,少保留高表达类。这可能说明 nu 同时编码了样本难度、类别频率和模型偏置,而不只是 epistemic 或 aleatoric uncertainty。换言之,目前它是有用的 ranking signal,但不是校准后的可靠 uncertainty estimate。
Relation To Prior Work
这篇工作最接近三条路线:attention MIL for pathology、PD-L1 TPS automated scoring、distributional regression/zero-inflated modeling。它没有在 MIL aggregator 上提出本质新结构,gated attention 和 patch-level prediction 都是已有思想的组合;也没有像 cell detection/segmentation 方法那样直接建模 tumor cell 与 positive staining 的细粒度计数。真正新增的信息是把 TPS 的统计结构显式放进 MIL 输出层,用 zero-inflated beta 来处理 0 类和比例型不确定性。
与传统自动 PD-L1 scoring 相比,它放弃了细胞级强监督或复杂标注,转向 slide-level 弱监督,因此更 scalable,但代价是可验证性下降:模型是否真的在数 tumor cells,文中不能证明。与普通 MIL 回归相比,它的本质差异是目标分布建模,而不是 attention 可视化。与不确定性估计工作相比,它不是 Bayesian ensemble 或 MC dropout,而是通过输出分布参数获得内生 concentration;这更轻量,但校准性也更弱。整体上它属于“任务分布假设 + 弱监督 MIL”的谱系,而不是 foundation model 或大规模预训练驱动的路线。
Dataset / Evaluation
评估覆盖范围偏窄。目标 TPS 数据只有 162 个病例,单中心、单疾病,几乎全为 SP263 染色,只有极少数 22C3;这不足以验证跨染色、跨中心、跨扫描仪或跨临床流程泛化。MIHIC 外部数据用于 embedding extraction,提供了一定 open-data 支撑,但它验证的是组织表征,不等于验证 PD-L1 TPS scoring 的泛化。
实验确实支持“深度 MIL 优于线性 mean pooling”和“attention 对信息过滤有帮助”。ZIBeta 对零类 F1 和回归误差有改善,也支持 zero-inflation 的方向。但核心 claim 中的 explainability 和 uncertainty 只被弱验证:attention patch 可视化是定性证据,nu-error 曲线是经验相关,不是校准评估。没有外部 test set、多专家标签、细胞级标注对照或临床阈值决策曲线,因此 benchmark 只能说明该 pipeline 在这个小 cohort 上有效,不能证明可部署级别的鲁棒性。
Limitation
方法成立依赖几个强前提。第一,MIHIC 学到的 tissue embedding 必须和目标 PD-L1 IHC slide 的视觉语义对齐;一旦染色、扫描、组织处理或 tumor morphology 发生 domain shift,MIL 头可能没有足够数据自行修正。第二,slide-level TPS 标签必须足够可靠,但文中标签来自单一 certified pathologist,高 TPS 还被离散到 10% 间隔,这会直接限制分布建模的上限。第三,zero-inflated beta 假设 TPS=0 是一个可分离的离散状态,但真实标注中的 <1%、0%、未采到阳性区域、病理医生估计误差可能混在一起,phi 学到的不一定是纯生物状态。
泛化文中未充分说明,特别是 phi 阈值需要在 validation set 上按 Kappa 调,说明零类决策仍依赖数据集特定校准。所谓 uncertainty 也没有经过 calibration、risk-coverage、external validation 或 subgroup analysis;nu 可能只是类别难度和数据分布的 proxy。增益来源不清:ZIBeta 相比 MIL_att 的数值优势很小,Kappa 还低于 MIL_base,因此不能排除主要增益来自 attention、external embedding、patch sampling 或数据 split 偶然性。更本质地说,这个方法把细胞级计数问题转移成了弱监督分布拟合问题,降低了标注成本,但没有消除病理证据归因的不确定性。
Takeaway
- 最值得记住的不是具体 architecture,而是 TPS 这类医学比例标签应按其统计结构建模:零膨胀、边界约束、异方差和临床阈值都应该进入目标函数,而不是事后靠回归器拟合。
- 第二,弱监督 pathology 中 attention 解决的是有效证据选择,distributional head 解决的是标签空间错配,两者是互补的。
- 第三,nu/浓度作为 confidence proxy 是有迁移价值的方向,但必须走向严格 calibration 和 external risk-coverage 验证,否则只是可解释的模型内部分数。
- 第四,未来真正值得做的是多中心、多染色、多专家标签下验证 zero-inflated distributional MIL,并与 cell-level counting 或 foundation pathology embeddings 结合,判断它到底是在学诊断机制,还是在小数据上学 cohort-specific shortcut。
一句话总结
这篇论文是弱监督数字病理 TPS 估计中一次有价值的目标分布重建尝试:它的实质贡献是把 PD-L1 非表达状态和比例型不确定性显式写进 MIL 输出层,而不是单纯改进 WSI 聚合网络。
