arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.03460cs.LGq-bio.MN
收藏

用于分子表示学习的重编程语言模型

Reprogramming Language Models for Molecular Representation Learning

University of California, Berkeley(加州大学伯克利分校);IBM Research(IBM研究院)

Ria Vinod, Pin-Yu Chen, Payel Das

摘要:

迁移学习的最新进展使其成为一种通过迁移学习到的表征进行领域自适应的有前途的方法。当替代任务具有有限的明确定义和标记的数据样本时,这一点尤其重要,这在分子数据领域很常见。这使得迁移学习成为解决分子学习任务的理想方法。虽然对抗性重编程已被证明是一种成功的方法,可以将神经网络重新用于替代任务,但大多数工作都考虑同一域中的源任务和替代任务。在这项工作中,我们提出了一种新的算法,通过字典学习(R2 DL)表示重编程,用于对抗性地重编程预先训练的语言模型的分子学习任务,通过利用学习表示在大规模的最先进的语言模型的动机。对抗程序学习密集源模型输入空间(语言数据)和稀疏目标模型输入空间(例如,化学和生物分子数据),通过字典学习,使用k-SVD求解器来近似编码数据的稀疏表示。R2 DL实现了由在领域特定数据上训练的现有技术毒性预测模型建立的基线,并且在有限的训练数据设置中优于基线,从而为具有分子数据的任务建立了领域不可知迁移学习的途径。

英文摘要:

Recent advancements in transfer learning have made it a promising approach for domain adaptation via transfer of learned representations. This is especially when relevant when alternate tasks have limited samples of well-defined and labeled data, which is common in the molecule data domain. This makes transfer learning an ideal approach to solve molecular learning tasks. While Adversarial reprogramming has proven to be a successful method to repurpose neural networks for alternate tasks, most works consider source and alternate tasks within the same domain. In this work, we propose a new algorithm, Representation Reprogramming via Dictionary Learning (R2DL), for adversarially reprogramming pretrained language models for molecular learning tasks, motivated by leveraging learned representations in massive state of the art language models. The adversarial program learns a linear transformation between a dense source model input space (language data) and a sparse target model input space (e.g., chemical and biological molecule data) using a k-SVD solver to approximate a sparse representation of the encoded data, via dictionary learning. R2DL achieves the baseline established by state of the art toxicity prediction models trained on domain-specific data and outperforms the baseline in a limited training-data setting, thereby establishing avenues for domain-agnostic transfer learning for tasks with molecule data.

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

↑