arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2009.00792cs.LGstat.ML
收藏

Select-ProtoNet:学习选择用于Few-Shot疾病亚型预测

Select-ProtoNet: Learning to Select for Few-Shot Disease Subtype Prediction

Macau University of Science and Technology(澳门科技大学)

Ziyi Yang, Jun Shu, Yong Liang, Deyu Meng, Zongben Xu

摘要:

当前的机器学习在计算机视觉和许多其他领域取得了很大的进展,这归功于大量高质量的训练样本,而它在基因组数据分析上的效果不是很好,因为它们被称为小数据。在我们的工作中,我们专注于Few-Shot疾病亚型预测问题,识别相似患者的亚组,通过对小数据的训练,可以指导特定个体的治疗决策。事实上,医生和临床医生总是通过同时研究几个相互关联的临床变量来解决这个问题。我们尝试模拟这样的临床观点,并引入元学习技术来开发一个新的模型,该模型可以从相关的临床任务中提取共同的经验或知识,并将其转移到帮助解决新的任务。我们的新模型建立在精心设计的元学习器之上,称为原型网络,它是一种简单而有效的元学习机器,用于Few-Shot图像分类。我们观察到基因表达数据与图像数据相比具有特定的高维和高噪声特性,提出了一种新的扩展,通过增加两个模块来解决这些问题。具体地说,我们增加了一个特征选择层,以自动过滤掉与疾病无关的基因,并结合了一个样本重加权策略,以自适应地去除噪声数据,同时扩展模型能够从有限数量的训练样本学习,并推广良好。仿真和真实基因表达数据实验证实了该方法在预测疾病亚型和识别潜在疾病相关基因方面的优越性。

英文摘要:

Current machine learning has made great progress on computer vision and many other fields attributed to the large amount of high-quality training samples, while it does not work very well on genomic data analysis, since they are notoriously known as small data. In our work, we focus on few-shot disease subtype prediction problem, identifying subgroups of similar patients that can guide treatment decisions for a specific individual through training on small data. In fact, doctors and clinicians always address this problem by studying several interrelated clinical variables simultaneously. We attempt to simulate such clinical perspective, and introduce meta learning techniques to develop a new model, which can extract the common experience or knowledge from interrelated clinical tasks and transfer it to help address new tasks. Our new model is built upon a carefully designed meta-learner, called Prototypical Network, that is a simple yet effective meta learning machine for few-shot image classification. Observing that gene expression data have specifically high dimensionality and high noise properties compared with image data, we proposed a new extension of it by appending two modules to address these issues. Concretely, we append a feature selection layer to automatically filter out the disease-irrelated genes and incorporate a sample reweighting strategy to adaptively remove noisy data, and meanwhile the extended model is capable of learning from a limited number of training examples and generalize well. Simulations and real gene expression data experiments substantiate the superiority of the proposed method for predicting the subtypes of disease and identifying potential disease-related genes.

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

↑