What Does a Chemical Language Model Know About Molecules?
About
Chemical language models (cLMs) are widely assumed to learn surface-level syntactic patterns rather than learning meaningful molecular semantics. Here, we apply sparse autoencoders (SAEs) to MolFormer, an encoder-only cLM, to mechanistically examine how molecular representations are built across layers. We discover that early layers rely on position-tracking latents to parse molecular grammar, while later layers encode atom-in-substructure and pharmacologically relevant features. Additionally, we show that non-canonical SMILES produce more disruptive representation shifts than invalid SMILES, driven by position-latent disruption propagating across layers. To support further exploration, we develop InterMol, an interactive visualizer for SAE activations on molecular strings and structures.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| ADMET Properties Prediction | TDC AMES | AUROC0.82 | 20 | |
| Drug discovery classification | ClinTox | ROC-AUC91 | 15 | |
| ADMET Properties Prediction | TDC HIA Hou | AUROC98 | 15 | |
| ADMET Properties Prediction | TDC Pgp Broccatelli | AUROC0.93 | 15 | |
| ADMET Properties Prediction | TDC Bioavailability Ma | AUROC0.72 | 15 | |
| Absorption Regression | FreeSolv (train-test) | RMSE1.2 | 8 | |
| Absorption Regression | Lipophilicity AstraZeneca (train test) | RMSE0.75 | 8 | |
| Absorption Regression | AqSolDB (train-test) | RMSE1.65 | 8 | |
| ADMET Classification | PAMPA NCATS | ROC-AUC76 | 8 | |
| ADMET Classification | BBB | ROC-AUC90 | 8 |