SAGE: Semantic Explainability of Attention-Based Survival Models in Computational Pathology
本文介绍了 SAGE,这是一个事后(post-hoc)框架,它利用病理视觉-语言模型为计算病理学中的注意力机制生存模型生成全局性的、基于语言的解释,从而揭示特定的组织学概念如何驱动不同患者队列的预测,并实现生物标志物的发现。
原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明
想象一下你是一名正在试图破解谜团的侦探,但你的“犯罪现场”不是一个犯罪现场,而是一个在显微镜下被放大了数千倍的、来自患者身体的微小且色彩斑斓的组织切片。这就是计算病理学的世界——在这里,科学家利用强大的计算机来阅读这些微观图像,并预测患者的疾病(如癌症)随时间演变的趋势。长期以来,这些计算机就像是才华横溢却沉默寡言的巫师:它们能观察一张切片并说,“这个患者可能会存活更久,”或者“这个人的风险更高,”其准确度令人印象深刻。但问题在于,当你询问计算机为什么做出这样的预测时,它通常只是指向图像中的一个特定位置并说:“我看了这里。”但它并没有告诉你它在那个位置看到了什么。它是看到了激烈的免疫战斗的迹象?是细胞死亡的迹象?还是某种特定类型的血管?如果不了解其中的“为什么”,医生就无法完全信任计算机,也无法从计算机发现的规律中学习关于疾病的新秘密。
这正是名为 SAGE 的新工具发挥作用的地方。把 SAGE 想象成一个翻译官,它能将计算机沉默的指点转化为讲述出来的故事。这篇论文介绍了一种方法,可以将那些“沉默的巫师”(具体来说是一种被称为“基于注意力的多实例学习”或 ABMIL 的 AI 模型)转化为能够使用包含 25 个常见生物学概念(如“淋巴细胞”、 “坏死”或“血管生成”)的词典进行解释的模型。研究人员并没有重新训练计算机或强迫它学习新知识;相反,他们构建了一个后验框架,通过读取计算机现有的注意力图,将其转化为人类可读的概念。他们在来自大规模数据库 TCGA 的七种不同类型的癌症上进行了测试。结果如何?SAGE 不仅证实了医生已知的知识——例如坏死(组织死亡)通常预示着不良预后——还揭示了令人惊讶的、具有癌症特异性的故事,例如在肾癌中,一种特定类型的血管生长过程实际上似乎是患者预后良好的迹象。
SAGE 的故事:赋予 AI 一个声音
想象你有一个超级聪明但害羞的机器人助手,它非常擅长给你的作业评分,但拒绝告诉你为什么你得了 A 或 F。它只是高亮显示页面并说:“关注这里。”这本质上就是目前病理学中的 AI 模型所做的。它们观察全切片图像(WSI)——这些图像就像是由数千个微小正方形补丁组成的巨大数字马赛克——并为不同的补丁分配“注意力权重”。如果一个补丁获得了很高的注意力权重,意味着模型认为该特定的小方块对于其预测很重要。但“重要”这个词很模糊。是因为它看起来像肿瘤?因为它看起来像一支免疫细胞军队?还是仅仅因为它颜色很奇怪?
这篇论文的作者 Abdallah Lamane 及其团队想要解开这个谜团。他们创建了一个名为 SAGE(语义注意力全局解释)的框架。把 SAGE 想象成侦探的笔记本,它能接收机器人模糊的指点,并用一套包含 25 个特定生物学概念的词汇表来填补空白。这些概念涵盖了从细胞类型(如淋巴细胞或癌细胞)到细胞行为(如有丝分裂,即细胞分裂)以及环境特征(如坏死或瘢痕)的广泛范围。
以下是魔法发生的步骤:
- 冻结的模型: 研究人员采用了一个现有的、“冻结的”AI 模型(一个已经训练好用于预测生存期的模型),并且没有改变它思考方式的任何一点。他们只是保留了它的注意力权重。
- 词典: 他们使用了一种特殊的“视觉-语言模型”(一种既能理解图像又能理解文字的 AI)来创建一个包含这 25 个概念的词典。
- 翻译: 对于组织图像中的每一个微小补丁,SAGE 都会询问视觉-语言模型:“这个补丁看起来有多少像‘坏死’?它看起来有多少像‘淋巴细胞’?”它会为每个概念获得一个分数。
- 聚合: 这是聪明之处。SAGE 不仅仅是计算这些分数的平均值。它将这些分数乘以原始 AI 模型分配给这些补丁的注意力权重。如果原始 AI 非常关注一个看起来很像“坏死”的补丁,SAGE 就会给“坏死”一个很高的分数。如果 AI 忽略了一个看起来像“淋巴细胞”的补丁,那么该概念的分数就会很低。
- 全局故事: 最后,SAGE 查看一组患者(一个队列)中的所有情况,并问道:“具有高‘坏死’分数的患者是否往往具有更差的 AI 生存预测结果?”通过这样做,它创建了一个“全局解释”,讲述的是关于整个群体的故事,而不仅仅是关于一张切片的故事。
他们的发现:证实已知,发现未知
当团队在七个不同的癌症数据集(包括乳腺癌、膀胱癌、宫颈癌、结肠癌、肾癌、脑癌和肺癌)上运行 SAGE 时,结果既包含了令人安心的确认,也包含了令人兴奋的发现。
首先,AI 的“沉默指点”是有道理的。在所有案例中,上皮细胞(组织的正常组成部分)这一概念与较好的生存预测密切相关。相反,坏死(死亡组织)则一致地与较差的预测相关联。这正是病理学家所预期的,这表明 AI 确实在学习真实的生物学,而不仅仅是随机噪声。
但真正的乐趣始于他们观察特定癌症时。在肾透明细胞癌 (KIRC) 中,AI 的准确性极高,而 SAGE 揭示了一个迷人的模式。模型将血管生成(新血管的生长)和内皮细胞(构成血管壁的细胞)与较好的生存率联系在一起。这听起来违反直觉——通常,更多的血管意味着肿瘤正在“进食”——但研究人员指出,这与已知的肾癌分子亚型相符,在这些亚型中,特定类型的血管生长实际上是疾病侵袭性较低的标志。
在宫颈癌 (CESC) 中,AI 将中性粒细胞(一种白细胞)与较差的结果联系起来,这与之前的研究一致。有趣的是,在某些情况下,AI 将高细胞密度(大量细胞聚集在一起)与较好的生存率联系起来。起初,这看起来很奇怪,因为细胞拥挤通常意味着癌症。然而,当研究人员观察实际图像时,他们看到那些“拥挤”的区域实际上是充满了正在与癌症战斗的免疫细胞,而不仅仅是肿瘤细胞。SAGE 帮助他们意识到,AI 捕捉到的是一种“好的”拥挤。
“为什么”很重要:注意力 vs. 仅仅是观察
团队提出的最重要问题之一是:“AI 的注意力真的重要吗?如果我们只是平等地观察所有内容,结果会有所不同吗?”
为了测试这一点,他们进行了一项实验,忽略了 AI 的注意力权重,而是直接对整个切片的概念得分取平均值。他们发现,虽然即使没有注意力权重,某些趋势仍然存在,但使用 AI 的注意力权重后,信号变得更加清晰。例如,在肾癌中,血管与良好生存率之间的联系在使用了 AI 的注意力权重后变得更加强烈。这表明 AI 不仅仅是在随机猜测;它是在专门学习如何专注于疾病中最具代表性的特征。
他们还测试了他们的 25 个单词词典本身是否足够好。他们建立了一个新的、简单的模型,该模型仅使用这 25 个概念的文本描述(完全不使用图像)来预测生存期。令人惊讶的是,这个“仅限文本”的模型表现得几乎与复杂的基于图像的模型一样好。这表明,他们选择的这 25 个概念捕捉到了预测患者预后所需的绝大部分生物学真相。
局限性与未来
作者们谨慎地指出,不要过度吹捧他们的结果。他们指出 SAGE 是一个“后验”工具,这意味着它解释的是一个已经做出了决定的模型。它生成的是假设,而不是绝对的因果关系证明。他们也承认,他们的 25 个概念列表可能会遗漏一些罕见或细微的特征,而且他们用于将图像转化为文字的视觉-语言模型并不完美,可能会出错。
然而,这项研究表明,SAGE 是一个强大且轻量级的工具,可以“插入”现有的 AI 模型中,而无需重新训练它们。它将黑箱变成了一个透明的窗口,让医生不仅能看到 AI 在看哪里,还能看到它看到了什么。通过揭示 AI 正在关注诸如“血管生成”或“淋巴聚集”之类的特征,SAGE 有助于建立医生与机器之间的信任,甚至可能帮助医生发现关于癌症如何运作的新生物学线索。
最终,SAGE 不仅仅告诉我们 AI 很聪明;它帮助我们理解 AI 用来描述人体语言的方式,将冰冷的数据转化为我们都能理解的故事。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。