← 最新论文
⚡ electrical engineering

Entropy-Guided Self-Supervised Learning for Medical Image Classification

本文提出了一种协同深度学习框架,将基于 ImageNet 预训练和基于熵引导的掩码自编码器预训练的 ConvNeXt-Tiny 模型集成到集成学习中,通过有效应对跨多样化数据集的标注数据有限和类内变异性高等挑战,在医学图像分类任务中实现了最先进的性能。

原作者: Joao Florindo, Viviane Moura

发布于 2026-05-22
📖 1 分钟阅读☕ 轻松阅读

原作者: Joao Florindo, Viviane Moura

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下,你正在尝试教导两名不同的学生如何从医学影像(如 X 光片或超声扫描)中识别特定疾病。挑战在于医学影像非常棘手:数据量不多,彼此之间看起来非常相似,而且“患病”与“健康”图像之间的差异可能极其细微。

本文提出了一种巧妙的方法来训练这些学生,使其成为专家,该方法称为熵引导自监督学习(Entropy-Guided Self-Supervised Learning)。以下是其工作原理,分解为简单概念:

两名学生(两个模型)

作者没有仅依赖一名学生,而是使用了两个不同的"ConvNeXt-Tiny"模型(可以将它们想象成两名非常聪明、专业的学生)。他们通过两种截然不同的方式训练这两名学生,赋予他们独特的优势:

  1. “世界旅行者”(ImageNet 预训练):

    • 学习方式: 这名学生首先在名为 ImageNet 的数据集上接受了海量日常照片(如猫、汽车和树木)的训练。
    • 优势: 他们擅长识别通用形状、边缘和基本模式。他们对世界的外观拥有广泛而通用的知识。
    • 类比: 将这名学生想象成一名全科医生,看过成千上万种不同类型的照片,知道“团块”或“线条”通常是什么样子的。
  2. “医学侦探”(熵引导的 MAE 预训练):

    • 学习方式: 这名学生仅在医学图像上接受训练,但没有任何标签(他们不知道哪些图像是患病或健康的)。他们使用了一种称为**掩码自编码器(Masked Autoencoder, MAE)**的技术。
    • 转折(熵引导): 通常,MAE 的工作原理是隐藏图像的部分区域,并要求计算机猜测缺失的内容。本文增加了一条特殊规则:它计算图像每个微小部分的“熵值”(一种衡量混乱程度或信息量的指标)。
      • 如果图像的某一部分枯燥且均匀(低熵),计算机会向其添加少量“噪声”(杂讯)。
      • 如果图像的某一部分复杂、细节丰富或令人困惑(高熵)——例如微妙的肿瘤或棘手的病变——计算机会向其添加更多的噪声。
    • 目标: 通过迫使这名学生重建图像中最令人困惑和最详细的区域,他们被迫格外关注最重要、信息最丰富的区域。
    • 类比: 想象一名侦探正在查看一张模糊的犯罪现场照片。侦探并非只是猜测整张图片,而是被迫专门聚焦于那些模糊、混乱的部分,因为线索就隐藏在其中。这种“熵引导”方法迫使模型成为识别医学中微妙且关键细节的专家。

团队合作(集成学习)

一旦两名学生都经过训练,他们就会在特定的医学任务(如识别乳腺癌或皮肤病变)上进行微调。

  • 策略: 系统不是只向一名学生询问答案,而是向两名学生同时询问。
  • 决策: 系统获取“世界旅行者”和“医学侦探”两者的概率分数(即每名学生有多自信),然后简单地取平均值
  • 为何有效: “世界旅行者”带来了通用的稳定性,而“医学侦探”则带来了关于医学特性的深刻、具体知识。两者结合,互补了彼此的弱点。这就像让一名全科医生和一名专科医生共同对诊断进行投票;最终的决定通常比任何单独一人做出的决定都要准确得多。

结果

作者在四个不同的医学数据集上测试了这种团队合作(乳腺超声、皮肤图像、内窥镜图像以及用于 COVID 的胸部 X 光片)。

  • 结果: 这支联合团队始终击败了现有的最佳单一模型和其他顶级方法。
  • 关键发现: “医学侦探”(熵引导模型)特别擅长从特定的医学数据中学习,而“世界旅行者”则提供了坚实的基础。当两者结合时,它们取得了破纪录的准确率,特别是在乳腺超声数据集上。

总结

本文不仅仅是在说“让我们使用人工智能”。它提出:“让我们训练一个 AI 成为通用专家,训练另一个 AI 成为专门关注医学图像中最令人困惑部分的专家。然后,让他们共同投票。”这种组合创造了一个比仅使用一种训练方法更准确、更可靠的系统,有助于医生做出更好的诊断。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →