Auditing frozen chemical language model embeddings for drug-delivery screening: fingerprint baselines, random-initialization controls, and scaffold-memorization control tasks

Frozen embeddings from pretrained chemical language models (cLMs) are widely used as features for ADMET and drug-delivery prediction, often with a linear probe and little validation of what the representation actually contributes. We audit two small public cLMs (ChemBERTa-77M-MLM and ChemBERTa-77M-MTR, 3.4M parameters each) on three delivery-relevant endpoints from openly licensed sources: blood–brain barrier penetration (B3DB, CC0), PAMPA permeability (NCATS via PubChem), and Caco-2 permeability (OpenADMET, CC-BY-4.0). For each endpoint we compare frozen mean-pooled embeddings with (i) Morgan-fingerprint linear probes with train-only tuned regularization, (ii) the same architecture with random weights (five initialization seeds), and (iii) scaffoldconstant control tasks in the style of Hewitt and Liang that quantify how much a linear probe can score by memorizing scaffold identity. On observed means, tuned fingerprint probes are slightly ahead of frozen embeddings on all three endpoints (paired per-split differences of 0.016–0.032 in AUROC or R2, smaller than split-to-split variation); trained embeddings exceed random-initialization controls by 0.04–0.10; endpoint probes are selective above scaffold-memorization controls, and the memorization ceiling rises with layer depth in the primary MLM model. Decodability of bulk physicochemical descriptors (molecular weight, TPSA, fraction sp3, aromatic rings) is largely preserved under random initialization, indicating an architectural rather than learned origin for those signals; learned increments appear for logP, hydrogen-bond donors, and rotatable bonds. We quantify three practitioner pitfalls: pooling choice changes endpoint AUROC by up to 0.05; tuning the fin- gerprint baseline’s regularization changes it by up to 0.03 AUROC or 0.20 R2; and scaffoldsplit construction details change means by up to 0.04, exceeding seed standard deviations. The source datasets are public and cited; all curation scripts, pipeline code, and per-seed results are included in the accompanying package. The study is compute-bounded to CPU and small models, which limits the generality of the conclusions to that regime.

Authors

Publication Details

Journal
ChemRxiv
Published
2026-11-02
DOI
https://doi.org/10.26434/chemrxiv.15011567/v1
Primary Topic
Computational Drug Discovery Methods
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Auditing frozen chemical language model embeddings for drug-delivery screening: fingerprint baselines, random-initialization controls, and scaffold-memorization control tasks

Badri N. Dubey, Shereen Anand, Swananda Marathe, Anmol Kumar
ChemRxiv
Computational Drug Discovery Methods
preprint

Auditing frozen chemical language model embeddings for drug-delivery screening: fingerprint baselines, random-initialization controls, and scaffold-memorization control tasks

Badri N. Dubey, Shereen Anand, Swananda Marathe, Anmol Kumar
preprint en

Abstract

Frozen embeddings from pretrained chemical language models (cLMs) are widely used as features for ADMET and drug-delivery prediction, often with a linear probe and little validation of what the representation actually contributes. We audit two small public cLMs (ChemBERTa-77M-MLM and ChemBERTa-77M-MTR, 3.4M parameters each) on three delivery-relevant endpoints from openly licensed sources: blood–brain barrier penetration (B3DB, CC0), PAMPA permeability (NCATS via PubChem), and Caco-2 permeability (OpenADMET, CC-BY-4.0). For each endpoint we compare frozen mean-pooled embeddings with (i) Morgan-fingerprint linear probes with train-only tuned regularization, (ii) the same architecture with random weights (five initialization seeds), and (iii) scaffoldconstant control tasks in the style of Hewitt and Liang that quantify how much a linear probe can score by memorizing scaffold identity. On observed means, tuned fingerprint probes are slightly ahead of frozen embeddings on all three endpoints (paired per-split differences of 0.016–0.032 in AUROC or R2, smaller than split-to-split variation); trained embeddings exceed random-initialization controls by 0.04–0.10; endpoint probes are selective above scaffold-memorization controls, and the memorization ceiling rises with layer depth in the primary MLM model. Decodability of bulk physicochemical descriptors (molecular weight, TPSA, fraction sp3, aromatic rings) is largely preserved under random initialization, indicating an architectural rather than learned origin for those signals; learned increments appear for logP, hydrogen-bond donors, and rotatable bonds. We quantify three practitioner pitfalls: pooling choice changes endpoint AUROC by up to 0.05; tuning the fin- gerprint baseline’s regularization changes it by up to 0.03 AUROC or 0.20 R2; and scaffoldsplit construction details change means by up to 0.04, exceeding seed standard deviations. The source datasets are public and cited; all curation scripts, pipeline code, and per-seed results are included in the accompanying package. The study is compute-bounded to CPU and small models, which limits the generality of the conclusions to that regime.

ChemRxiv
Computational Drug Discovery Methods
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon
Auditing frozen chemical language model embeddings for drug-delivery screening: fingerprint baselines, random-initialization controls, and scaffold-memorization control tasks — Badri N. Dubey, Shereen Anand, et al. · ChemRxiv (2026) | TGRS Research Map | TGRS