Phenotypical Ontology Driven Framework for Multi-Task Learning
| Autor: | Zijun Yao, Mohamed Ghalwash, James Codella, Daby Sow, Prithwish Chakraporty |
|---|---|
| Jazyk: | angličtina |
| Rok vydání: | 2020 |
| Předmět: |
FOS: Computer and information sciences
Network architecture Computer Science - Machine Learning Computer science business.industry Computer Science - Artificial Intelligence Deep learning Multi-task learning Machine Learning (stat.ML) Construct (python library) Ontology (information science) USable Machine learning computer.software_genre Machine Learning (cs.LG) Artificial Intelligence (cs.AI) Statistics - Machine Learning Key (cryptography) Graph (abstract data type) Artificial intelligence business computer |
| Zdroj: | CHIL |
| Popis: | Despite the large number of patients in Electronic Health Records (EHRs), the subset of usable data for modeling outcomes of specific phenotypes are often imbalanced and of modest size. This can be attributed to the uneven coverage of medical concepts in EHRs. In this paper, we propose OMTL, an Ontology-driven Multi-Task Learning framework, that is designed to overcome such data limitations. The key contribution of our work is the effective use of knowledge from a predefined well-established medical relationship graph (ontology) to construct a novel deep learning network architecture that mirrors this ontology. It can effectively leverage knowledge from a well-established medical relationship graph (ontology) by constructing a deep learning network architecture that mirrors this graph. This enables common representations to be shared across related phenotypes, and was found to improve the learning performance. The proposed OMTL naturally allows for multitask learning of different phenotypes on distinct predictive tasks. These phenotypes are tied together by their semantic distance according to the external medical ontology. Using the publicly available MIMIC-III database, we evaluate OMTL and demonstrate its efficacy on several real patient outcome predictions over state-of-the-art multi-task learning schemes. To be appear on ACM CHIL 2021 |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|