Unimodal-Uniform Constrained Wasserstein Training for Medical Diagnosis

Abstract

The labels in medical diagnosis task are usually discrete and successively distributed. For example, the Diabetic Retinopathy Diagnosis (DR) involves five health risk levels: no DR (0), mild DR (1), moderate DR (2), severe DR (3) and proliferative DR (4). This labeling system is common for medical disease. Previous methods usually construct a multi-binary-classification task or propose some reparameter schemes in the output unit. In this paper, we target on this task from the perspective of loss function. More specifically, the Wasserstein distance is utilized as an alternative, explicitly incorporating the inter-class correlations by pre-defining its ground metric. Then, the ground metric which serves as a linear, convex or concave increasing function w.r.t. the Euclidean distance in a line is explored from an optimization perspective.Meanwhile, this paper also proposes of constructing the smoothed target labels that model the inlier and outlier noises by using a unimodal-uniform mixture distribution. Different from the one-hot setting, the smoothed label endues the computation of Wasserstein distance with more challenging features. With either one-hot or smoothed target label, this paper systematically concludes the practical closed-form solution. We evaluate our method on several medical diagnosis tasks (e.g., Diabetic Retinopathy and Ultrasound Breast dataset) and achieve state-of-the-art performance.

Embedding Semantic Hierarchy in Discrete Optimal Transport for Risk Minimization

“…In this paper, we resort to the optimal transport distance as an alternative for empirical risk minimization [9,10,11,12]. With the low-cost modification of the loss function perspective, our solution can be added on any up-to-date general deep networks in a plug-and-play fashion.…”

Navigation-Oriented Scene Understanding for Robotic Autonomy: Learning to Segment Driveability in Egocentric Images

“…Soft labelling schemes have successfully been applied to semantic segmentation in prior work [26], [24], however these works only consider a binary classification case, and generate pixel labels not based on inter-class relations, but on spatial location, to capture ambiguity along object boundaries. Conversely, works which demonstrate the use of soft labels for ordinal classification [26], [22], [27], apply it to other tasks such as full-image ranking (e.g. age estimation, aesthetic quality prediction or medical diagnosis) or pixel-wise regression (e.g.…”

Deep 3D-CNN for Depression Diagnosis with Facial Video Recording of Self-Rating Depression Scale Questionnaire

“…With the fast development of deep learning for recognition [11], [12], [13], [14], [15], [16], [17], [18], a hierarchical approach is developed for automatically interpreting depression based on the SDS assessment, its associated FE, and action video recording, among other things. To be more specific, we effectively extract the temporal information from each question-wise video by adjusting the 3D convolutional neural networks to the particular question (3D-CNN) [19].…”