Boosting Multi-task Learning Through Combination of Task Labels - with Applications in ECG Phenotyping

Ming En Hsieh, Vincent S. Tseng

研究成果: Conference contribution同行評審

6 引文 斯高帕斯(Scopus)

摘要

Multi-task learning has increased in importance due to its superior performance by learning multiple different tasks simultaneously and its ability to perform several different tasks using a single model. In medical phenotyping, task labels are costly to acquire and might contain a certain degree of label noise. This decreases the efficiency of using additional human labels as auxiliary tasks when applying multi-task learning to medical phenotyping. In this work, we proposed an effective multi-task learning framework, CO-TASK, to boost multi-task learning performance by generating auxiliary tasks through COmbination of TASK Labels. The proposed CO-TASK framework generates auxiliary tasks without additional labeling effort, is robust to a certain degree of label noise, and can be applied in parallel with various multitask learning techniques. We evaluated our performance using the CIFAR-MTL dataset and demonstrated its effectiveness in medical phenotyping using two large-scale ECG phenotyping datasets, an 18 diseases multi-label ECG-P18 dataset and an echocardiogram diagnostic from electrocardiogram dataset ECG-EchoLVH. On the CIFAR-MTL dataset, we doubled the average per-task performance gain of the multi-task learning model from 4.38% to 9.78%. With the proposed task-aware imbalance data sampler, the CO-TASK framework can effectively deal with the different imbalance ratios for the different tasks in electrocardiogram phenotyping datasets. The proposed framework combined with noisy annotations as minor tasks increased the sensitivity by 7.1% compared to the single-task model while maintaining the same specificity as the doctor annotations on the ECG-EchoLVH dataset.

原文English
主出版物標題35th AAAI Conference on Artificial Intelligence, AAAI 2021
發行者Association for the Advancement of Artificial Intelligence
頁面7771-7779
頁數9
ISBN(電子)9781713835974
出版狀態Published - 2021
事件35th AAAI Conference on Artificial Intelligence, AAAI 2021 - Virtual, Online
持續時間: 2 2月 20219 2月 2021

出版系列

名字35th AAAI Conference on Artificial Intelligence, AAAI 2021
9A

Conference

Conference35th AAAI Conference on Artificial Intelligence, AAAI 2021
城市Virtual, Online
期間2/02/219/02/21

指紋

深入研究「Boosting Multi-task Learning Through Combination of Task Labels - with Applications in ECG Phenotyping」主題。共同形成了獨特的指紋。

引用此