Transfer Learning via Optimal Transportation for Integrative Cancer Patient Stratification

Transfer Learning via Optimal Transportation for Integrative Cancer Patient Stratification

Ziyu Liu, Wei Shao, Jie Zhang, Min Zhang, Kun Huang

Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence
Main Track. Pages 2760-2766. https://doi.org/10.24963/ijcai.2021/380

The Stratification of early-stage cancer patients for the prediction of clinical outcome is a challenging task since cancer is associated with various molecular aberrations. A single biomarker often cannot provide sufficient information to stratify early-stage patients effectively. Understanding the complex mechanism behind cancer development calls for exploiting biomarkers from multiple modalities of data such as histopathology images and genomic data. The integrative analysis of these biomarkers sheds light on cancer diagnosis, subtyping, and prognosis. Another difficulty is that labels for early-stage cancer patients are scarce and not reliable enough for predicting survival times. Given the fact that different cancer types share some commonalities, we explore if the knowledge learned from one cancer type can be utilized to improve prognosis accuracy for another cancer type. We propose a novel unsupervised multi-view transfer learning algorithm to simultaneously analyze multiple biomarkers in different cancer types. We integrate multiple views using non-negative matrix factorization and formulate the transfer learning model based on the Optimal Transport theory to align features of different cancer types. We evaluate the stratification performance on three early-stage cancers from the Cancer Genome Atlas (TCGA) project. Comparing with other benchmark methods, our framework achieves superior accuracy for patient outcome prediction.
Keywords:
Machine Learning: Transfer, Adaptation, Multi-task Learning
Machine Learning Applications: Applications of Unsupervised Learning
Machine Learning Applications: Bio/Medicine