Fibro-CoSANet: pulmonary fibrosis prognosis prediction using a convolutional self attention network
Abstract
Idiopathic pulmonary fibrosis (IPF) is a restrictive interstitial lung disease that causes lung function decline by lung tissue scarring. Although lung function decline is assessed by the forced vital capacity (FVC), determining the accurate progression of IPF remains a challenge. To address this challenge, we proposed Fibro-CoSANet, a novel end-to-end multi-modal learning based approach, to predict the FVC decline. Fibro-CoSANet utilized computed tomography images and demographic information in convolutional neural network frameworks with a stacked attention layer. Extensive experiments on the OSIC Pulmonary Fibrosis Progression Dataset demonstrated the superiority of our proposed Fibro-CoSANet by achieving new state-of-the-art modified Laplace log-likelihood score of -6.68. This network may benefit research areas concerned with designing networks to improve the prognostic accuracy of IPF. The source-code for Fibro-CoSANet is available at: https://github.com/zabir-nabil/Fibro-CoSANet.
- Publication:
-
Physics in Medicine and Biology
- Pub Date:
- November 2021
- DOI:
- 10.1088/1361-6560/ac36a2
- arXiv:
- arXiv:2104.05889
- Bibcode:
- 2021PMB....66v5013A
- Keywords:
-
- pulmonary fibrosis;
- computed tomography (CT);
- convolutional neural network;
- self-attention;
- computer-aided diagnosis;
- Electrical Engineering and Systems Science - Image and Video Processing;
- Computer Science - Computer Vision and Pattern Recognition;
- Computer Science - Machine Learning
- E-Print:
- 12 Pages