Mask Distillation Network for Conjunctival Hyperemia Severity Classification  

作　　者：Mingchao Li Kun Huang Xiao Ma Yuexuan Wang Wen Fan Qiang Chen 

机构地区：[1]School of Computer Science and Engineering,Nanjing University of Science and Technology,Nanjing,210094,China [2]Department of Ophthalmology,The First Affiliated Hospital with Nanjing Medical University,Nanjing,210029,China

出　　处：《Machine Intelligence Research》2023年第6期909-922,共14页机器智能研究（英文版）

基　　金：This work was supported in part by National Natural Science Foundation of China(Nos.62172223 and 61671242);the Fundamental Research Funds for the Central Universities(No.30921013105).

摘　　要：To achieve automatic,fast and accurate severity classification of bulbar conjunctival hyperemia severity,we proposed a novel prior knowledge-based framework called mask distillation network(MDN).The proposed MDN consists of a segmentation network and a classification network with teacher-student branches.The segmentation network is used to generate a bulbar conjunctival mask and the classification network divides the severity of bulbar conjunctival hyperemia into four grades.In the classification network,we feed the original image and the image with the bulbar conjunctival mask into the student and teacher branches respectively,and an attention consistency loss and a classification consistency loss are used to keep a similar learning mode for these two branches.This design of“different input but same output”,named mask distillation(MD),aims to introduce the regional prior knowledge that“bulbar conjunctival hyperemia severity classification is only related to the bulbar conjunctiva region”.Extensive experiments on 5117 anterior segment images have proven the effectiveness of mask distillation technology:1)The accuracy of the MDN student branch is 3.5%higher than that of a single optimal baseline network and 2%higher than that of the baseline network combination.2)In the test phase,only the student branch is needed,and no additional segmentation network is required.The framework only takes 0.003 s to classify a single image,achieving the fastest speed in all the methods we compared.3)Compared with a single baseline network,the attention of both teacher and student branches in the MDN has been intuitively improved.

关 键 词：Mask distillation(MD) conjunctiva hyperemia attention mechanism severity classification deep learning 

分 类 号：TN911.7[电子电信—通信与信息系统] TP183[电子电信—信息与通信工程] R318[自动化与计算机技术—控制理论与控制工程]