NeurIPS 2020 | ChemBERTa+:用于分子性质预测的大规模自监督预训练

本文介绍了ChemBERTa模型,它基于HuggingFace中的RoBERTa transformer实现,在多个PubChem子集上训练。在MoleculeNet的分类任务上评估该模型,下游任务中预训练数据集扩大可改善性能。还比较了不同tokenizer算法,建议尽量使用ChemBERTa - 2。

原文标题:ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Prediction

Chithrananda, Seyone, Gabriel Grand, and Bharath Ramsundar. "ChemBERTa: large-scale self-supervised pretraining for molecular property prediction." arXiv preprint arXiv:2010.09885 (2020).

地址:https://arxiv.org/abs/2209.01712v1

基于HuggingFace中的RoBERTa transformer实现。对RoBERTa的实现使用了12个注意力头和6层,产生了72种不同的注意力机制。mask每个输入字符串中15%的令牌。用的是最大值。词汇表大小为52Ktoken和最大值。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值