bioRxivBioinformaticsPreprint
Probing isn't Causality in Molecular Models
Satya Pratik SrivastavaiD, Sharath Krishna Chundru, Harshit SinghiD, Rajeev Kumar SinghiD
Abstract
Linear probes establish whether a chemical variable can be recovered from a molecular representation; they do not establish how that variable contributes to subsequent predictions. We replace a concept-associated scalar component at an intermediate block and measure the change through a separately fitted downstream outcome readout. Comparisons with random directions and architecture-matched random encoders distinguish recoverability, intervention selectivity, and a pretraining-associated selectivity advantage. In Uni-Mol, block-12 total-distortion edits predict the observed energy-change sign for 97.6\% of held-out pairs, with Spearman $\rho=0.787$. Structure-clustered intervals for the primary direction and encoder contrasts exclude zero. A stricter 20-direction sensitivity analysis does not survive correction across six cells. A post-review analysis finds substantial concept--outcome alignment; intervention effects persist after removal of the same-block outcome direction, but this does not establish that subsequent encoder computation is necessary. In MolCLR, pretraining improves dipole recovery by $0.080\,R^2$, whereas concept-minus-random sign-accuracy contrasts are $0.029$ and $-0.015$ at the two tested blocks. Their paired uncertainty is unresolved, so they do not establish an absence of selective effects. We propose an operational evidence ladder whose conclusions are conditional on the concept, intervention, control family, and fitted readout.
- Citations
- 0
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Beyond performance: How design choices shape chemical language models
Inken Fender, Jannik Adrian Gut, Thomas Lemmin·bioRxiv·May 23, 2025
- MolDeBERTa: Foundational Model for Physicochemical and Substructure-Informed Molecular Representation Learning
Gabriel Bianchin de Oliveira, Fahad Saeed·bioRxiv·Jun 10, 2026
- MOL-AE: Auto-Encoder Based Molecular Representation Learning With 3D Cloze Test Objective
Junwei Yang, Kangjie Zheng, … Hao Zhou·bioRxiv·Apr 13, 2024
- Deep learning models for chemical perturbation prediction do not yet utilise drug molecular features
Jinming Bai, Sharon Prince, Geoff Nitschke·bioRxiv·May 15, 2026
- MolMAE: A Surface-Centric Multimodal Masked Autoencoder for Molecular Representation Learning
Jiaqing Li·bioRxiv·Jul 14, 2026
- MolRep: A Deep Representation Learning Library for Molecular Property Prediction
Jiahua Rao, Shuangjia Zheng, … Yuedong Yang·bioRxiv·Jan 13, 2021