Two major challenges hinder the deployment of deep learning-based systems into clinical practice: the need for numerous high-quality well-labeled data and the lack of explainability. Attention models, originated from natural language processing, have been popular to address the label scarcity problem and encourage model explainability. In this work, we developed a domain knowledge-guided attention model for disease diagnosis with only coarse scan-level labels and the population-level domain knowledge. The use of guided attention models encourages the deep learning-based diagnosis model to focus on the area of interests in an end-to-end manner. The research interest is to diagnose subjects with idiopathic pulmonary fibrosis (IPF) among subjects with interstitial lung disease (ILD) using an axial chest high resolution computed tomography (HRCT) scan. Our dataset contains 279 IPF patients and 423 non-IPF ILD patients. The network’s performance was evaluated by the area under the receiver operating characteristic curve (AUC). We observe that without attention-based loss function, the IPF diagnosis model reaches satisfactory performance (AUC=0.972), but lack explainability; when increasing the relative importance of attention-based loss, the IPF diagnosis model increases performance (AUC=0.988), along with the model explainability. Our contributions are (1) developing an IPF diagnosis model that only uses scan-level weak supervision; (2) incorporating population-level domain knowledge into the training of IPF diagnosis model in an end-to-end manner; (3) enhancing the explainability of deep learning systems by introducing attention mechanisms.
Access to the requested content is limited to institutions that have purchased or subscribe to SPIE eBooks.
You are receiving this notice because your organization may not have SPIE eBooks access.*
*Shibboleth/Open Athens users─please
sign in
to access your institution's subscriptions.
To obtain this item, you may purchase the complete book in print or electronic format on
SPIE.org.
INSTITUTIONAL Select your institution to access the SPIE Digital Library.
PERSONAL Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.