atlas

Research map / Large language models / Agents, code and math

From Artificially Real to Real: Leveraging Pseudo Data from Large Language Models for Low-Resource Molecule Discovery

AAAI 2024 · Large language models › Agents, code and math · approach: LLM-guided program synthesis · 5 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Generates 1M pseudo molecule-description pairs with ChatGPT via retrieval-based one-shot prompting, then uses them for domain adaptation (pre-training then fine-tuning) or data augmentation to train small T5 models for molecule captioning and text-to-molecule generation.

Problem

Cross-modal molecule discovery (molecule captioning and text-based de novo generation) is limited by the scarcity of parallel molecule-description pairs, which are costly to annotate with chemistry expertise.

Approach

Use LLM-generated "artificially-real" parallel data built from unlabeled PubChem molecules, with retrieval of Tanimoto-similar annotated molecules as one-shot in-context examples, and use the pseudo data as a second pre-training (domain adaptation) stage before fine-tuning on real data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use pseudo data for domain adaptation, then fine-tune on real data instead of mixing pseudo and real data during fine-tuningT5 · ChEBI-20, PCdes, and DrugBank-23 molecular captioning and molecule generationtask metricsHelped
Increase pseudo-data volume during domain adaptation (ablation)Ada-T5 · ChEBI-20 molecular captioning and molecule generationoverall metricsHelped
Increase pseudo-data volume during fine-tuning augmentation (ablation)Aug-T5 · ChEBI-20 molecular captioningoverall metricsHurt
Increase model scale from T5_small to T5_base for data augmentation (ablation)Aug-T5 · Molecular captioningoverall metricsNo clear effect
Increase model scale from T5_small to T5_base for domain adaptation (ablation)Ada-T5 · Molecular captioningoverall metricsHelped
Fine-tune with a mixture of real and pseudo data rather than fine-tuning on real data after pseudo-data domain adaptationAda-T5 versus Aug-T5 · Molecule generationoverall task metricsHurt
Increase pseudo-data volume during fine-tuning augmentation (ablation)Aug-T5 · ChEBI-20 molecule generationoverall metricsMixed

Surprising findings

Limitations

Experimental setup

Datasets: ChEBI-20 PCdes DrugBank-23 PseudoMD-1M PubChem unlabeled SMILES

Benchmarks: ChEBI-20 PCdes DrugBank-23

Metrics: BLEU-4 ROUGE-2 METEOR Accuracy Validity MACCS FTS RDK FTS Morgan FTS FCD Levenshtein distance

Baselines: T5 MolT5 ChatGPT MolXPT Text&Chem T5

Builds on: MolT5 Text2Mol in-context learning Text&Chem T5

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy