
原文标题: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和最大值。

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

4158

被折叠的 条评论
为什么被折叠?



