Classification of lung nodules in CT images based on Wasserstein distance in differential geometry

TL;DR

This paper introduces a novel 3D shape analysis method using Wasserstein distance within differential geometry to improve classification accuracy of lung nodules in CT images, addressing limitations of previous shape-based methods.

Contribution

It presents the first application of Wasserstein distance for lung nodule classification, utilizing a new spherical optimal mass transport algorithm for more accurate shape analysis.

Findings

Wasserstein distance effectively differentiates benign and malignant nodules.

The new spherical optimal mass transport algorithm improves computational efficiency.

Shape classification accuracy is enhanced by invariance to rigid motions and scalings.

Abstract

Lung nodules are commonly detected in screening for patients with a risk for lung cancer. Though the status of large nodules can be easily diagnosed by fine needle biopsy or bronchoscopy, small nodules are often difficult to classify on computed tomography (CT). Recent works have shown that shape analysis of lung nodules can be used to differentiate benign lesions from malignant ones, though existing methods are limited in their sensitivity and specificity. In this work we introduced a new 3D shape analysis within the framework of differential geometry to calculate the Wasserstein distance between benign and malignant lung nodules to derive an accurate classification scheme. The Wasserstein distance between the nodules is calculated based on our new spherical optimal mass transport, this new algorithm works directly on sphere by using spherical metric, which is much more accurate and efficient than previous methods. In the process of deformation, the area-distortion factor gives a probability measure on the unit sphere, which forms the Wasserstein space. From known cases of benign and malignant lung nodules, we can calculate a unique optimal mass transport map between their correspondingly deformed Wasserstein spaces. This transportation cost defines the Wasserstein distance between them and can be used to classify new lung nodules into either the benign or malignant class. To the best of our knowledge, this is the first work that utilizes Wasserstein distance for lung nodule classification. The advantages of Wasserstein distance are it is invariant under rigid motions and scalings, thus it intrinsically measures shape distance even when the underlying shapes are of high complexity, making it well suited to classify lung nodules as they have different sizes, orientations, and appearances.