机构:[1]Harbin Inst Technol Shenzhen, Shenzhen, Peoples R China[2]Peng Cheng Lab, Shenzhen, Peoples R China[3]Capital Med Univ, Beijing, Peoples R China[4]Capital Med Univ, Xuanwu Hosp, Beijing, Peoples R China首都医科大学宣武医院
The brain age has been proven a phenotype with relevance to cognitive performance and brain disease. With the development of deep learning, brain age estimation accuracy has been greatly improved. However, such methods may incur over-fitting and suffer from poor generalizations, especially for insufficient brain imaging data. This paper presents a novel regularization method that penalizes the predictive distribution using knowledge distillation and introduces additional knowledge to reinforce the learning process. During knowledge distillation, we propose a gated distillation mechanism to enable the student model to attentively learn key knowledge from the teacher model, given the assumption that the teacher may not always be correct. Moreover, to enhance the capability of knowledge transfer, the hint representation similarity is also adopted to regularize the model training. We evaluate the model by a cohort of 3655 subjects from 4 public datasets, demonstrating that the proposed method improves the prediction performance over several well-established models, where the mean absolute error of the estimated ages is 2.129 years.
基金:
Innovation Team and Talents Cultivation Program of National Administration of Traditional Chinese Medicine [ZYYCXTD-C-202004]; Basic Research Foundation of Shenzhen Science and Technology Stable Support Program [GXWD20201230155427003-20200822115709001]; National Natural Science Foundation of China [62106113, 62106115]
语种:
外文
被引次数:
WOS:
第一作者:
第一作者机构:[1]Harbin Inst Technol Shenzhen, Shenzhen, Peoples R China[2]Peng Cheng Lab, Shenzhen, Peoples R China
通讯作者:
通讯机构:[1]Harbin Inst Technol Shenzhen, Shenzhen, Peoples R China[2]Peng Cheng Lab, Shenzhen, Peoples R China[3]Capital Med Univ, Beijing, Peoples R China[4]Capital Med Univ, Xuanwu Hosp, Beijing, Peoples R China
推荐引用方式(GB/T 7714):
Yang Yanwu,Guo Xutao,Ye Chenfei,et al.Regularizing Brain Age Prediction via Gated Knowledge Distillation[J].INTERNATIONAL CONFERENCE ON MEDICAL IMAGING WITH DEEP LEARNING, VOL 172.2022,172:1430-1443.
APA:
Yang, Yanwu,Guo, Xutao,Ye, Chenfei,Xiang, Yang&Ma, Ting.(2022).Regularizing Brain Age Prediction via Gated Knowledge Distillation.INTERNATIONAL CONFERENCE ON MEDICAL IMAGING WITH DEEP LEARNING, VOL 172,172,
MLA:
Yang, Yanwu,et al."Regularizing Brain Age Prediction via Gated Knowledge Distillation".INTERNATIONAL CONFERENCE ON MEDICAL IMAGING WITH DEEP LEARNING, VOL 172 172.(2022):1430-1443