MIST: Molecular Insight SMILES Transformers
MIST is a family of molecular foundation models for molecular property prediction.
The models were pre-trained on SMILES strings from the
Enamine REAL Space dataset using the Masked Language Modeling (MLM) objective, then fine-tuned for downstream prediction tasks.
Further information is available in
our pre-print on arXiv.
Model Details
Model Description
This fine-tuned MIST variant consists of the MIST-28M encoder finetuned on various solvent properties:
- Catalan Solvatochromic Parameters
SPP1, SB3, SA4: (Source - https://www.stenutz.eu/chem/solv25.php) Details of the SPP solubility parameters (Solvent Polarity/Polarizability), together with a more comprehensive list and a discussion on solvent scales in general can be found here.
pKa: Deprecated use pKa in DMSO model mist-models/mist-28M-6zlgl2qn-pKa- Kamlet-Taft parameters:
alpha/ beta: (Source - DOI 10.1021/acscentsci.1c01151) The Kamlet-Taft solvatochromatic parameters dataset was used. Kamlet-Taft parameters are used to characterize solvent properties, particularly their ability to participate in hydrogen bonding and their polarizability. The three parameters are α (hydrogen bond acidity, the solvent's ability to donate a hydrogen bond), β (hydrogen bond basicity, the solvent's ability to accept a hydrogen bond) and π* (polarizability, the solvent's overall polarity). The fine-tuning dataset consisted of 182 Kamlet-Taft values measured using NMR.
Fine-tuned MIST models consist of the pretrained MIST model (the encoder), followed by a task network.
The task networks consist of a two-layer MLP with Gaussian Error Linear Units (GELU) activations and dropout.
The final hidden state vectors for all tokens in the sequence are pooled to produce a single embedding vector.
Consistent with prior works, the encoder hidden states were pooled by taking the hidden state of the first token.
- Developed by: Electrochemical Energy Group, University of Michigan, Ann Arbor.
- Model type: Self-supervised pre-trained MIST encoder with supervised finetuning.
- Finetuned from model:
mist-28M-ti624ev1
Model Sources
Getting Started
Setting Up Your Environment
Create a virtual environment and install dependencies:
1python -m venv .venv
2source .venv/bin/activate # On Windows: .venv\Scripts\activate
3pip install -r requirements.txt
Note: SMIRK tokenizers require Rust to be installed. See the
Rust installation guide for details.
Property Prediction
1from transformers import AutoModel
2from smirk import SmirkTokenizerFast
3# Load the model
4model = AutoModel.from_pretrained(
5 "path/to/model",
6 trust_remote_code=True
7)
8
9# Make predictions for normalized empirical solvent polarity parameter E_T^N
10smiles_batch = [
11 "CCO", # Ethanol
12 "CC(=O)O", # Acetic acid
13 "C1=CC=CC=C1" # Benzene
14]
15# Returns predictions for the normalized Dimroth-Reichardt solvent polarity parameter ETN
16results = model.predict(smiles_batch)
Use and Restrictions
Model weights are provided as-is for research purposes only, without guarantees of correctness, fitness for purpose, or warranties of any kind.
- Research use only
- No redistribution without permission
- No commercial use without licensing agreement
Training Details
Training Data
Pretraining
We use the the Enamine REAL Space dataset to pretrain MIST models.
At time of writing, Enamine REAL Space is the largest database of commercially available compounds.
The dataset was constructed using forward synthetic analysis: experimentally validated building blocks were converted into synthons annotated with reactivity features.
Enamine REAL Space was selected as the pretraining dataset since it was the largest database of molecular SMILES at the time of training, it is easily accessible for academic use and molecules relevant to downstream tasks, such as drug candidates, electrolytes, fragrances, live in synthetically accessible regions of chemical space.
Finetuning
Training Procedure
Inputs
- Inputs: The input to MIST models are SMILES strings for molecules.
Unless specified otherwise, models were pretrained and fine-tuned on Kekulized SMILES strings.
- Outputs: Regression prediction for ETN
Evaluation
Testing
Testing Data
Dataset was split 80/10/10 using a random split.
Metrics
Mean Absolute Error (MAE) for ETN predictions
Technical Specifications
Model Architecture and Objective
- Encoder:
RoBERTa-PreLayerNorm encoder with 8 layers, a hidden size of 512, intermediate size of 2048, 8 attention heads and maximum sequence length of 2048.
- Task Network: Two-layer MLP (Multi-layer perceptron)
- Objective
- Pretraining: MLM (Masked Language Modeling)
- Fine-tuning: Regression
- Loss:
- Pretraining: Cross-Entropy Loss
- Fine-tuning: Mean Squared Error (MSE) Loss
- Optimizer:
- Pretraining:
deepspeed.ops.lamb.FusedLAMB
- Fine-tuning:
torch.optim.AdamW
Compute Infrastructure
Hardware
This model was pre-trained on 2 NVIDIA A100-SXM4-80GB GPUs in 12 hours 15 minutes.
It was finetuned on 1 NVIDIA A100 GPU.
Software
This model was trained with PyTorchLightning using the DeepSpeed strategy for data distributed parallelism.
Model are exported in a Safetensors format.
Citation
If you use this model in your research, please cite:
1@online{MIST,
2 title = {Foundation Models for Discovery and Exploration in Chemical Space},
3 author = {Wadell, Alexius and Bhutani, Anoushka and Azumah, Victor and Ellis-Mohr, Austin R. and Kelly, Celia and Zhao, Hancheng and Nayak, Anuj K. and Hegazy, Kareem and Brace, Alexander and Lin, Hongyi and Emani, Murali and Vishwanath, Venkatram and Gering, Kevin and Alkan, Melisa and Gibbs, Tom and Wells, Jack and Varshney, Lav R. and Ramsundar, Bharath and Duraisamy, Karthik and Mahoney, Michael W. and Ramanathan, Arvind and Viswanathan, Venkatasubramanian},
4 date = {2025-10-20},
5 eprint = {2510.18900},
6 eprinttype = {arXiv},
7 eprintclass = {physics},
8 doi = {10.48550/arXiv.2510.18900},
9 url = {http://arxiv.org/abs/2510.18900},
10}
Model Card Authors
Anoushka Bhutani, Alexius Wadell
Model Card Contact
For questions, issues, or licensing inquiries, please contact Venkat Viswanathan
venkvis@umich.edu.
