coding103856/d4lens-weights

D4LensPINN Weights

This repository contains the trained checkpoints for the D4LensPINN gravitational lens classification model and its ResNet-18 baseline.

Checkpoints

• d4phase2best.pth: Best D4LensPINN checkpoint.
• resnet18baseline_best.pth: ResNet-18 baseline checkpoint.

Intended Use

These weights are provided for reproducibility and research use in mechanistic interpretability experiments on hybrid physics-ML architectures.

Notes

• Inputs are 150×150 grayscale images.
• D4LensPINN expects the full physics pipeline described in the paper.
• The checkpoint files are not meant to be used as standalone generic image classifiers without the corresponding model code.

Loading

Example:

import torch

ckpt = torch.load("d4phase2best.pth", map_location="cpu")

The exact model class definitions must match the training code used to create the checkpoints.

Reproducibility

If you use these weights, please cite the associated paper.

License

Apache-2.0 License.