Published July 30, 2026 | Version v1
Model Open

CDI Reference Models for CBED

Authors/Creators

Description

# CDI Reference Models for CBED

This repository contains pretrained reference models for Conditional Diffusion Inference (CDI) experiments on the CBED dataset for magnesium diboride (MgB₂). The models are provided as tar archives containing complete CDI model directories with saved configurations, network weights, optimizer states, and training histories.

## Files

- `reference_diffusion_model.tar` — CDI conditional diffusion model for   uncertainty-aware CBED parameter inference.
- `reference_efficientnet_b2.tar` — deterministic EfficientNet-B2 regression baseline.
- `reference_resnet_50.tar` — deterministic ResNet-50 regression baseline.

## Archive Contents

Each archive unpacks to one model directory:

```text
reference_diffusion_model/
reference_efficientnet_b2/
reference_resnet_50/
```

The model directories include the saved training configuration (`config.json`), PyTorch model weights (`net_*.pth`), optimizer and scheduler states when available, and training/validation history files.

## Related Resources

- CDI code repository: https://github.com/realtime-intelligence/cdi-uq
- CBED-ML test split: https://zenodo.org/records/18340683
- CBED-ML training split, Part 1: https://zenodo.org/records/18344486
- CBED-ML training split, Part 2: https://zenodo.org/records/18356685
- CBED-ML training split, Part 3: https://zenodo.org/records/18476248

Files

Files (963.6 MB)

Name Size
md5:29b84b067688a92994be402a154d1a49
587.2 MB Download
md5:19a441ba911f68e6f29bcd63c4fc482c
93.5 MB Download
md5:8a5ffc859d21edd97f6d43c7107d5f73
282.9 MB Download