眼轴预测模型的训练方法、眼轴预测方法和装置

本公开的实施例涉及人工智能，具体地，涉及适用于一种眼轴预测模型的训练方法、眼轴预测方法和装置。

背景技术：

1、眼轴为把眼睛看作一台光学仪器来描述，从眼球接收光线的最表层，即从角膜-晶状体-玻璃体-视网膜(感受光线最里的一层)的距离看成是物理中的光学系统的一条中轴线。

2、相关技术中，在进行未来眼轴预测时，如果采用现有技术只能将时间变量作为输入数据直接输入到模型中对模型进行训练，现有技术的模型主要有两种方式，一种是根据当前单一的固定时间点的眼轴测量数据去预测未来某一时间点的眼轴，第二种是根据当前n个固定的时间点的眼轴测量数据去预测未来某一时间点的眼轴，例如根据时间点1、时间点2、......、时间点n预测时间点n+1的眼轴。以上两种方式，都存在输入的数据与时间位置固定，输入形式必须对应的问题，但是在实际测量场景中，这种模型无法利用不定时的输入数据，与实际数据采集情况不符，难以推广实现，模型也难以进行预测；即使按照模型固定输入形式输入了数据，由于输入数据的时间位置必须固定对应，依然会导致模型无法智能的利用其他非预设时间点输入的数据，存在模型预测眼轴的准确度较低的技术问题。

技术实现思路

1、本文中描述的实施例提供了一种眼轴预测模型的训练方法、眼轴预测方法和装置，克服了上述问题。

2、第一方面，根据本公开的内容，提供了一种眼轴预测模型的训练方法，包括：

3、训练信息提取模型，所述信息提取模型用于提取眼底图像对应的层级特征向量，所述层级特征向量包括：第一层级向量和第二层级向量，所述信息提取模型包括第一子模型和第二子模型，所述第一子模型用于提取所述第一层级向量，所述第二子模型用于提取所述第二层级向量，所述第一层级向量用于描述与近视病灶无关的向量，所述第二层级向量用于描述与近视病灶相关的向量；

4、基于所述信息提取模型确定目标眼底图像对应的层级特征向量，所述目标眼底图像中包括目标对象不同年龄/相同年龄的不同月份的眼底图像，所述目标眼底图像对应的层级特征向量用于描述所述目标对象的轴性近视相关信息向量；

5、基于词向量模型确定所述目标对象的生理信息文本向量；

6、基于所述目标眼底图像对应的层级特征向量和所述生理信息文本向量，对网络模型结构进行训练，得到眼轴预测模型，所述眼轴预测模型用于对所述目标对象进行时间序列眼轴预测，所述时间序列眼轴预测用于预测所述目标对象的未来眼轴；

7、其中，所述眼轴预测模型中包括时间块机制，所述时间块机制用于划分所述目标对象的年龄/月份，使得所述目标眼底图像对应的层级特征向量和所述生理信息文本向量均隶属于所述目标对象的不同年龄/相同年龄的不同月份。

8、第二方面，根据本公开的内容，提供了一种眼轴预测方法，包括：

9、获取待预测眼底图像，所述待预测眼底图像中包括目标对象不同年龄/相同年龄的不同月份的眼底图像；

10、将所述待预测眼底图像输入信息提取模型中，根据所述信息提取模型的输出确定所述待预测眼底图像对应的层级特征向量；

11、基于词向量模型确定所述目标对象的生理信息文本向量；

12、将所述待预测眼底图像对应的层级特征向量以及所述生理信息文本向量输入眼轴预测模型中，根据所述眼轴预测模型的输出确定所述目标对象的未来眼轴；

13、其中，所述眼轴预测模型为第一方面中任一项所述的方法所训练生成的眼轴预测模型，所述信息提取模型为第一方面中任一项所述的方法所训练生成的信息提取模型。

14、第三方面，根据本公开的内容，提供了一种眼轴预测模型的训练装置，包括：

15、第一训练构建模块，用于训练信息提取模型，所述信息提取模型用于提取眼底图像对应的层级特征向量，所述层级特征向量包括：第一层级向量和第二层级向量，所述信息提取模型包括第一子模型和第二子模型，所述第一子模型用于提取所述第一层级向量，所述第二子模型用于提取所述第二层级向量，所述第一层级向量用于描述与近视病灶无关的向量，所述第二层级向量用于描述与近视病灶相关的向量；

