NanoPhotoNet-MPM Model Checkpoints
Copyright © 2026 Islam I. Abdulaal and Omar A. M. Abdelraouf. Model checkpoints are released under CC BY 4.0. Paper: forthcoming.
This repository contains inference-ready NanoPhotoNet-MPM checkpoints for a physics-informed neural surrogate of modal phase-matched SPDC inverse design in monoclinic NbOCl2 ridge waveguides.
The paired dataset release is available at DOI 10.57967/hf/9066. The archived code release is available at DOI 10.5281/zenodo.18653064.
Files
| File | Description |
|---|---|
checkpoints/transverse_solver.safetensors |
EigenmodeDeepONet transverse solver weights |
checkpoints/propagator.safetensors |
CWE-PINN longitudinal propagator weights |
checkpoints/unified_model.safetensors |
Combined training-time model state dictionary |
inference.py |
Standalone forward-inference script |
requirements.txt |
Minimal inference dependencies |
The release uses safetensors checkpoint files rather than TorchScript or pickle-based .pt files. This keeps the distributed model weights tensor-only and avoids the automated "suspicious file" warning that applies to serialized PyTorch executable artifacts.
Quick Start
pip install -r requirements.txt
python inference.py --width 693.49 --height 200.0 --length 5.0
The script reconstructs the transverse solver and longitudinal propagator architectures locally, loads the tensor-only checkpoint files, and prints effective indices, overlap proxy, phase mismatch, and output envelopes for the requested geometry.
Inputs
| Argument | Default | Meaning |
|---|---|---|
--width |
693.49 |
Waveguide width in nm |
--height |
200.0 |
Waveguide height in nm |
--length |
5.0 |
Interaction length in mm |
--transverse-checkpoint |
checkpoints/transverse_solver.safetensors |
Transverse solver checkpoint |
--propagator-checkpoint |
checkpoints/propagator.safetensors |
Longitudinal propagator checkpoint |
Verification Results
| Metric | Value | Notes |
|---|---|---|
| Effective-index MAE | 1.43e-3 | Across 140 paired target-mode samples |
| Zero-mismatch design point | width = 693.49 nm, height = 200.00 nm | Continuous model phase-matching zero |
| Width tolerance FWHM | 0.191 nm | At interaction length L = 5 mm |
| Pump-wavelength FWHM | 22.1 nm | Acceptance bandwidth |
| Nonlinear overlap proxy | K = 0.385 | d22-weighted spatial overlap |
| Field reconstruction overlap | Pump: 0.839, signal: 0.978 | Signed-vector overlap vs FDE |
| Stage 2 validation loss | RK4: 9.37e-10, CWE: 8.30e-8 | Reference solver and PDE residuals |
| HOM purity ceiling | P = 0.1186 | Degenerate purity without filtering |
Limitations
The checkpoints are intended for the geometry, material, wavelength, and modal design space represented by the paired dataset. Designs outside that range should be revalidated with an independent electromagnetic and coupled-wave solver.
Citation
@misc{abdulaal2026nanophotonet_model,
author = {Islam I. Abdulaal},
title = {{NanoPhotoNet-MPM} Model Checkpoints},
year = {2026},
publisher = {Hugging Face},
doi = {10.57967/hf/9068},
url = {https://doi.org/10.57967/hf/9068}
}