arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.10185cs.LGstat.ML
收藏

基于EHR时间序列数据的多任务学习和多任务预训练的综合评价

A Comprehensive Evaluation of Multi-task Learning and Multi-task Pre-training on EHR Time-series Data

MIT(麻省理工学院);University of Toronto(多伦多大学);NYU(纽约大学);Vector Institute(Vector研究所);SickKids(病童医院)

Matthew B. A. McDermott, Bret Nestor, Evan Kim, Wancong Zhang, Anna Goldenberg, Peter Szolovits, Marzyeh Ghassemi

摘要:

多任务学习(MTL)是一种机器学习技术,旨在通过利用跨多个任务的信息来提高模型性能。它已被广泛用于各种数据模式,包括电子健康记录(EHR)数据。然而,尽管大量使用EHR数据,很少有系统的调查MTL的效用在不同的可能的任务和培训计划的利益在医疗保健。在这项工作中,我们检查MTL跨电池的任务EHR时间序列数据。我们发现,虽然MTL确实遭受常见的负迁移,但我们可以通过MTL预训练结合单任务微调来实现显着的收益。我们证明了这些收益可以以任务独立的方式实现,并且不仅在传统学习下提供了微小的改进,而且在少样本学习环境中也有显着的收益,从而表明这可能是一种可扩展的工具,可以在重要的医疗保健环境中提供更好的性能。

英文摘要:

Multi-task learning (MTL) is a machine learning technique aiming to improve model performance by leveraging information across many tasks. It has been used extensively on various data modalities, including electronic health record (EHR) data. However, despite significant use on EHR data, there has been little systematic investigation of the utility of MTL across the diverse set of possible tasks and training schemes of interest in healthcare. In this work, we examine MTL across a battery of tasks on EHR time-series data. We find that while MTL does suffer from common negative transfer, we can realize significant gains via MTL pre-training combined with single-task fine-tuning. We demonstrate that these gains can be achieved in a task-independent manner and offer not only minor improvements under traditional learning, but also notable gains in a few-shot learning context, thereby suggesting this could be a scalable vehicle to offer improved performance in important healthcare contexts.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