Multi-task Learning with Weak Class Labels: Leveraging iEEG to Detect Cortical Lesions in Cryptogenic Epilepsy

by   Bilal Ahmed, et al.

Multi-task learning (MTL) is useful for domains in which data originates from multiple sources that are individually under-sampled. MTL methods are able to learn classification models that have higher performance as compared to learning a single model by aggregating all the data together or learning a separate model for each data source. The performance of these methods relies on label accuracy. We address the problem of simultaneously learning multiple classifiers in the MTL framework when the training data has imprecise labels. We assume that there is an additional source of information that provides a score for each instance which reflects the certainty about its label. Modeling this score as being generated by an underlying ranking function, we augment the MTL framework with an added layer of supervision. This results in new MTL methods that are able to learn accurate classifiers while preserving the domain structure provided through the rank information. We apply these methods to the task of detecting abnormal cortical regions in the MRIs of patients suffering from focal epilepsy whose MRI were read as normal by expert neuroradiologists. In addition to the noisy labels provided by the results of surgical resection, we employ the results of an invasive intracranial-EEG exam as an additional source of label information. Our proposed methods are able to successfully detect abnormal regions for all patients in our dataset and achieve a higher performance as compared to baseline methods.


page 1

page 2

page 3

page 4


Relational Multi-Task Learning: Modeling Relations between Data and Tasks

A key assumption in multi-task learning is that at the inference time th...

Multi-task Learning with Crowdsourced Features Improves Skin Lesion Diagnosis

Machine learning has a recognised need for large amounts of annotated da...

Evaluating Classifiers Without Expert Labels

This paper considers the challenge of evaluating a set of classifiers, a...

From Multi-label Learning to Cross-Domain Transfer: A Model-Agnostic Approach

In multi-label learning, a particular case of multi-task learning where ...

Weak Supervision for Affordable Modeling of Electrocardiogram Data

Analysing electrocardiograms (ECGs) is an inexpensive and non-invasive, ...

Weakly Supervised Multi-task Learning for Concept-based Explainability

In ML-aided decision-making tasks, such as fraud detection or medical di...

Learning More for Free - A Multi Task Learning Approach for Improved Pathology Classification in Capsule Endoscopy

The progress in Computer Aided Diagnosis (CADx) of Wireless Capsule Endo...

Please sign up or login with your details

Forgot password? Click here to reset