16、第一确定模块，用于基于所述信息提取模型确定目标眼底图像对应的层级特征向量，所述目标眼底图像中包括目标对象不同年龄/相同年龄的不同月份的眼底图像，所述目标眼底图像对应的层级特征向量用于描述所述目标对象的轴性近视相关信息向量；

17、第二确定模块，用于基于词向量模型确定所述目标对象的生理信息文本向量；

18、第二训练模块，用于基于所述目标眼底图像对应的层级特征向量和所述生理信息文本向量，对网络模型结构进行训练，得到眼轴预测模型，所述眼轴预测模型用于对所述目标对象进行时间序列眼轴预测，所述时间序列眼轴预测用于预测所述目标对象的未来眼轴；

19、其中，所述眼轴预测模型中包括时间块机制，所述时间块机制用于划分所述目标对象的年龄/月份，使得所述目标眼底图像对应的层级特征向量和所述生理信息文本向量均隶属于所述目标对象的不同年龄/相同年龄的不同月份。

20、第四方面，根据本公开的内容，提供了一种眼轴预测装置，包括：

21、获取模块，用于获取待预测眼底图像，所述待预测眼底图像中包括目标对象不同年龄/相同年龄的不同月份的眼底图像；

22、提取模块，用于将所述待预测眼底图像输入信息提取模型中，根据所述信息提取模型的输出确定所述待预测眼底图像对应的层级特征向量；

23、第三确定模块，用于基于词向量模型确定所述目标对象的生理信息文本向量；

24、第四确定模块，用于将所述待预测眼底图像对应的层级特征向量以及所述生理信息文本向量输入眼轴预测模型中，根据所述眼轴预测模型的输出确定所述目标对象的未来眼轴；

25、其中，所述眼轴预测模型为第一方面中任一项所述的方法所训练生成的眼轴预测模型，所述信息提取模型为第一方面中任一项所述的方法所训练生成的信息提取模型。

26、第五方面，提供了一种计算机设备，包括存储器和处理器，存储器中存储有计算机程序，处理器执行计算机程序时实现如以上任意一个实施例中眼轴预测模型的训练方法的步骤，或者，实现如以上任意一个实施例中眼轴预测方法的步骤。

27、第六方面，提供了一种计算机可读存储介质，计算机可读存储介质上存储有计算机程序，计算机程序被处理器执行时实现如以上任意一个实施例中眼轴预测模型的训练方法的步骤，或者，实现如以上任意一个实施例中眼轴预测方法的步骤。

28、本技术实施例提供的眼轴预测模型的训练方法，训练信息提取模型，信息提取模型用于提取眼底图像对应的层级特征向量，层级特征向量包括：第一层级向量和第二层级向量，信息提取模型包括第一子模型和第二子模型，第一子模型用于提取第一层级向量，第二子模型用于提取第二层级向量，第一层级向量用于描述与近视病灶无关的向量，第二层级向量用于描述与近视病灶相关的向量；基于信息提取模型确定目标眼底图像对应的层级特征向量，目标眼底图像中包括目标对象不同年龄/相同年龄的不同月份的眼底图像，目标眼底图像对应的层级特征向量用于描述目标对象的轴性近视相关信息向量；基于词向量模型确定目标对象的生理信息文本向量；基于目标眼底图像对应的层级特征向量和生理信息文本向量，对网络模型结构进行训练，得到眼轴预测模型，眼轴预测模型用于对目标对象进行时间序列眼轴预测，时间序列眼轴预测用于预测目标对象的未来眼轴；其中，眼轴预测模型中包括时间块机制，时间块机制用于划分目标对象的年龄/月份，使得目标眼底图像对应的层级特征向量和生理信息文本向量均隶属于目标对象的不同年龄/相同年龄的不同月份。如此，通过设定时间token以及模型结构，能够有效利用多次随访数据，不需要固定的输入数据形式，在任意时间点采集的数据均可以利用训练模型训练得到的眼轴预测模型，能够对目标对象的未来眼轴进行有效预测，提升未来眼轴的预测准确度，也解决了现有技术的预测模型难以推广实现的技术问题。

29、上述说明仅是本技术实施例技术方案的概述，为了能够更清楚了解本技术实施例的技术手段，而可依照说明书的内容予以实施，并且为了让本技术实施例的上述和其它目的、特征和优点能够更明显易懂，以下特举本技术的具体实施方式。