cs.LGJun 6, 2026

De novo molecular generation with optical property preconditioning at the token level

Authors: Haozhe HuangManuel Gonzalez LastreHyun Suk ParkJorge A. Campos-Gonzalez-AnguloXinjian LiuAlán Aspuru-Guzik

Organizations: Department of Computer Science, University of Toronto, 40 St George St., Toronto, ON M5S 2E4, Canada · 2Vector Institute for Artificial Intelligence, W1140-108 College St., Schwartz Reisman Innovation Campus, Toronto, ON M5G 0C6, Canada · 3Departamento de Física Teórica de la Materia Condensada, Universidad Autónoma de Madrid, 28049, Madrid, Spain. · Department of Chemistry, University of Toronto, Lash Miller Chemical Laboratories, 80 St. George Street, ON M5S 3H6, Toronto, Canada · Department of Materials Science & Engineering, University of Toronto, 184 College St., M5S 3E4,2026 Toronto, Canada · Department of Chemical Engineering & Applied Chemistry, University of Toronto, 200 College St. ON M5S 3E5, Toronto, CanadaJun · 7Acceleration Consortium, 700 University Ave., M7A 2S4, Toronto, Canada · 8Senior Fellow, Canadian Institute for Advanced Research (CIFAR), 661 University Ave., M5G 1M1, Toronto, Canada · 9NVIDIA, 431 King St W #6th, M5V 1K4, Toronto, Canada

Abstract

Designing OLED molecules with targeted optical properties remains challenging due to the scarcity of high-quality data and the limited reliability of conditional control in generative models across chemical motifs. Here, we benchmark a token-conditioned autoregressive language model for OLED molecular generation in a realistic low-data regime. A GPT2 model is pretrained on large chemical corpora, augmented with discrete property tokens, and fine-tuned using multi-task optimisation. Conditioning targets vertical absorption energy and oscillator strength, with the HOMO-LUMO gap included as an auxiliary electronic descriptor. Generated molecules are evaluated at the TDDFT level to assess distributional fidelity and controllability. The generated library reproduces the dominant optical-property support of the training distribution while shifting towards lower molecular weight and fewer heavy atoms. Token-level control is consistently directional across conditioning bins, but is not fully orthogonal and exhibits local calibration irregularities. A chemotype-resolved analysis further shows that controllability depends strongly on local electronic environments: moderately conjugated aromatic-carbon motifs are associated with improved joint target satisfaction, whereas electron-withdrawing motifs, particularly aryl nitriles, show systematic red-shifting and reduced controllability. These results establish a quantitative benchmark for conditional OLED molecular generation and show that model reliability must be assessed in chemically meaningful subspaces rather than from aggregate property distributions alone.

Explore similar work

CardsList
  1. OLEDLM: A Unified Language Model for OLED Molecular Design

    Jul 22, 2026Fukang Wen, Yuchong Tang, Jingyuan Li +9Molecular Generation