Multi-Label Softmax Networks for Pulmonary Nodule Classification Using Unbalanced and Dependent Categories.

Abstract

Radiographic attributes of lung nodules remedy the shortcomings of lung cancer computer-assisted diagnosis systems, which provides interpretable diagnostic reference for doctors. However, current studies fail to dedicate multi-label classification of lung nodules using convolutional neural networks (CNNs) and are inferior in exploiting statistical dependency between the labels. In addition, data imbalance is an indispensable problem to be reckoned with when employing CNNs to perform lung nodule classification. It introduces greater challenges especially in the multi-label classification. In this paper, we propose a method called MLSL-Net to discriminate lung nodule characteristics and simultaneously address the challenges. Particularly, the proposal employs multi-label softmax loss (MLSL) as the performance index, aiming to reduce the ranking errors between the labels and within the labels during training, thereby optimizing ranking loss and AUC directly. Such criterions can better evaluate the classifier's performance on the multi-label imbalanced dataset. Furthermore, a scale factor is introduced based on the investigation of the max surrogate function. Different from preceding usages, the small factor is used so that to narrow the discrepancy of gradients produced by different labels. More interestingly, this factor also facilitates the exploit of label dependency. Experimental results on the LIDC-IDRI dataset as well as another akin dataset demonstrate that MLSL-Net can effectively perform multi-label classification despite the imbalance issue. Meanwhile, the results confirm the responsibility of the factor for capturing label correlations, accordingly leading to more accurate predictions.