CN112614119B - 医学图像感兴趣区域可视化方法、装置、存储介质和设备 - Google Patents
医学图像感兴趣区域可视化方法、装置、存储介质和设备 Download PDFInfo
- Publication number
- CN112614119B CN112614119B CN202011587331.1A CN202011587331A CN112614119B CN 112614119 B CN112614119 B CN 112614119B CN 202011587331 A CN202011587331 A CN 202011587331A CN 112614119 B CN112614119 B CN 112614119B
- Authority
- CN
- China
- Prior art keywords
- image
- model
- interest
- region
- training
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
- 238000007794 visualization technique Methods 0.000 title abstract description 8
- 238000000034 method Methods 0.000 claims abstract description 49
- 238000012549 training Methods 0.000 claims abstract description 43
- 238000013527 convolutional neural network Methods 0.000 claims abstract description 28
- 238000012800 visualization Methods 0.000 claims abstract description 14
- 208000020016 psychiatric disease Diseases 0.000 claims abstract description 12
- 238000013145 classification model Methods 0.000 claims abstract description 10
- 208000020925 Bipolar disease Diseases 0.000 claims abstract description 4
- 230000036541 health Effects 0.000 claims abstract description 4
- 201000000980 schizophrenia Diseases 0.000 claims abstract description 4
- 230000004913 activation Effects 0.000 claims description 30
- 238000013528 artificial neural network Methods 0.000 claims description 24
- 230000006870 function Effects 0.000 claims description 15
- 238000010586 diagram Methods 0.000 claims description 14
- 238000012545 processing Methods 0.000 claims description 13
- 238000012360 testing method Methods 0.000 claims description 11
- 239000011159 matrix material Substances 0.000 claims description 9
- 238000004590 computer program Methods 0.000 claims description 8
- 230000003213 activating effect Effects 0.000 claims description 6
- 201000010099 disease Diseases 0.000 claims description 5
- 208000037265 diseases, disorders, signs and symptoms Diseases 0.000 claims description 5
- 238000000605 extraction Methods 0.000 claims description 5
- 230000000877 morphologic effect Effects 0.000 claims description 3
- 238000003672 processing method Methods 0.000 claims description 3
- 230000000007 visual effect Effects 0.000 claims description 3
- 230000008569 process Effects 0.000 abstract description 6
- 230000004927 fusion Effects 0.000 abstract description 4
- 210000004556 brain Anatomy 0.000 description 13
- 238000004422 calculation algorithm Methods 0.000 description 9
- 238000011176 pooling Methods 0.000 description 8
- 230000000875 corresponding effect Effects 0.000 description 7
- 239000013598 vector Substances 0.000 description 6
- 238000004458 analytical method Methods 0.000 description 5
- 238000013135 deep learning Methods 0.000 description 5
- 238000003745 diagnosis Methods 0.000 description 4
- 238000004364 calculation method Methods 0.000 description 3
- 238000013136 deep learning model Methods 0.000 description 3
- 238000013507 mapping Methods 0.000 description 3
- 230000003287 optical effect Effects 0.000 description 3
- 238000013473 artificial intelligence Methods 0.000 description 2
- 230000001054 cortical effect Effects 0.000 description 2
- 238000001514 detection method Methods 0.000 description 2
- 230000000694 effects Effects 0.000 description 2
- 238000002474 experimental method Methods 0.000 description 2
- 238000011478 gradient descent method Methods 0.000 description 2
- 238000010191 image analysis Methods 0.000 description 2
- 230000001788 irregular Effects 0.000 description 2
- 238000010606 normalization Methods 0.000 description 2
- 230000000644 propagated effect Effects 0.000 description 2
- 238000011160 research Methods 0.000 description 2
- 238000005070 sampling Methods 0.000 description 2
- 238000012706 support-vector machine Methods 0.000 description 2
- 206010008190 Cerebrovascular accident Diseases 0.000 description 1
- 206010012289 Dementia Diseases 0.000 description 1
- 206010028980 Neoplasm Diseases 0.000 description 1
- 208000006011 Stroke Diseases 0.000 description 1
- 210000003484 anatomy Anatomy 0.000 description 1
- 238000013459 approach Methods 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 230000002490 cerebral effect Effects 0.000 description 1
- 230000008859 change Effects 0.000 description 1
- 238000004891 communication Methods 0.000 description 1
- 230000002596 correlated effect Effects 0.000 description 1
- 238000007405 data analysis Methods 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 230000001419 dependent effect Effects 0.000 description 1
- 238000005516 engineering process Methods 0.000 description 1
- 230000002708 enhancing effect Effects 0.000 description 1
- 239000000835 fiber Substances 0.000 description 1
- 238000013467 fragmentation Methods 0.000 description 1
- 238000006062 fragmentation reaction Methods 0.000 description 1
- 230000014509 gene expression Effects 0.000 description 1
- 230000000971 hippocampal effect Effects 0.000 description 1
- 238000002372 labelling Methods 0.000 description 1
- 238000007477 logistic regression Methods 0.000 description 1
- 238000010801 machine learning Methods 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000003062 neural network model Methods 0.000 description 1
- 210000002569 neuron Anatomy 0.000 description 1
- 239000013307 optical fiber Substances 0.000 description 1
- 238000005457 optimization Methods 0.000 description 1
- 230000004044 response Effects 0.000 description 1
- 230000000698 schizophrenic effect Effects 0.000 description 1
- 239000004065 semiconductor Substances 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 238000012546 transfer Methods 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/0002—Inspection of images, e.g. flaw detection
- G06T7/0012—Biomedical image inspection
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/20—Image preprocessing
- G06V10/25—Determination of region of interest [ROI] or a volume of interest [VOI]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20092—Interactive image processing based on input by user
- G06T2207/20104—Interactive definition of region of interest [ROI]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/30—Subject of image; Context of image processing
- G06T2207/30004—Biomedical image processing
- G06T2207/30016—Brain
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- General Physics & Mathematics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- Evolutionary Computation (AREA)
- General Engineering & Computer Science (AREA)
- General Health & Medical Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Computing Systems (AREA)
- Bioinformatics & Computational Biology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Molecular Biology (AREA)
- Evolutionary Biology (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Multimedia (AREA)
- Medical Informatics (AREA)
- Nuclear Medicine, Radiotherapy & Molecular Imaging (AREA)
- Radiology & Medical Imaging (AREA)
- Quality & Reliability (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种医学图像感兴趣区域可视化方法、装置、存储介质和设备,使用卷积神经网络作为分类模型,模型的训练过程中,采用多尺度输入数据进行训练:1)原始图像,2)由原始图像上随机取得的图像块;随机提取的图像块和原始图像交替作为模型的输入,模型可以综合学习来自图像不同尺度的输入信息,模型也可以学习到图像的整体信息。训练标签采用精神疾病标签类别,即模型输出为健康、精神分裂症、双相情感障碍等。本方案利用特定的训练策略,实现模型对图像局部信息和整体信息的融合,实现图像的快速、精确分类,同时可以实现相比现有方法更加精细的对于指定类别的感兴趣区域可视化结果。
Description
技术领域
本发明涉及图像数据分析技术,尤其涉及一种医学图像感兴趣区域可视化方法、装置、存储介质和设备。
背景技术
目前基于深度学习,尤其是卷积神经网络(Convolution neural network,CNN)的图像分类方法是图像分析领域的主流研究方法之一,且已在多种应用场景中取得很好效果,但是在细粒度图像分类任务中,CNN等深度学习算法的性能仍有待提升。以基于大脑结构磁共振图像的精神疾病分类任务为例,由于大脑的解剖结构较为复杂,精神疾病对大脑结构的影响相对较为细微,同时与肿瘤、脑卒中等具有确切病灶的疾病不同,精神疾病对大脑的影响往往分布于大脑中的多个位置,因此对于精神疾病的诊断任务而言,全脑结构磁共振图像具有大量的冗余信息,因此直接将全脑结构磁共振图像作为深度学习模型的输入,很难实现精神疾病的精确诊断。
针对此问题,目前最新的研究提出利用先验信息,定位出特定图像区域,该特定图像区域根据先验被认为是与分类标签高度相关的图像区域,如痴呆症诊断任务中的大脑磁共振图像中的海马体区域、汽车品牌分类任务中的车标区域等。将定位区域作为图像块提取出来,作为深度学习模型的输入,进行分类。这种策略可以很大程度上减少原始图像中的冗余信息,从而提升分类精度。但上述策略存在四个问题:1)需要预先人工定义图像感兴趣区域,对于一些难分类的图像(如精神分裂症患者的大脑结构磁共振图像)很难预先精确定义哪些特定的图像区域发生病变;2)对于不同图像,很难确定包含感兴趣区域的图像块的大小,因此感兴趣区域可能超出预定义图像块的范围,或包含在多个图像块中,因此造成分类相关特征的缺失或割裂;3)定义好感兴趣区域后,在每个图像数据上提取感兴趣区域,需要目标检测算法的协助,因此目标检测算法的精度将极大地决定后续分类模型的性能;4)预定义的感兴趣区域彼此之间可能存在关联,因此直接提取出感兴趣区域进行分类,将导致这种潜在关联信息的丢失,从而没有实现数据的有效利用。
深度学习分类模型的一个弊端在于其难解释性,即很难给出模型进行推理判断具体依据哪些特征。为解决此问题,目前最新的研究采用类别激活图(Class activationmap,CAM)和基于梯度的类别激活图(Gradient-based CAM,Grad-CAM)的策略,即利用网络模型全连接层的神经元权重、或各通道特征图的平均值为网络每个通道的梯度激活情况进行加权,得到模型对于每幅输入图像的特定类别标签的响应情况,以此来可视化深度学习模型对图像的哪些区域做出较大响应。目前,CAM和Grad-CAM广泛应用于图像处理的各个领域,但此类方法需要利用深度神经网络模型中下采样得到的特征图上的梯度激活情况,因此可视化过程中需要大比例的上采样操作,因此得到的类别激活图较为模糊,如果应用场景需要精细的解释(如医学影像分析中需要接近体素级别的可视化),CAM类方法无法得到较为精细的可视化结果。
综上,现有技术的模型对图像局部信息和整体信息的融合差,对图像的分类速度和精确低;对于制定类别的感兴趣区域可视化结果精细度不高。亟需一种对图像分类及感兴趣区域可视化的高速、高精度方法。
发明内容
为了克服现有技术的不足,本发明的目的在于提供一种医学图像感兴趣区域可视化方法、装置、存储介质和设备,其能解决上述问题。
本发明的目的采用以下技术方案实现。
一种医学图像自动分类及感兴趣区域可视化方法,方法包括:
步骤1,建立分类模型;
步骤2,从训练数据中的一个原始图像上任意位置随机提取一个任意尺寸的图像块作为模型的输入进行训练;
步骤3,将步骤2中的原始图像整幅作为模型的输入进行训练,获匹配该图像块的类别标签并确定相应的权值;
步骤4,改变步骤2中原始图像的选取位置,提取任意尺寸的图像块作为模型的输入进行训练;
步骤5,再将步骤2中的原始图像整幅作为模型的输入进行训练;
步骤6,重复步骤4和步骤5,选取的图像块的位置遍历整个原始图像,通过迭代训练学习图像全局及不同位置的局部纹理信息,获得训练好的生成器;
步骤7,验证测试,加载所述训练好的生成器,从原始图像中随机取T个图像块,使采样得到的图像块集合可以有重叠地覆盖整幅图像,通过融合每个图像块中的激活图,得到原始图像X针对类别标签c的感兴趣区域激活图Lc(X);
步骤8,通过原始图像X的感兴趣区域激活图Lc(X)实现神经网络对于输入测试图像各个类别的激活情况,标记出神经网络的感兴趣区域,实现神经网络的可视化。
优选的,在步骤1中,采用卷积神经网络作为分类模型。
优选的,图像块尺寸采用正方体尺寸块。
优选的,训练标签采用精神疾病标签类别,模型输出为健康、精神分裂症、和双相情感障碍。
优选的,疾病标签类别c,神经网络对于相应图像块的感兴趣区域激活图lc(x)为:
式中:fk(x)为输入图像块为x时,卷积神经网络最后一层卷积层输出的k个通道的特征图;为最后一层卷积层k个特征通道上,对于标签类别c的梯度值的平均值,作为各特征图的权重;ReLU(·)为线性整流单元激活函数;Upsamp(·)为上采样操作,即将按通道加权求和后的特征图尺寸上采样至输入图像块的尺寸。
优选的,原始图像X的感兴趣区域激活图Lc(X)为:
式中:lc(xt)为根据式1计算得到的第t个图像块的感兴趣区域激活图;为将第t个图像块的感兴趣区域激活图lc(xt)按照xt在原始图像X上的空间位置复制至一与原始图像X具有相同尺寸的空白图像/>的操作;/>为卷积神经网络模型将图像块xt预测为类别c的概率,在神经网络中即为阈值化操作前的输出层输出结果,将其作为第t个图像块的感兴趣区域激活图lc(xt)的权重;°为哈达玛积、即矩阵图像对应位置元素相乘;/>为重叠情况记录矩阵,其大小与原始图像X相同,其每个元素为前述图像/>对应位置上粘贴操作/>的重叠次数的倒数。
本发明还提供了一种识别医学图像的装置,所述装置包括:
图像获取模块,用于获取医学图像;
图像处理模块,通过图像处理方法对医学图像进行初步处理为便于学习的二值化图像;
区域提取模块,对处理后的二值化图像进行形态学随机块选取处理以得到感兴趣区域;
可视化分类标识,通过前述方法训练的卷积神经网络模型对感兴趣区域进行识别。
本发明还提供了一种计算机存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现前述的方法。
本发明还提供了一种电子设备,设备包括:处理器和存储器;所述存储器上存储有计算机可读指令,所述计算机可读指令被所述处理器执行时实施前述的方法。
相比现有技术,本发明的有益效果在于:1.利用特定的训练策略,实现模型对图像局部信息和整体信息的融合,实现图像的快速、精确分类;
2.提供一种新的深度学习可视化方法,可以实现相比现有方法更加精细的对于指定类别的感兴趣区域可视化结果。
附图说明
图1为本发明图像分类及可视化系统的训练及测试示意图。
具体实施方式
为使本发明实施例的目的、技术方案和优点更加清楚,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本发明的一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动的前提下所获得的所有其他实施例,都属于本发明保护的范围。
此外,所描述的特征、结构或特性可以以任何合适的方式结合在一个或更多实施例中。在下面的描述中,提供许多具体细节从而给出对本发明的实施例的充分理解。然而,本领域技术人员将意识到,可以实践本发明的技术方案而没有特定细节中的一个或更多,或者可以采用其它的方法、组元、装置、步骤等。在其它情况下,不详细示出或描述公知方法、装置、实现或者操作以避免模糊本发明的各方面。
若附图中存在方框图,所示的方框图仅仅是功能实体,不一定必须与物理上独立的实体相对应。即,可以采用软件形式来实现这些功能实体,或在一个或多个硬件模块或集成电路中实现这些功能实体,或在不同网络和/或处理器装置和/或微控制器装置中实现这些功能实体。
附图中所示的流程图仅是示例性说明,不是必须包括所有的内容和操作/步骤,也不是必须按所描述的顺序执行。例如,有的操作/步骤还可以分解,而有的操作/步骤可以合并或部分合并,因此实际执行的顺序有可能根据实际情况改变。
在详细阐述本发明的实施例的技术方案之前,以下介绍一些相关的技术方案、术语和原理。
卷积神经网络(Convolutional NeuralNetwork,CNN)
CNN是一种多层监督学习神经网络,用来处理图像相关的机器学习问题。
典型的CNN由卷积层(Convolution)、池化层(Pooling)和全连接层(FullyConnection)组成。其中,低隐层一般由卷积层和池化层交替组成,卷积层的作用是通过卷积运算使图像的原信号特征增强并降低噪音,池化层的作用在于根据图像局部相关性的原理减少计算量同时保持图像旋转不变性。全连接层位于CNN的高层,其输入是由卷积层和池化层进行特征提取得到的特征图像,输出可连接分类器,通过采用逻辑回归、Softmax回归、或者是支持向量机(Support VectorMachine,SVM)对输入图像进行分类。
CNN的训练过程一般采用梯度下降法最小化损失函数,通过全连接层后连接的损失层,对网络中各层的权重参数逐层反向调节,并通过频繁的迭代训练提高网络的精度。CNN的训练样本集通常由形如“输入向量,理想输出向量”的向量对构成,在开始训练前,网络所有层的权重参数可以用一些不同的小随机数进行初始化。由于CNN本质上可视为一种输入到输出的映射,能够学习大量的输入与输出之间的映射关系,而不需要任何输入和输出之间的精确数学表达式,因此可以用已知向量对组成的训练样本集对CNN加以训练,使其具有输入输出对之间的映射能力。
Softmax层
Softmax函数将多个标量映射为一个概率分布,其输出的每一个值范围在(0,1)。softmax函数经常用在神经网络的最后一层,作为输出层,进行多分类。
残差神经网络(ResidualNeuralNetwork,ResNet)
CNN的典型网络结构包括ResNet、AlexNet、VGGNet、GoogleNet、SENet等。
与其他网络结构相比,ResNet最大的不同在于,其可以设置旁路支线将输入直接连到网络后面的层中,使网络后面的层也可以直接学习残差。这种方式能够解决传统CNN在信息传递时,或多或少会丢失原始信息的问题,从而保护数据的完整性。
ImageNet数据集
ImageNet数据集是一个用于视觉对象识别软件研究的大型可视化数据库。超过1400万的图像URL被ImageNet手动注释,以指示图片中的对象;在至少一百万个图像中,还提供了边界框。
ASAP(Automated Slide Analysis Platform,自动切片分析平台)
ASAP是一个集成了浏览、标记等功能的组织病理学WSI(Whole Slide Image,全切片图像)分析的开源平台。ASAP基于多个成熟的开源软件包,如OpenSlide、Qt和OpenCV等搭建。
TensorFlow
TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理。Tensor(张量)意味着N维数组,Flow(流)意味着基于数据流图的计算,TensorFlow为张量从流图的一端流动到另一端计算过程。TensorFlow是将复杂的数据结构传输至人工智能神经网中进行分析和处理过程的系统。
TFRecord数据格式
TFRecord是一种数据格式,可以允许将任意的数据转换为TensorFlow所支持的格式,其可以使TensorFlow的数据集更容易与网络应用架构相匹配。
PNPoly算法
本算法是由W.Randolph Franklin提出的。算法的思路如下:从待测点出发作一条射线,然后判断这条射线与不规则区域的交点数量。如果点的两边交点的个数都是奇数个则该点在多边形内,否则在多边形外。这个算法对于任意不规则图形都适用。
以下对本发明实施例的技术方案的原理和实现细节进行详细阐述。
实施例一
一种医学图像自动分类及感兴趣区域可视化方法,以基于大脑结构磁共振图像的精神疾病诊断任务为例,方法具体如下。
使用卷积神经网络(CNN)作为分类模型,模型的训练过程中,采用多尺度输入数据进行训练,即训练数据包括两部分:1)原始图像,2)由原始图像上随机取得的图像块。
对于每个训练数据,首先从图像上任意位置随机提取一个图像块(如48×48×48)作为模型的输入进行训练,再将整幅图像作为模型的输入进行训练。
由于每个训练数据要参与模型训练的多次迭代(实验中设定为2000次,可根据实际情况进行调整),图像块的采样位置可以遍历整幅图像,因此模型可以学习到图像不同位置的局部纹理信息;而由于未裁剪过的原始图像也作为输入训练数据,因此模型也可以学习到图像的整体信息。根据上述描述,模型训练过程中,随机提取的图像块和原始图像交替作为模型的输入,通过上述训练策略,模型可以综合学习来自图像不同尺度的输入信息,得到针对精神疾病大脑变化模式的更加可靠的判断。训练标签采用精神疾病标签类别,即模型输出为健康、精神分裂症、双相情感障碍等。
当模型训练结束后,对于任意测试图像,将其输入模型中,即可得到模型对于疾病标签的预测结果。
对于训练完成的模型,既可以像上述一样输入整幅图像作为测试图像,也可以输入图像块(如48×48×48),得到该图像块区域上的疾病预测结果。对于疾病标签类别c,可以通过下式计算神经网络对于该图像块的感兴趣区域激活图lc(x):
上式中,fk(x)为输入CNN图像块为x时,CNN最后一层卷积层输出的k个通道的特征图;为最后一层卷积层k个特征通道上,对于标签类别c的梯度值的平均值,作为各特征图的权重;ReLU(·)为线性整流单元激活函数;Upsamp(·)为上采样操作,即将按通道加权求和后的特征图尺寸上采样至输入图像块的尺寸。需要注意,与普通CAM方法需要将特征图上采样至原始输入图像尺寸不同,本方法只需将特征图上采样至输入图像块尺寸即可,极大地降低了由于上采样带来的感兴趣区域激活图模糊的问题。
根据上述计算,测试阶段输入模型的每一个图像块都可得到对应的激活情况。从原图像中随机取T个图像块(实验中T=2000),使采样得到的图像块集合可以有重叠地覆盖整幅图像。通过融合每个图像块中的激活图,即可得到原始图像X针对类别标签c的感兴趣区域激活图Lc(X):
上式中,lc(xt)为根据前述计算得到的第t个图像块的感兴趣区域激活图;为将第t个图像块的感兴趣区域激活图lc(xt)按照xt在原始图像X上的空间位置复制至一与原始图像X具有相同尺寸的空白图像/>的操作;/>为CNN模型将图像块xt预测为类别c的概率(在神经网络中即为阈值化操作前的输出层输出结果),将其作为第t个图像块的感兴趣区域激活图lc(xt)的权重;°为哈达玛积(即矩阵(图像)对应位置元素相乘;/>为重叠情况记录矩阵,其大小与原始图像X相同,其每个元素为前述图像/>对应位置上粘贴操作/>的重叠次数的倒数。
根据上述计算,得到的原始图像X的感兴趣区域激活图可以实现神经网络对于输入测试图像各个类别的激活情况,标记出了神经网络的感兴趣区域,提供了一种神经网络的可视化方法,可以为后续分析,以及感兴趣区域的定位、再分类提供依据,同时相较现有的CAM类方法保留了更多的细节。
具体地,模型的训练过程如图1所示。分类网络为一深度卷积神经网络,其网络结构主要包括5层卷积核尺寸为3×3×3、步长为1的卷积层,每层卷积层后包括一层批归一化层、一层ReLU激活函数层、以及一层池化核尺寸为2×2×2、步长为2的最大池化层,每经过一组卷积层、批归一化层、激活函数层和池化层的处理,特征图尺寸减半、通道数加倍,最终输出128维向量,经一层全连接层处理,输出n维向量(n为待分类样本集合的类别数)。
模型的损失函数采用神经网络算法中常用的交叉熵损失函数,优化方式采用自适应矩估计算法,学习率设为1e-4,模型参数通过梯度下降法更新。
效果测试:
训练得到的图像分类器及可视化系统,在搭载CentOs 6.5系统的Intel(R)Xeon(R)CPU E5-2670v22.50GHz处理器上,利用Pytorch搭建模型并测试,可在0.1秒内实现单幅图像的标签判断,对采集自不同中心的3200例大脑磁共振结构像数据上测试,得到5类精神疾病的分类准确率为85%;同时对可视化得到的感兴趣区域进行皮层特征提取,提取得到的特征使用线性SVM进行分类测试,得到的结果相比于无感兴趣区域先验、直接全脑皮层提取特征作为线性SVM输入的分类精度提升了10%。
特殊说明:
1、网络结构方面,分类网络的网络结构不局限于前述描述的结构,可以包括但不限于ResNet、DenseNet、LSTM、GRU等;
2、分类模型训练阶段,输入模型的图像块固定尺寸不局限于前述提到的48×48×48尺寸,还可为其他尺寸;
3、分类模型训练阶段,输入模型的图像块不局限于固定尺寸,也可使用可变尺寸图像块作为输入,即以多种尺寸的图像块对模型进行训练;
4、损失函数部分,不局限于前述的交叉熵损失函数,也可为Focal loss、BCEloss等分类损失函数。
5、本发明以磁共振结构图像为例进行讲解,但其适用对象包括但不限于其余模态的医学影像、遥感图像、显微镜切片图像、自然图像等
6、本发明构建一种图像分类及感兴趣区域可视化方法。该方法首次利用特定的训练策略,实现模型对图像局部信息和整体信息的融合,实现图像的快速、精确分类
7、本发明提供一种新的深度学习可视化方法,可以实现相比现有方法更加精细的对于指定类别的感兴趣区域可视化结果
实施例二
本发明还提供了一种识别医学图像的装置,所述装置包括:
图像获取模块,用于获取医学图像;
图像处理模块,通过图像处理方法对医学图像进行初步处理为便于学习的二值化图像;
区域提取模块,对处理后的二值化图像进行形态学随机块选取处理以得到感兴趣区域;
可视化分类标识,通过前述方法训练的卷积神经网络模型对感兴趣区域进行识别。
实施例三
本发明还提供了一种计算机存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现前述的方法。
本发明所示的计算机可读介质可以是计算机可读信号介质或者计算机可读存储介质或者是上述两者的任意组合。计算机可读存储介质例如可以是——但不限于——电、磁、光、电磁、红外线、或半导体的系统、装置或器件,或者任意以上的组合。计算机可读存储介质的更具体的例子可以包括但不限于:具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、随机访问存储器(RAM)、只读存储器(ROM)、可擦式可编程只读存储器(EPROM或闪存)、光纤、便携式紧凑磁盘只读存储器(CD-ROM)、光存储器件、磁存储器件、或者上述的任意合适的组合。在本发明中,计算机可读存储介质可以是任何包含或存储程序的有形介质,该程序可以被指令执行系统、装置或者器件使用或者与其结合使用。而在本发明中,计算机可读的信号介质可以包括在基带中或者作为载波一部分传播的数据信号,其中承载了计算机可读的程序代码。这种传播的数据信号可以采用多种形式,包括但不限于电磁信号、光信号或上述的任意合适的组合。计算机可读的信号介质还可以是计算机可读存储介质以外的任何计算机可读介质,该计算机可读介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。计算机可读介质上包含的程序代码可以用任何适当的介质传输,包括但不限于:无线、电线、光缆、RF等等,或者上述的任意合适的组合。
计算机可读介质可以是上述实施例中描述的电子设备中所包含的;也可以是单独存在,而未装配入该电子设备中。上述计算机可读介质承载有一个或者多个程序,当上述一个或者多个程序被一个该电子设备执行时,使得该电子设备实现如下述实施例中所述的方法。例如,所述的电子设备可以实现前述方法的各个步骤。
实施例四
本发明还提供了一种电子设备,设备包括:处理器和存储器;所述存储器上存储有计算机可读指令,所述计算机可读指令被所述处理器执行时实施前述的方法。
方法可以被实现为计算机软件程序。例如,本发明的实施例包括一种计算机程序产品,其包括承载在计算机可读介质上的计算机程序,该计算机程序包含用于执行流程图所示的方法的程序代码。在这样的实施例中,该计算机程序可以通过通信部分从网络上被下载和安装,和/或从可拆卸介质被安装。在该计算机程序被中央处理单元(CPU)执行时,执行本申请方法、装置和设备中限定的各种功能。
综上所述,
最后应说明的是:以上实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的精神和范围。
Claims (9)
1.一种医学图像感兴趣区域可视化方法,其特征在于,方法包括:
步骤1,建立分类模型;
步骤2,从训练数据中的一个原始图像上任意位置随机提取一个任意尺寸的图像块作为模型的输入进行训练;
步骤3,将步骤2中的原始图像整幅作为模型的输入进行训练,获匹配该图像块的类别标签并确定相应的权值;
步骤4,改变步骤2中原始图像的选取位置,提取任意尺寸的图像块作为模型的输入进行训练;
步骤5,再将步骤2中的原始图像整幅作为模型的输入进行训练;
步骤6,重复步骤4和步骤5,选取的图像块的位置遍历整个原始图像,通过迭代训练学习图像全局及不同位置的局部纹理信息,获得训练好的生成器;
步骤7,验证测试,加载所述训练好的生成器,从原始图像中随机取T个图像块,使采样得到的图像块集合可以有重叠地覆盖整幅图像,通过融合每个图像块中的激活图,得到原始图像X针对类别标签c的感兴趣区域激活图Lc(X);
步骤8,通过原始图像X的感兴趣区域激活图Lc(X)实现神经网络对于输入测试图像各个类别的激活情况,标记出神经网络的感兴趣区域,实现神经网络的可视化。
2.根据权利要求1所述的方法,其特征在于:在步骤1中,采用卷积神经网络作为分类模型。
3.根据权利要求1所述的方法,其特征在于:图像块尺寸采用正方体尺寸块。
4.根据权利要求1所述的方法,其特征在于:训练标签采用精神疾病标签类别,模型输出为健康、精神分裂症、和双相情感障碍。
5.根据权利要求1或4所述的方法,其特征在于:疾病标签类别c,神经网络对于相应图像块的感兴趣区域激活图lc(x)为:
式中:
fk(x)为输入图像块为x时,卷积神经网络最后一层卷积层输出的k个通道的特征图;
为最后一层卷积层k个特征通道上,对于标签类别c的梯度值的平均值,作为各特征图的权重;
ReLU(·)为线性整流单元激活函数;
Upsamp(·)为上采样操作,即将按通道加权求和后的特征图尺寸上采样至输入图像块的尺寸。
6.根据权利要求5所述的方法,其特征在于:原始图像X的感兴趣区域激活图Lc(X)为:
式中:
lc(xt)为根据式1计算得到的第t个图像块的感兴趣区域激活图;
为将第t个图像块的感兴趣区域激活图lc(xt)按照xt在原始图像X上的空间位置复制至一与原始图像X具有相同尺寸的空白图像Q的操作;
为卷积神经网络模型将图像块xt预测为类别c的概率,在神经网络中即为阈值化操作前的输出层输出结果,将其作为第t个图像块的感兴趣区域激活图lc(xt)的权重;
°为哈达玛积、即矩阵图像对应位置元素相乘;
为重叠情况记录矩阵,其大小与原始图像X相同,其每个元素为前述图像Q对应位置上粘贴操作/>的重叠次数的倒数。
7.一种识别医学图像的装置,其特征在于,所述装置包括:
图像获取模块,用于获取医学图像;
图像处理模块,通过图像处理方法对医学图像进行初步处理为便于学习的二值化图像;
区域提取模块,对处理后的二值化图像进行形态学随机块选取处理以得到感兴趣区域;
可视化分类标识,通过权利要求1-6任一项所述的方法训练的卷积神经网络模型对感兴趣区域进行识别。
8.一种计算机存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现根据权利要求1-6任一项所述的方法。
9.一种电子设备,其特征在于,包括:处理器和存储器;所述存储器上存储有计算机可读指令,所述计算机可读指令被所述处理器执行时实施权利要求1-6中任一项所述的方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011587331.1A CN112614119B (zh) | 2020-12-28 | 2020-12-28 | 医学图像感兴趣区域可视化方法、装置、存储介质和设备 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011587331.1A CN112614119B (zh) | 2020-12-28 | 2020-12-28 | 医学图像感兴趣区域可视化方法、装置、存储介质和设备 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112614119A CN112614119A (zh) | 2021-04-06 |
CN112614119B true CN112614119B (zh) | 2024-04-12 |
Family
ID=75248829
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202011587331.1A Active CN112614119B (zh) | 2020-12-28 | 2020-12-28 | 医学图像感兴趣区域可视化方法、装置、存储介质和设备 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112614119B (zh) |
Families Citing this family (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113239993B (zh) * | 2021-04-29 | 2022-12-16 | 中国人民解放军海军军医大学第三附属医院 | 一种病理图像的分类系统、终端及计算机可读存储介质 |
CN113269752A (zh) * | 2021-05-27 | 2021-08-17 | 中山大学孙逸仙纪念医院 | 一种图像检测方法、装置终端设备及存储介质 |
CN113327221A (zh) * | 2021-06-30 | 2021-08-31 | 北京工业大学 | 融合roi区域的图像合成方法、装置、电子设备及介质 |
CN113592807B (zh) * | 2021-07-28 | 2024-04-09 | 北京世纪好未来教育科技有限公司 | 一种训练方法、图像质量确定方法及装置、电子设备 |
CN113781465A (zh) * | 2021-09-18 | 2021-12-10 | 长春理工大学 | 基于Grad-CAM的医学图像分割模型可视化方法 |
CN114419309A (zh) * | 2022-01-07 | 2022-04-29 | 福州大学 | 一种基于大脑T1-w磁共振影像的高维特征自动提取方法 |
CN114782676B (zh) * | 2022-04-02 | 2023-01-06 | 北京广播电视台 | 一种视频感兴趣区域的提取方法和系统 |
CN114964476B (zh) * | 2022-05-27 | 2023-08-22 | 中国石油大学(北京) | 油气管道系统动设备的故障诊断方法、装置及设备 |
CN116245832B (zh) * | 2023-01-30 | 2023-11-14 | 浙江医准智能科技有限公司 | 一种图像处理方法、装置、设备及存储介质 |
Citations (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107977969A (zh) * | 2017-12-11 | 2018-05-01 | 北京数字精准医疗科技有限公司 | 一种内窥镜荧光图像的分割方法、装置及存储介质 |
CN109389587A (zh) * | 2018-09-26 | 2019-02-26 | 上海联影智能医疗科技有限公司 | 一种医学图像分析系统、装置及存储介质 |
CN109635835A (zh) * | 2018-11-08 | 2019-04-16 | 深圳蓝韵医学影像有限公司 | 一种基于深度学习与迁移学习的乳腺病变区域检测方法 |
CN109741346A (zh) * | 2018-12-30 | 2019-05-10 | 上海联影智能医疗科技有限公司 | 感兴趣区域提取方法、装置、设备及存储介质 |
CN110245657A (zh) * | 2019-05-17 | 2019-09-17 | 清华大学 | 病理图像相似性检测方法及检测装置 |
CN110909756A (zh) * | 2018-09-18 | 2020-03-24 | 苏宁 | 用于医学图像识别的卷积神经网络模型训练方法和装置 |
WO2020162834A1 (en) * | 2019-02-08 | 2020-08-13 | Singapore Health Services Pte Ltd | Method and system for classification and visualisation of 3d images |
WO2020215557A1 (zh) * | 2019-04-24 | 2020-10-29 | 平安科技(深圳)有限公司 | 医学影像解释方法、装置、计算机设备及存储介质 |
WO2020243556A1 (en) * | 2019-05-29 | 2020-12-03 | Leica Biosystems Imaging, Inc. | Neural network based identification of areas of interest in digital pathology images |
-
2020
- 2020-12-28 CN CN202011587331.1A patent/CN112614119B/zh active Active
Patent Citations (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107977969A (zh) * | 2017-12-11 | 2018-05-01 | 北京数字精准医疗科技有限公司 | 一种内窥镜荧光图像的分割方法、装置及存储介质 |
CN110909756A (zh) * | 2018-09-18 | 2020-03-24 | 苏宁 | 用于医学图像识别的卷积神经网络模型训练方法和装置 |
CN109389587A (zh) * | 2018-09-26 | 2019-02-26 | 上海联影智能医疗科技有限公司 | 一种医学图像分析系统、装置及存储介质 |
CN109635835A (zh) * | 2018-11-08 | 2019-04-16 | 深圳蓝韵医学影像有限公司 | 一种基于深度学习与迁移学习的乳腺病变区域检测方法 |
CN109741346A (zh) * | 2018-12-30 | 2019-05-10 | 上海联影智能医疗科技有限公司 | 感兴趣区域提取方法、装置、设备及存储介质 |
WO2020162834A1 (en) * | 2019-02-08 | 2020-08-13 | Singapore Health Services Pte Ltd | Method and system for classification and visualisation of 3d images |
WO2020215557A1 (zh) * | 2019-04-24 | 2020-10-29 | 平安科技(深圳)有限公司 | 医学影像解释方法、装置、计算机设备及存储介质 |
CN110245657A (zh) * | 2019-05-17 | 2019-09-17 | 清华大学 | 病理图像相似性检测方法及检测装置 |
WO2020243556A1 (en) * | 2019-05-29 | 2020-12-03 | Leica Biosystems Imaging, Inc. | Neural network based identification of areas of interest in digital pathology images |
Non-Patent Citations (2)
Title |
---|
李维嘉 ; 陈爽 ; 张雷 ; 吴正灏 ; .基于深度学习图像处理的肺部造影检测研究.自动化与仪器仪表.2019,(12),108-110+115. * |
管姝 ; 张骞予 ; 谢红薇 ; 强彦 ; 程臻 ; .CT影像识别的卷积神经网络模型.计算机辅助设计与图形学学报.2018,(08),150-155. * |
Also Published As
Publication number | Publication date |
---|---|
CN112614119A (zh) | 2021-04-06 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112614119B (zh) | 医学图像感兴趣区域可视化方法、装置、存储介质和设备 | |
Shin et al. | Deep convolutional neural networks for computer-aided detection: CNN architectures, dataset characteristics and transfer learning | |
CN111476292B (zh) | 医学图像分类处理人工智能的小样本元学习训练方法 | |
Meraj et al. | Lung nodules detection using semantic segmentation and classification with optimal features | |
Brosch et al. | Deep 3D convolutional encoder networks with shortcuts for multiscale feature integration applied to multiple sclerosis lesion segmentation | |
Tan et al. | A novel computer‐aided lung nodule detection system for CT images | |
Li et al. | Deep convolutional neural networks for detecting secondary structures in protein density maps from cryo-electron microscopy | |
Panigrahi et al. | Capsule network based analysis of histopathological images of oral squamous cell carcinoma | |
CN113728335A (zh) | 用于3d图像的分类和可视化的方法和系统 | |
Greenblatt et al. | Quaternion neural networks applied to prostate cancer gleason grading | |
Amit et al. | Hybrid mass detection in breast MRI combining unsupervised saliency analysis and deep learning | |
CN114600155A (zh) | 用于细胞检测和分割的弱监督多任务学习 | |
Ahmed et al. | Symmetric image contents analysis and retrieval using decimation, pattern analysis, orientation, and features fusion | |
Han et al. | Nuclei counting in microscopy images with three dimensional generative adversarial networks | |
Chen et al. | Improved fast r-cnn with fusion of optical and 3d data for robust palm tree detection in high resolution uav images | |
Dana | Computational texture and patterns: From textons to deep learning | |
Shin et al. | Three aspects on using convolutional neural networks for computer-aided detection in medical imaging | |
Ibrahim et al. | Deep learning based Brain Tumour Classification based on Recursive Sigmoid Neural Network based on Multi-Scale Neural Segmentation | |
Kalyani et al. | Deep learning-based detection and classification of adenocarcinoma cell nuclei | |
Selvia et al. | Skin lesion detection using feature extraction approach | |
Park et al. | Classification of cervical cancer using deep learning and machine learning approach | |
Sri et al. | Detection Of MRI Brain Tumor Using Customized Deep Learning Method Via Web App | |
CN112991280B (zh) | 视觉检测方法、系统及电子设备 | |
Raj et al. | Shape Feature Extraction Techniques for Computer Vision Applications | |
Priyanka Pramila et al. | Automated skin lesion detection and classification using fused deep convolutional neural network on dermoscopic images |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |