吴恩达团队新研究！进行膝关节磁共振影像诊断已媲美医生｜论文

AI资讯1年前 (2023)发布 AI工具箱

237 0 0

日前，吴恩达在Twitter上po了自己团队的最新研究。

一种能够预测“膝关节磁共振影像检查”异常的算法。

这个算法主要用于膝关节磁共振影像影像检查中的一般异常检测与特殊诊断，特殊诊断分别是前十字韧带撕裂和半月板撕裂。

在实际的测试中，模型在异常检测、前十字韧带撕裂检测和半月板撕裂检测方面的精度分别为93.7%、96.5%和84.7%，与普通的放射科医生相比，性能指标没有显著差异。

也就是说，它已经媲美医生了。研究结果显示，在一些方面，还超过了人类医生。

研究方法

数据集

研究采用的数据集来自斯坦福大学。2011年1月1日到2012年12月31日期间，斯坦福大学医学中心收集了1370例膝关节磁共振影像检查数据。

其中，异常检测中有1104例（80.6%）；前十字韧带撕裂319例（23.3%）；半月板撕裂508例(37.1%)；前十字韧带撕裂和半月板撕裂同时发生194例(38.2%)。

在研究中，数据集被分为三个部分。分别是训练集：1130例检查数据，1088名患者；调优集：120例检查数据，111名患者；以及验证集：120例检查数据，113名患者。为了更好的构建验证集和调优集，使用了分层随机抽样，确保每个数据集中有50个“积极的”标签（异常、前十字韧带撕裂和半月板撕裂）出现。每个患者的所有检测结果都是一样的。

外部验证数据集使用的是来自Clinical Hospital Centre Rijeka的前十字韧带损伤数据集，数据集一共有917个检测案例，作者将前十字韧带损伤的程度将数据集标签分成了3个等级：非受伤(690)、部分撕裂(172)和完全撕裂(55)。

在吴恩达团队的研究中，使用了分层随机抽样，将数据集按照60:20:20的比例分成了训练、调优和验证三个数据集。

具体的过程是：先在没有对外部数据进行再训练的情况下直接应用MRNet，然后使用外部数据中的训练集和调优集对MRNet进行优化。

分类任务是区分未受伤的前十字韧带和受伤的前十字韧带（部分撕裂或完全撕裂）。

模型设计

预测模型的主要组成部分是MRNet：一个卷积神经网络（CNN）图像分析系统，能够把3D的磁共振影像映射为概率。

MRNet的输入尺寸是s×3×256×256。其中s是磁共振影像系列中的图像数量（3是彩色通道的数量）。

首先，把每个2D的磁共振影像图像切片，通过特征提取器，获得包含每个切片特征的s×256×7×7张量。

然后应用全局平均池层，将这些特征减少到s×256。接着对切片应用最大池化，以获得256维向量，将向量传递到完全连接的层，获得预测概率。