[发明专利]基于医学图像进行疾病预测的装置、方法和可读存储介质

权利要求书

1.一种基于医学图像进行疾病预测的装置，其特征在于，包括：

处理器；以及

存储器，其存储有基于医学图像进行疾病预测的程序指令，当所述程序指令由处理器运行时，使得所述装置执行以下操作：

使用分类模型对待测医学图像进行疾病分类，以获得基于所述待测医学图像的初始分类结果，其中所述分类模型包括主干网络和与所述主干网络连接的第一分类器和第二分类器，所述第一分类器用于输出待分类病种的单病种得分，所述第二分类器用于输出与待分类病种的混淆关系相关的病种细分得分；以及

使用自然语言模型对所述初始分类结果进行推理，以获得基于所述待测医学图像的疾病预测结果；

其中所述装置在使用自然语言模型进行推理中执行以下操作：

将单病种得分对应的待分类病种进行编码，以生成用于表示所述待分类病种的第一词向量；

将每个第一词向量与相应的单病种得分相乘，以得到第一相乘结果；

将病种细分得分对应的混淆关系进行编码，以生成用于表示所述混淆关系的第二词向量；

将每个第二词向量与相应的病种细分得分相乘，以得到第二相乘结果；以及

使用自然语言模型对所述第一相乘结果和所述第二相乘结果形成的序列进行推理。

2.根据权利要求1所述的装置，其特征在于，当所述程序指令由处理器运行时，还使得所述装置在使用分类模型获得初始分类结果中执行以下操作：

使用主干网络对待测医学图像进行特征提取，以获得所述待测医学图像中的病灶特征；以及

使用第一分类器基于所述病灶特征输出待分类病种的单病种得分。

3.根据权利要求2所述的装置，其特征在于，当所述程序指令由处理器运行时，还使得所述装置在使用分类模型获得初始分类结果中执行以下操作：

使用第二分类器基于所述病灶特征输出与所述待分类病种相关的病种细分得分。

4.根据权利要求1所述的装置，其特征在于，当所述程序指令由处理器运行时，还使得所述装置在使用分类模型获得初始分类结果之前执行以下操作：

根据待分类任务中的待分类病种的数量，确定第一分类器的数量，使得每个第一分类器用于输出相应的一个待分类病种的单病种得分；以及

根据待分类病种之间混淆关系的数量，确定第二分类器的数量，使得每个第二分类器用于输出相应的一个混淆关系中的病种细分得分。

5.根据权利要求1所述的装置，其特征在于，当所述程序指令由处理器运行时，还使得所述装置在使用自然语言模型进行推理中执行以下操作：

使用预设阈值对初始分类结果进行筛选，以获得筛选后的分类结果；以及

使用自然语言模型对筛选后的分类结果进行推理。

6.根据权利要求5所述的装置，其特征在于，当所述程序指令由处理器运行时，还使得所述装置在使用预设阈值对初始分类结果进行筛选，以获得筛选后的分类结果中执行以下操作：

将每个第一分类器输出的单病种得分与相应的预设阈值进行比较，以筛选出大于预设阈值的阳性单病种得分；

根据阳性单病种得分对应的待分类病种，筛选出相应的混淆关系；以及

根据筛选出的混淆关系，将相应的病种细分得分确定为筛选后的病种细分得分。

7.根据权利要求1-6任一所述的装置，其特征在于，所述待测医学图像包括眼底图像。

8.根据权利要求1所述的装置，其特征在于，当所述程序指令由处理器运行时，还使得所述装置在使用自然语言模型进行推理中执行以下操作：

将待测医学图像所属患者的基本信息、病史信息和检查信息中的至少一种与所述初始分类结果一起输入至自然语言模型进行推理。