中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Using multi-task learning to improve diagnostic performance of convolutional neural networks

文献类型:会议论文

作者Fang, Mengjie2,3; Dong, Di2,3; Sun, Ruijia1; Fan, Li4; Sun, Yingshi1; Liu, Shiyuan4; Tian, Jie2,3
出版日期2019
会议日期2019-2
会议地点San Diego, California, USA
英文摘要

Due to the complex biological and physical mechanisms, the correlations between the classification objects of clinical tasks and the medical imaging phenotype are always ambiguous and implied, which makes it difficult to train a powerful diagnostic convolutional neural network (CNN) model efficiently. In this study, we propose a generic multi-task learning (MTL) CNN framework to achieve higher classification accuracy and better generalization. The proposed framework is designed to carry out the major diagnostic task and several auxiliary tasks simultaneously. It encourages the models to learn more beneficial representation following the underlying relation among patients’ clinical characteristics, obvious imaging findings and quantitative imaging phenotype. We evaluate our approach on two clinical applications, namely advanced gastric cancer (AGC) serosa invasion diagnosis and discrimination of lung invasive adenocarcinoma manifesting as ground-glass nodule (GGN). Two datasets are utilized, which contain 357 AGC patients’ venous phase contrast-enhanced CT volumes and 236 GGN patients’ non-contrast CT volumes respectively. Several subjective CT morphology characteristics and common clinical characteristics are collected and used as the auxiliary tasks. To evaluate the generality of our strategy, CNNs with and without natural image-based pre-training are successively incorporated into the framework. The experimental results demonstrate that the proposed MTL CNN framework is able to improve the diagnostic performance significantly (7.4%-12.8% AUC increase and 3.5%-7.9% accuracy increase).

源URL[http://ir.ia.ac.cn/handle/173211/48553]  
专题自动化研究所_中国科学院分子影像重点实验室
通讯作者Tian, Jie
作者单位1.Radiology Department, Peking University Cancer Hospital & Institute, Beijing 100142, China
2.CAS Key Laboratory of Molecular Imaging, Institute of Automation, Chinese Academy of Sciences, Beijing 100190, China
3.University of Chinese Academy of Sciences, Beijing 100049, China
4.Department of Radiology, Changzheng Hospital, Second Military Medical University, Shanghai 200003, China
推荐引用方式
GB/T 7714
Fang, Mengjie,Dong, Di,Sun, Ruijia,et al. Using multi-task learning to improve diagnostic performance of convolutional neural networks[C]. 见:. San Diego, California, USA. 2019-2.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。