arrow
返回

Prior knowledge-based multi-task learning network for pulmonary nodule classification

delete2025-04-01
delete0
PRE
AI
P
Peng Xue
H
Hang Lü
Y
Yu Fu
M
Meirong Ren
T
Taohui Xiao
Z
Zhili Zhang
E
Enqing Dong *
DOI:10.1016/j.compmedimag.2025.102511delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
The morphological characteristics of pulmonary nodule, also known as the attributes, are crucial for classification of benign and malignant nodules. In clinical, radiologists usually conduct a comprehensive analysis of correlations between different attributes, to accurately judge pulmonary nodules are benign or malignant. However, most of pulmonary nodule classification models ignore the inherent correlations between different attributes, leading to unsatisfactory classification performance. To address these problems, we propose a prior knowledge-based multi-task learning (PK-MTL) network for pulmonary nodule classification. To be specific, the correlations between different attributes are treated as prior knowledge, and established through multi- order task transfer learning. Then, the complex correlations between different attributes are encoded into hypergraph structure, and leverage hypergraph neural network for learning the correlation representation. On the other hand, a multi-task learning framework is constructed for joint segmentation, benign-malignant classification and attribute scoring of pulmonary nodules, aiming to improve the classification performance of pulmonary nodules comprehensively. In order to embed prior knowledge into multi-task learning framework, a feature fusion block is designed to organically integrate image-level features with attribute prior knowledge. In addition, a channel-wise cross attention block is constructed to fuse the features of encoder and decoder, to further improve the segmentation performance. Extensive experiments on LIDC-IDRI dataset show that our proposed method can achieve 91.04% accuracy for diagnosing malignant nodules, obtaining the state-of-art results.
Keyword:
Pulmonary nodule classification
Transfer learning
Multi-task learning
Hypergraph neural network
Feature fusion

期刊

Computerized Medical Imaging and Graphics 封面图
Computerized Medical Imaging and Graphics
IF:
4.9
论文数:
2.4K
被引数:
5.0K

机构

S
shandong university
学者数:
9.5W
论文数: 6.4W
被引数: 94
引用论文

引用论文

Search-Based Extraction of Component-Based Architecture from Object-Oriented Systems
err2008-01-01
err0
PREAI
errSylvain Chardigny; Abdelhak Seriai; Mourad Oussalah; Dalila Tamzalit
err分享
err收藏
Phase II study of epirubicin plus oxaliplatin and infusional 5-fluorouracil as first-line combination therapy in patients with metastatic or advanced gastric cancer
err2007-06-01
err0
PREAI
errChen X. Zhang; Sui Huang; Nong Xu; Jia W. Fang; Peng Shen; Yin H. Bao; Bo H. Mou; Ming G. Shi; Xing L. Zhong; Ping J. Xiong
err分享
err收藏
MSCS-DeepLN: Evaluating lung nodule malignancy using multi-scale cost-sensitive neural networksMscs-deepln: 使用多尺度成本敏感神经网络评估肺结节恶性程度
err2020-10-01
err77
PREAI
errXu, Xiuyuan; Wang, Chengdi; Guo, Jixiang; Gan, Yuncui; Wang, Jianyong; Bai, Hongli; Zhang, Lei; Li, Weimin; Yi, Zhang
err分享
err收藏
学者 查看更多内容