Journal of Shanghai Jiao Tong University ›› 2022, Vol. 56 ›› Issue (8): 1078-1088.doi: 10.16183/j.cnki.jsjtu.2021.113

• Biomedical Engineering • Previous Articles     Next Articles

Multi-Feature Extraction of Pulmonary Nodules Based on LSTM and Attention Structure

NI Yangfan1,2, YANG Yuanyuan2, XIE Zhe1,2, ZHENG Dezhong1,2, WANG Weidong3()   

  1. 1. Laboratory for Medical Imaging Informatics, Shanghai Institution of Technical Physics, Chinese Academy of Sciences, Shanghai 200080, China
    2. University of Chinese Academy of Sciences, Beijing 100049, China
    3. Chinese PLA General Hospital, Beijing 100089, China
  • Received:2021-04-14 Online:2022-08-28 Published:2022-08-26
  • Contact: WANG Weidong


The accurate classification of shape, edge, and internal features of pulmonary nodules can not only assist the radiologists in their daily diagnosis, but also improve the writing efficiency of imaging reports. This paper proposes a multi-task classification model based on long-short term memory (LSTM) and attention structure, which merges the shared features among different classification tasks through attention mechanism to improve the feature extraction performance of the current task. The classifier based on temporal sequence LSTM structure can effectively screen the shared features and improve the efficiency of information transmission in the multi-task model. Experimental results show that compared with the traditional multi-task structure, the proposed model can achieve better classification results on the public dataset LIDC-IDRI, and assist doctors to quickly obtain nodule characteristics.

Key words: pulmonary nodule, attention structure, long-short term memory(LSTM)network, multi-task classification

CLC Number: