[發(fā)明專利]眼底圖像分類模型訓(xùn)練方法及設(shè)備

有效

• 申請(qǐng)?zhí)枺?/td>: 202010140098.6
• 申請(qǐng)日: 2020-03-03
• 公開（公告）號(hào): CN111383210B
• 公開（公告）日: 2023-06-06
• 發(fā)明（設(shè)計(jì)）人: 賀婉佶;王欣;熊健皓;趙昕;和超;張大磊
• 申請(qǐng)（專利權(quán)）人: 上海鷹瞳醫(yī)療科技有限公司
• 主分類號(hào): G06T7/00
• 分類號(hào): G06T7/00;G06V40/18;G06V10/774;G16H50/20;G06V10/82;G06N3/045
• 代理公司: 北京泛華偉業(yè)知識(shí)產(chǎn)權(quán)代理有限公司 11280
• 代理人: 王勇

1.一種眼底圖像分類模型訓(xùn)練方法，其特征在于，包括：

獲取訓(xùn)練數(shù)據(jù)，包括眼底圖像及用于表達(dá)其疾病類型的標(biāo)簽；

利用經(jīng)過訓(xùn)練的第一機(jī)器學(xué)習(xí)模型對(duì)所述眼底圖像進(jìn)行分類，得到用于表達(dá)其屬于各種疾病類型的第一置信度信息；

利用所述訓(xùn)練數(shù)據(jù)以及所述第一置信度信息對(duì)第二機(jī)器學(xué)習(xí)模型進(jìn)行訓(xùn)練，訓(xùn)練過程中所述第二機(jī)器學(xué)習(xí)模型對(duì)所述眼底圖像進(jìn)行分類，得到用于表達(dá)其屬于各種疾病類型的第二置信度信息，并根據(jù)所述第二置信度信息與所述標(biāo)簽的差異以及所述第二置信度信息與所述第一置信度信息的差異優(yōu)化所述第二機(jī)器學(xué)習(xí)模型的參數(shù)；

其中所述第一機(jī)器學(xué)習(xí)模型中包括多個(gè)神經(jīng)網(wǎng)絡(luò)；以及

其中利用經(jīng)過訓(xùn)練的第一機(jī)器學(xué)習(xí)模型對(duì)所述眼底圖像進(jìn)行分類，得到用于表達(dá)其疾病類型的第一置信度信息，包括：

所述多個(gè)神經(jīng)網(wǎng)絡(luò)分別對(duì)所述眼底圖像進(jìn)行分類，分別輸出用于表達(dá)其屬于各種疾病類型的置信度信息；

獲取所述多個(gè)神經(jīng)網(wǎng)絡(luò)的權(quán)重；

根據(jù)所述多個(gè)神經(jīng)網(wǎng)絡(luò)輸出的置信度信息及對(duì)應(yīng)的權(quán)重計(jì)算得到所述第一置信度信息。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述多個(gè)神經(jīng)網(wǎng)絡(luò)的結(jié)構(gòu)不同。

3.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述權(quán)重為所述神經(jīng)網(wǎng)絡(luò)的性能值。

4.根據(jù)權(quán)利要求1-3中任一項(xiàng)所述的方法，其特征在于，訓(xùn)練所述第二機(jī)器學(xué)習(xí)模型使用的損失函數(shù)為：

λ*crossentropy(conf_s,label)+(1-λ)*crossentropy(conf_t,conf_s)

其中，label為所述標(biāo)簽，conf_t為所述第一置信度信息，conf_s為所述第二置信度信息，λ為超參數(shù)。

5.根據(jù)權(quán)利要求1-3中任一項(xiàng)所述的方法，其特征在于，所述眼底圖像為視盤局部圖像，所述疾病類型是與視盤相關(guān)的疾病類型。

6.一種眼底圖像分類方法，其特征在于，包括：

利用權(quán)利要求1-5中任一項(xiàng)所述的方法訓(xùn)練的所述第二機(jī)器學(xué)習(xí)模型對(duì)眼底圖像進(jìn)行分類，輸出用于表達(dá)該眼底圖像屬于各種疾病類型的置信度信息。

7.一種眼底圖像分類模型訓(xùn)練設(shè)備，其特征在于，包括：至少一個(gè)處理器；以及與所述至少一個(gè)處理器通信連接的存儲(chǔ)器；其中，所述存儲(chǔ)器存儲(chǔ)有可被所述一個(gè)處理器執(zhí)行的指令，所述指令被所述至少一個(gè)處理器執(zhí)行，以使所述至少一個(gè)處理器執(zhí)行如權(quán)利要求1-5中任意一項(xiàng)所述的眼底圖像分類模型訓(xùn)練方法。

8.一種眼底圖像分類設(shè)備，其特征在于，包括：至少一個(gè)處理器；以及與所述至少一個(gè)處理器通信連接的存儲(chǔ)器；其中，所述存儲(chǔ)器存儲(chǔ)有可被所述一個(gè)處理器執(zhí)行的指令，所述指令被所述至少一個(gè)處理器執(zhí)行，以使所述至少一個(gè)處理器執(zhí)行如權(quán)利要求6所述的眼底圖像分類方法。