nach0: Multimodal Natural and Chemical Languages Foundation Model
arXiv:2311.12410 · doi:10.1039/D4SC00966E
Abstract
Large Language Models (LLMs) have substantially driven scientific progress in various domains, and many papers have demonstrated their ability to tackle complex problems with creative solutions. Our paper introduces a new foundation model, nach0, capable of solving various chemical and biological tasks: biomedical question answering, named entity recognition, molecular generation, molecular synthesis, attributes prediction, and others. nach0 is a multi-domain and multi-task encoder-decoder LLM pre-trained on unlabeled text from scientific literature, patents, and molecule strings to incorporate a range of chemical and linguistic knowledge. We employed instruction tuning, where specific task-related instructions are utilized to fine-tune nach0 for the final set of tasks. To train nach0 effectively, we leverage the NeMo framework, enabling efficient parallel optimization of both base and large model versions. Extensive experiments demonstrate that our model outperforms state-of-the-art baselines on single-domain and cross-domain tasks. Furthermore, it can generate high-quality outputs in molecular and textual formats, showcasing its effectiveness in multi-domain setups.
Accepted to Chemical Science Journal. Models are publicly available via https://huggingface.co/insilicomedicine/nach0_base and https://huggingface.co/insilicomedicine/nach0_large
References in corpus (11)
- BioBERT: a pre-trained biomedical language representation model for biomedical text mining
- Domain-Specific Language Model Pretraining for Biomedical Natural Language Processing
- BioGPT: Generative Pre-trained Transformer for Biomedical Text Generation and Mining
- SELFIES and the future of molecular string representations
- Galactica: A Large Language Model for Science
- Keeping it Simple: Language Models can learn Complex Molecular Distributions
- SciFive: a text-to-text transformer model for biomedical literature
- Does Synthetic Data Generation of LLMs Help Clinical Text Mining?
- Group SELFIES: A Robust Fragment-Based Molecular String Representation
- Results of the seventh edition of the BioASQ Challenge
- Language models can generate molecules, materials, and protein binding sites directly in three dimensions as XYZ, CIF, and PDB files