SwinMSI-weights

SwinMSI — Trained Model Weights

Trained checkpoints for the paper "SwinMSI: Window-Attention Super-Resolution Exploiting Long-Range Self-Similarity for Mass Spectrometry Imaging."

SwinMSI reconstructs a full high-resolution MSI dataset (all m/z channels) from a single fast, low-resolution acquisition, so that high spatial resolution can be traded for far shorter acquisition time (≈7–8× in our DESI experiments). Its Swin-Transformer (window-attention) backbone exploits the long-range self-similarity of MSI ion images to recover fine detail, operating on the MSI data alone (no co-registered optical modality).

• 💻 Code: https://github.com/zhihaozhao98/SwinMSI
• 📊 Dataset (self-acquired DESI-MSI): https://huggingface.co/datasets/ZhihaoZhao/MSI-SuperRes-DESI

Files

The interpolation baselines (bilinear, bicubic) are parameter-free and require no weights. MOSR is an external, pretrained model released by its original authors and is not redistributed here (see the MOSR paper and its official release).

Usage

Download a checkpoint and run 4× super-resolution on a low-resolution .imzML file:

from huggingface_hub import hf_hub_download

weights = hf_hub_download("ZhihaoZhao/SwinMSI-weights", "swinmsi_best.pth")

# from the code repository
python evaluation/apply_superres_unet.py \
    --input  LR.imzML \
    --output SR.imzML \
    --model  swinmsi_best.pth \
    --device cuda

A one-click Colab demo that downloads a sample dataset and a checkpoint is provided in the code repository at examples/quickstart_colab.ipynb.

Training

All models were trained from scratch on 70 public METASPACE MSI datasets (60 rodent-brain and 10 kidney sections), with low-resolution inputs synthesized by Gaussian point-spread-function blur followed by 4× bicubic downsampling. Training used PyTorch with BF16 mixed precision on a single NVIDIA RTX 4090. See the code repository for the full pipeline and configuration.

Citation

@article{zhao2026swinmsi,
  title   = {SwinMSI: Window-Attention Super-Resolution Exploiting Long-Range Self-Similarity for Mass Spectrometry Imaging},
  author  = {Zhao, Zhihao and Lv, Yuanxia and Zhuang, Yaoyu and Zhang, Zhimin and Wu, Qian and Lu, Hongmei},
  year    = {2026},
  note    = {Manuscript under review}
}

License

These weights are released under the Creative Commons Attribution 4.0 International (CC-BY-4.0) license, matching the accompanying dataset.