On Modelling Label Uncertainty in Deep Neural Networks: Automatic Estimation of Intra- Observer Variability in 2D Echocardiography Quality Assessment

• Published in: IEEE transactions on medical imaging Vol. 39; no. 6; pp. 1868 - 1883
• Main Authors: Liao, Zhibin, Girgis, Hany, Abdi, Amir, Vaseli, Hooman, Hetherington, Jorden, Rohling, Robert, Gin, Ken, Tsang, Teresa, Abolmaesumi, Purang
• Format: Journal Article
• Language: English
• Published: United States IEEE 01.06.2020; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: 2D echocardiography; Artificial neural networks; Cardiovascular diseases; cross-entropy; Deep learning; deep neural networks; DenseNet; Echocardiography; Image quality; Label uncertainty; Labeling; Labels; loss regularization; LSTM; Machine learning; Mapping; Medical imaging; Model accuracy; Modelling; Network reliability; Neural networks; Observational learning; Observers; Quality; Quality assessment; Quality control; Regression; Reliability analysis; Statistical analysis; Time dependence; Two dimensional models; Uncertainty; Variability
• ISSN: 0278-0062; 1558-254X
• DOI: 10.1109/TMI.2019.2959209

Uncertainty of labels in clinical data resulting from intra-observer variability can have direct impact on the reliability of assessments made by deep neural networks. In this paper, we propose a method for modelling such uncertainty in the context of 2D echocardiography (echo), which is a routine procedure for detecting cardiovascular disease at point-of-care. Echo imaging quality and acquisition time is highly dependent on the operator's experience level. Recent developments have shown the possibility of automating echo image quality quantification by mapping an expert's assessment of quality to the echo image via deep learning techniques. Nevertheless, the observer variability in the expert's assessment can impact the quality quantification accuracy. Here, we aim to model the intra-observer variability in echo quality assessment as an aleatoric uncertainty modelling regression problem with the introduction of a novel method that handles the regression problem with categorical labels. A key feature of our design is that only a single forward pass is sufficient to estimate the level of uncertainty for the network output. Compared to the 0.11 ± 0.09 absolute error (in a scale from 0 to 1) archived by the conventional regression method, the proposed method brings the error down to 0.09 ± 0.08, where the improvement is statistically significant and equivalents to 5.7% test accuracy improvement. The simplicity of the proposed approach means that it could be generalized to other applications of deep learning in medical imaging, where there is often uncertainty in clinical labels.