Neuromation新研究:利用卷积神经网络进行儿童骨龄评估

近日,Neuromation 团队在 Medium 上撰文介绍其最新研究成果:利用卷积神经网络(CNN)评估儿童骨龄,这一自动骨龄评估系统可以得到与放射科专家相似或更好的结果。该团队评估了手骨不同区域,发现仅对掌骨和近端指骨进行评估,得到的结果与对整个手骨进行评估的结果相差无几。为了克服放射图像的质量和多样性问题,该团队引入了严格的清理和标准化过程,以增强模型的鲁棒性和准确率,提升骨龄评估的准确率、提高结果的可复现性以及临床医生的效率。

Alexander 的论文《Pediatric Bone Age Assessment Using Deep Convolutional Neural Networks》(Vladimir Iglovikov、Alexander Kalinin 以及 Alexey Shvets 联合完成的一项工作)已经发表在第 4 届 Deep Learning in Medical Image Analysis(DLMIA 2018)Workshop 上。这不是 Neuromation 在医学影像方面发表的第一篇论文,但这是一个深入了解该研究细节的好机会。与我们之前发表的关于医疗概念规范化的文章(https://medium.com/neuromation-io-blog/neuromation-research-medical-concept-normalization-in-social-media-posts-f06504200cec)相似,这是一件严肃且相当复杂的事。

骨龄反映了你的年龄:骨龄评估

骨龄,基本上就是指你的骨头看起来多大了。随着孩子的成长,他的骨骼也在逐渐成长。这就意味着你可以通过观察儿童的骨骼,根据具备此类骨骼的儿童的平均年龄来了解这个孩子多大了。这时候你可能会想这是不是一篇考古学方面的文章:毕竟活着的小孩应该并不怎么拍 X 光。

这么想是对的,但是也不对。正常发育的话,儿童的骨龄应该在生理年龄(日历年龄)的上下 10% 区间内波动。但也有例外,例如,孩子青春期的生长突增可能会影响骨龄。因此,如果骨龄比生理年龄大几年的话,该儿童就会提前停止生长,而如果骨骼比较「年轻」的话,则可能会出现延迟的生长突增。此外,在给定身高和骨龄的情况下,你可以准确预测这个孩子的成年身高,这也可以派上一些用场:如果你的孩子喜欢篮球,你可能想知道他未来身高能否超过 7 英尺。

还有一些例外情况:骨龄和生理年龄严重不匹配是预示着身体出现问题,如成长障碍和内分泌问题等。骨龄只能告诉医生患者在特定时间时骨骼的相对成熟度,还要和其他临床指标结合才能区分出是正常生长还是生长相对提前或迟缓。连续的骨龄读数可以说明儿童的发育方向或治疗进展。通过评估骨龄,儿科医生可以诊断儿童发育过程中的内分泌和代谢紊乱问题,如骨发育不良或受营养、代谢或其他未知因素影响的生长缺陷,这可能会损害骨骺或骨质成熟。在这种生长迟缓情况下,骨龄和身高可能会有相同程度的延迟,但是经过治疗,此类儿童仍可达到正常成年人的身高。

综上所述,儿科医生常常会用 X 光照射儿童手部来评估其骨龄……所以自动化是一个很大的问题。

读手骨:从手掌和手腕评估骨龄

骨骼成熟度主要是通过骨骺中次级骨化中心的发育和骨化程度来评估的。几十年来,评估骨骼成熟度通常是根据对手骨和手腕的骨骼发育情况进行可视化评估来进行的。以下是放射科医生在查看手部 X 光片时看到的内容:

现在评估骨龄的两种最常用的技术是 Greulich-Pyle 法和 Tanner-Whitehouse(TW2)法。这两种方法都是根据成熟度指标,用左手手掌和手腕的放射图像评估骨骼成熟度,即管状骨的骨骺从骨化的最早阶段到它们和骨干融合的过程中放射影像呈现出的变化,或者是扁骨在变成成年人骨骼形状之前在放射影像上呈现出来的变化……别担心,我们在此之前也没听过这些。我们将这一过程展示在下图中:

像 GP 或 TW2 这样的传统评估骨骼成熟度的技术耗时较久,从一定程度上说也太过主观,因此一些高级的放射科医生也不总是认可这些结果。因此,使用计算机辅助诊断系统提升骨龄评估的准确率、提高结果的可复现性以及临床医生的效率就变得很诱人了。

近期,一些基于深度学习的方法已经证明在解决生物医学领域的多个问题时性能优于传统的机器学习方法。而在医学成像领域,卷积神经网络已经成功应用于糖尿病视网膜病变筛查、乳腺癌组织学成像分析、骨骼疾病预测以及一些其他问题中。如果你想深入了解这些应用,请参阅我们之前的文章(https://medium.com/neuromation-io-blog/artificial-intelligence-technologies-could-help-solve-the-global-healthcare-crisis-441a51ac45b8)。

因此我们试着将现代深度神经网络架构应用于骨龄评估。下文将描述用全自动深度学习方法解决骨龄评估问题的完整过程,我们所用的数据来自儿科骨龄估计挑战赛(http://rsnachallenges.cloudapp.net/competitions/4),该比赛由北美放射学会(RSNA)举办。尽管我们的主要目标是要获得尽可能高的准确率,但还是要保持系统的鲁棒性,克服不同医疗中心的不同硬件所生成 X 光片的质量问题和多样性问题。

数据

数据集来自于举办了儿科骨龄挑战赛 2017 的 RSNA。他们从斯坦福儿童医院和科罗拉多儿童医院获取了放射图像,这些图像是在不同时间不同条件下用不同设备拍出来的。专业的儿科放射医师对这些图像进行注释,他们将这些图像与 Greulich 和 Pyle 所著《手部骨骼发育放射图解》进行比较,并在报告中记录骨龄。大赛组委会从这些报告中提取了骨龄信息,并将其作为模型训练的真值。

放射图像在比例、方向、曝光等方面各不相同,而且通常会带有特定的标记。完整的 RSNA 数据集包含 12,611 个训练图像、1,425 个验证图像以及 200 个测试图像。显而易见测试集特别小,且在开发阶段我们并不知道其标签,因此我们从训练集中取出 1000 张放射图像,在这些图像上对模型进行测试。

训练数据包含 5778 名女性和 6833 名男性的放射图像。他们的年龄从 1 个月到 228 个月不等,受试者多为 5 到 15 岁的儿童:

预处理 I:分割和对比

我们工作的关键贡献之一是进行了严格的预处理。为了防止模型因图像伪影学习到错误信息,我们首先通过图像分割来移除图像背景。

我们用 U-Net 深度架构(https://arxiv.org/abs/1505.04597)进行图像分割。自 2015 年 U-Net 出现以来,它就成为分割任务中的主力。它由一个获取上下文信息的收缩路径(contracting path)和一个用于精准定位的对称扩张路径(symmetric expanding path)组成;因为本文主题不是 U-Net,我们简单地展示一下其架构即可,细节请参阅论文。

我们还在训练过程中使用了批归一化使函数收敛。我们的算法使用的是广义损失函数(generalized loss function)

式中 H 是标准的二进制交叉熵损失函数

,式中 y_i 是像素 i 的真值,

是像素的预测概率,J 是 Jaccard 系数的可微泛化:

我们通过移除小的无关的连接组件和均衡对比度完成了分割步骤。下图是我们进行预处理的流程:

如你所见,放射图像的质量和对比度有了显著的提升。你可以到这里停止预处理,用我们的预处理方法和标准技术(如缩放和旋转)来增强训练集,然后训练一个标准的卷积神经网络用于分类或回归。我们试了这种方法,尽管结果不如我们的最终模型,但是也很令人满意。

之前的 GP 和 TW 方法关注的是手骨的特定部位(包括指骨、掌骨和腕骨)。我们决定试着用这些信息,针对几个解析度较高的特定区域分别训练模型,再评估和比较它们的性能。为了正确定位这些区域,我们要将所有图的大小和位置转换成一样的,也就是说将它们放在相同的坐标空间中,该过程通常被称为图像配准(image registration)。

预处理 II:基于特征点进行图像配准

我们的图像配准策略很简单:检测手骨中几个特征点的坐标,然后计算仿射变换参数(缩放、旋转、平移和反射),使图像与标准位置一致。

为了创建特征点模型的训练集,我们用 VGG 图像注释器(VIA)手动标注了 800 张放射影像。我们选择了三个特征点:第三根手指远端指骨的顶端,拇指远端指骨的顶端以及头状骨的中心。我们将特征点的像素坐标作为回归模型的训练目标。

再说一次,特征点模型是受一系列 VGG 族模型启发、通过深度卷积神经网络实现的回归模型。VGG 模块由两个带有指数线性单元激活、批归一化和最大池化的卷积层组成。下图是其架构:

该模型使用均方误差(MSE)损失和 Adam 优化器训练得到:

为了提升泛化性能,我们对输入应用了包括旋转、平移和缩放在内的标准增强。模型输出 6 个坐标,即 3 个特征点中每个特征点有 2 个坐标。

找到特征点后,我们就可以计算所有放射影像的仿射变换(缩放、旋转、平移)。我们的目标是在保持图像纵横比的情况下,将其调整到统一的位置:

1. 中指顶端水平对齐,在图像上边缘以下约 100 个像素处。

2. 头状骨水平对齐,在图像下边缘以上约 480 个像素处。

通常用左手的放射影像评估骨龄,但有时候数据集中的图像会呈镜像。为了检测这些图像并对其进行适当调整,我们使用拇指作为特征点。

来看一个图像配准模型起作用的例子。如你所见,手骨成功地旋转到我们希望的标准位置了:

此处展示了更多经过完整预处理的样本。分割、归一化以及配准结果见第四行:

骨龄评估模型

遵循 Gilsanz 和 Ratib 的《Hand Bone Age: a Digital Atlas of Skeletal Maturity》,我们从经过配准的放射影像中选择了三个特定区域,针对每一个区域训练了一个单独的模型:

1. 完整的手骨;

2. 腕骨;

3. 掌骨和近端指骨。

下图是真实放射影像中的上述区域以及相对应的分割样本:

卷积神经网络常用于分类任务,但是骨龄评估本质上是一个回归问题:我们必须要预测年龄,而年龄是一个连续的变量。因此,我们需要比较两种 CNN 架构设置——回归和分类。我们分别实现了二者,它们的参数和训练方法类似,区别只在于最后两层。

我们的第一个模型是自定义的 VGG 架构,其输出结果是回归结果。网络堆叠了 6 个卷积核大小分别为 32、64、128、256、384 的卷积模块,接着是两个有 2048 个神经元的全连接层和一个输出层(见下图)。输入的大小取决于需要预测的图像区域。为了更好地进行泛化,我们在全连接层之前应用了 dropout 层。我们还重新调整了回归目标也就是骨龄的范围,将其定在 [-1,1]。为了避免过拟合,我们还使用缩放、旋转和位移等方式增强训练数据。该网络是用 Adam 优化器通过最小化平均绝对误差(MAE)训练得到的:

第二个模型是分类模型,该模型和回归模型很像,只是最后两层不同。一个主要的区别是分配给每个骨龄的类别不同。在数据集中,骨龄是以月为单位的,所以我们可以将其视为 240 类,将倒数第二层变成有 240 个输出的 softmax 层。这一层输出的是概率的向量,这里的类别概率真值范围在 [0,1] 区间内。最后一层中,概率向量与不同骨龄的向量 [1,…,239,240] 相乘。因此,该模型会输出骨龄的期望值。我们用和回归模型同样的训练方法训练了这个模型。

下图是分类模型的架构,除了多了 softmax 层和分箱层(binning layer)外,其他和回归模型相同:

结果

我们从训练集中取出 1000 张放射图像作为验证集来评估模型。遵循 GP 和 TW 方法将性别考虑在内,对每个空间区域我们都针对性别训练了性别特定模型,并与在整个数据集上训练得到的不区分性别的模型做了比较。以下是结果总结:

结果证明在输入中添加性别可以将模型准确率平均提高 1.4 个月。最左边这一列表示针对两个性别的回归模型的表现。掌骨和近端指骨区域(区域 C)的 MAE 为 8.42 个月,整个手骨(区域 A)的 MAE 为 8.08 个月。对这三个区域进行线性集成可以将准确率提高到 7.52 个月(表中最后一行)。

性别特定的回归模型(第二列和第三列)将针对男性的模型的 MAE 改善为 6.30 个月,将针对女性的模型的 MAE 改善到 6.49。需要注意的是,就女性而言,掌骨和近端指骨区域的 MAE 为 6.79 个月,甚至比对整个手骨进行预测的结果还准确,后者的 MAE 仅为 7.12 个月。

性别特定的分类模型(第四列和第五列)的结果比回归模型的结果略好,它们的 MAE 分别为 6.16 个月和 6.39 个月(最后一行)。

最后,在第六列,我们将所有性别特定的模型(分类和回归)进行集成。在验证集上,集成模型得到的准确率为 6.10 个月,这也是当前最佳的结果,无论从骨龄评估挑战赛还是从实际应用来看,这都是一个很好的结果。

结论

本文介绍了如何开发可以准确评估骨骼成熟度的自动骨龄评估系统,该系统可以得到与放射科专家相似或更好的结果。我们评估了手骨不同区域,发现可以仅对掌骨和近端指骨进行评估,得到的结果与对整个手骨进行评估的结果相差无几。为了克服放射图像的质量和多样性问题,我们引入了严格的清理和标准化过程,以显著增强模型的鲁棒性和准确率。

我们的模型有极大的潜力部署在临床环境中,帮助临床医生实时判断骨龄。这可以保证在患病儿童出现成长障碍时能进行及时的诊断和治疗。

原文地址:https://medium.com/neuromation-io-blog/neuromation-research-pediatric-bone-age-assessment-with-convolutional-neural-networks-ad539ffb671f

原文地址:https://www.cnblogs.com/alan-blog-TsingHua/p/9766868.html

时间: 2024-11-08 03:29:40

Neuromation新研究:利用卷积神经网络进行儿童骨龄评估的相关文章

机器学习:利用卷积神经网络实现图像风格迁移 (一)

相信很多人都对之前大名鼎鼎的 Prisma 早有耳闻,Prisma 能够将一张普通的图像转换成各种艺术风格的图像,今天,我们将要介绍一下Prisma 这款软件背后的算法原理.就是发表于 2016 CVPR 一篇文章, " Image Style Transfer Using Convolutional Neural Networks" 算法的流程图主要如下: 总得来说,就是利用一个训练好的卷积神经网络 VGG-19,这个网络在ImageNet 上已经训练过了. 给定一张风格图像 a 和

Tensorflow学习教程------利用卷积神经网络对mnist数据集进行分类_训练模型

原理就不多讲了,直接上代码,有详细注释. #coding:utf-8 import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data mnist = input_data.read_data_sets('MNIST_data',one_hot=True) #每个批次的大小 batch_size = 100 n_batch = mnist.train._num_examples // batch_

利用卷积神经网络对大规模可穿戴传感器数据进行运动分类

Exercise Motion Classification from Large-scale Wearable Sensor Data Using Convolutional Neural Networks 本文使用CNN来对可穿戴传感器收集的大规模运动数据进行分类,同时对传感器数据转换为不同的图像格式输入做出了比较.最佳性能配置以92.1%的准确率将50种健身房运动分类. 作者在这里使用的是CNN而不是RNN(常用来处理时序数据),是因为每次的重复运动练习(如举哑铃)是花费的时间是非常短的少

利用卷积神经网络处理cifar图像分类

这是一个图像分类的比赛CIFAR( CIFAR-10 - Object Recognition in Images ) 首先我们需要下载数据文件,地址: http://www.cs.toronto.edu/~kriz/cifar.html CIFAR-10数据集包含10个类别的60000个32x32彩色图像,每个类别6000个图像.有50000张训练图像和10000张测试图像. 数据集分为五个训练批次和一个测试批次,每个批次具有10000张图像.测试批次包含每个类别中恰好1000张随机选择的图像

卷积神经网络(CNN)基础介绍

本文是对卷积神经网络的基础进行介绍,主要内容包含卷积神经网络概念.卷积神经网络结构.卷积神经网络求解.卷积神经网络LeNet-5结构分析.卷积神经网络注意事项. 一.卷积神经网络概念 上世纪60年代.Hubel等人通过对猫视觉皮层细胞的研究,提出了感受野这个概念.到80年代.Fukushima在感受野概念的基础之上提出了神经认知机的概念,能够看作是卷积神经网络的第一个实现网络,神经认知机将一个视觉模式分解成很多子模式(特征),然后进入分层递阶式相连的特征平面进行处理,它试图将视觉系统模型化,使其

深度卷积神经网络在目标检测中的进展

作者:travelsea链接:https://zhuanlan.zhihu.com/p/22045213来源:知乎著作权归作者所有.商业转载请联系作者获得授权,非商业转载请注明出处. 近些年来,深度卷积神经网络(DCNN)在图像分类和识别上取得了很显著的提高.回顾从2014到2016这两年多的时间,先后涌现出了R-CNN,Fast R-CNN, Faster R-CNN, ION, HyperNet, SDP-CRC, YOLO,G-CNN, SSD等越来越快速和准确的目标检测方法. 基于Reg

卷积神经网络之VGG

2014年,牛津大学计算机视觉组(Visual Geometry Group)和Google DeepMind公司的研究员一起研发出了新的深度卷积神经网络:VGGNet,并取得了ILSVRC2014比赛分类项目的第二名(第一名是GoogLeNet,也是同年提出的).论文下载 Very Deep Convolutional Networks for Large-Scale Image Recognition.论文主要针对卷积神经网络的深度对大规模图像集识别精度的影响,主要贡献是使用很小的卷积核(\

CNN卷积神经网络新想法

最近一直在看卷积神经网络,想改进改进弄出点新东西来,看了好多论文,写了一篇综述,对深度学习中卷积神经网络有了一些新认识,和大家分享下. 其实卷积神经网络并不是一项新兴的算法,早在上世纪八十年代就已经被提出来,但当时硬件运算能力有限,所以当时只用来识别支票上的手写体数字,并且应用于实际.2006年深度学习的泰斗在<科学>上发表一篇文章,论证了深度结构在特征提取问题上的潜在实力,从而掀起了深度结构研究的浪潮,卷积神经网络作为一种已经存在的.有一定应用经验的深度结构,重新回到人们视线,此时硬件的运算

卷积神经网络新想法

最近一直在看卷积神经网络,想改进改进弄出点新东西来,看了好多论文,写了一篇综述,对深度学习中卷积神经网络有了一些新认识,和大家分享下. 其实卷积神经网络并不是一项新兴的算法,早在上世纪八十年代就已经被提出来,但当时硬件运算能力有限,所以当时只用来识别支票上的手写体数字,并且应用于实际.2006年深度学习的泰斗在<科学>上发表一篇文章,论证了深度结构在特征提取问题上的潜在实力,从而掀起了深度结构研究的浪潮,卷积神经网络作为一种已经存在的.有一定应用经验的深度结构,重新回到人们视线,此时硬件的运算