Time-distanced gates in long short-term memory networks

被引:20
作者
Gao, Riqiang [1 ]
Tang, Yucheng [1 ]
Xu, Kaiwen [1 ]
Huo, Yuankai [1 ]
Bao, Shunxing [1 ]
Antic, Sanja L. [2 ]
Epstein, Emily S. [2 ]
Deppen, Steve [3 ]
Paulson, Alexis B. [4 ]
Sandler, Kim L. [4 ]
Massion, Pierre P. [2 ]
Landman, Bennett A. [1 ,4 ]
机构
[1] Vanderbilt Univ, Elect Engn & Comp Sci, Nashville, TN 37235 USA
[2] Vanderbilt Univ, Sch Med, Med, Nashville, TN 37235 USA
[3] Vanderbilt Univ, Med Ctr, Thorac Surg, Nashville, TN 37235 USA
[4] Vanderbilt Univ, Med Ctr, Radiol, Nashville, TN 37235 USA
关键词
Lung cancer diagnosis; Longitudinal; Distanced LSTM; Temporal Emphasis Model;
D O I
10.1016/j.media.2020.101785
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The Long Short-Term Memory (LSTM) network is widely used in modeling sequential observations in fields ranging from natural language processing to medical imaging. The LSTM has shown promise for interpreting computed tomography (CT) in lung screening protocols. Yet, traditional image-based LSTM models ignore interval differences, while recently proposed interval-modeled LSTM variants are limited in their ability to interpret temporal proximity. Meanwhile, clinical imaging acquisition may be irregularly sampled, and such sampling patterns may be commingled with clinical usages. In this paper, we propose the Distanced L STM (DL STM) by introducing time-distanced (i.e., time distance to the last scan) gates with a temporal emphasis model (TEM) targeting at lung cancer diagnosis (i.e., evaluating the malignancy of pulmonary nodules) . Briefly, (1) the time distance of every scan to the last scan is modeled explicitly, (2) time-distanced input and forget gates in DLSTM are introduced across regular and irregular sampling sequences, and (3) the newer scan in serial data is emphasized by the TEM. The DLSTM algorithm is evaluated with both simulated data and real CT images (from 1794 National Lung Screening Trial (NLST) patients with longitudinal scans and 1420 clinical studied patients). Experimental results on simulated data indicate the DLSTM can capture families of temporal relationships that cannot be detected with traditional LSTM. Cross-validation on empirical CT datasets demonstrates that DLSTM achieves leading performance on both regularly and irregularly sampled data (e.g., improving LSTM from 0.6785 to 0.7085 on F1 score in NLST). In external-validation on irregularly acquired data, the benchmarks achieved 0.8350 (CNN feature) and 0.8380 (with LSTM) on AUC score, while the proposed DLSTM achieves 0.8905. In conclusion, the DLSTM approach is shown to be compatible with families of linear, quadratic, exponential, and log-exponential temporal models. The DLSTM can be readily extended with other temporal dependence interactions while hardly increasing overall model complexity. (c) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:10
相关论文
共 35 条
  • [1] Reduced Lung-Cancer Mortality with Low-Dose Computed Tomographic Screening
    Aberle, Denise R.
    Adams, Amanda M.
    Berg, Christine D.
    Black, William C.
    Clapp, Jonathan D.
    Fagerstrom, Richard M.
    Gareen, Ilana F.
    Gatsonis, Constantine
    Marcus, Pamela M.
    Sicks, JoRean D.
    [J]. NEW ENGLAND JOURNAL OF MEDICINE, 2011, 365 (05) : 395 - 409
  • [2] End-to-end lung cancer screening with three-dimensional deep learning on low-dose chest computed tomography
    Ardila, Diego
    Kiraly, Atilla P.
    Bharadwaj, Sujeeth
    Choi, Bokyung
    Reicher, Joshua J.
    Peng, Lily
    Tse, Daniel
    Etemadi, Mozziyar
    Ye, Wenxing
    Corrado, Greg
    Naidich, David P.
    Shetty, Shravya
    [J]. NATURE MEDICINE, 2019, 25 (06) : 954 - +
  • [3] Bayer Justin, 2014, ARXIV14117610
  • [4] Boas F. E., 2012, IMAGING MED, V4, P229, DOI [DOI 10.2217/IIM.12.13, 10.2217/iim.12.13]
  • [5] Cai J., 2017, ARXIV170704912
  • [6] Cho K, 2014, PREPRINT
  • [7] Chung J., 2015, P NIPS, P2980
  • [8] Cox D, 2016, INT C LEARN REPR ICL
  • [9] LUNG-TUMOR GROWTH CORRELATES WITH GLUCOSE-METABOLISM MEASURED BY FLUORIDE-18 FLUORODEOXYGLUCOSE POSITRON EMISSION TOMOGRAPHY
    DUHAYLONGSOD, FG
    LOWE, VJ
    PATZ, EF
    VAUGHN, AL
    COLEMAN, E
    WOLFE, WG
    [J]. ANNALS OF THORACIC SURGERY, 1995, 60 (05) : 1348 - 1352
  • [10] Finn C., 2006, ADV NEURAL INFORM PR, P64