Unsupervised domain adaptation for radioisotope identification in gamma spectroscopy
Organizations: Pacific Northwest National Laboratory, Richland, WA 99352 USA · University of Washington, Seattle, WA 98195 USA
Abstract
Training machine learning models for radioisotope identification using gamma spectroscopy remains an elusive challenge for many practical applications, largely stemming from the difficulty of acquiring and labeling large, diverse experimental datasets. Simulations can mitigate this challenge, but the accuracy of models trained on simulated data can deteriorate substantially when deployed to an out-of-distribution operational environment. In this study, we demonstrate that unsupervised domain adaptation (UDA) can improve the ability of a model trained on synthetic data to generalize to a new testing domain, provided unlabeled data from the target domain is available. Conventional supervised techniques are unable to utilize this data because the absence of isotope labels precludes defining a supervised classification loss. We compare a range of different UDA techniques, finding that feature alignment strategies, particularly via maximum mean discrepancy (MMD) minimization or domain-adversarial training, yield the most consistent improvement to testing scores. For instance, using a custom transformer-based neural network, we achieve a testing accuracy of on an experimental LaBr test set after performing unsupervised feature alignment via MMD minimization, compared to before alignment. Overall, our results highlight the potential of using UDA to adapt a radioisotope classifier trained on synthetic data for real-world deployment.
Figures & tables
| Scenario | Domain | Acquisition Mode | Isotopic Composition | Number of Isotopes | Size |
| sim-to-sim (HPGe) | Source | GADRAS | Mixed | 55 | |
| Target | Geant4 | Mixed | 55 | ||
| sim-to-real (LaBr 3 ) | Source | GADRAS | Single | 32 | |
| Target | Experiment | Single | 32 | 15,091 | |
| sim-to-real (NaI(Tl)) | Source | GADRAS | Single | 32 | |
| Target | Experiment | Single | 32 | 10,440 |
| Scenario | AUROC (features) | AUROC (inputs) | MMD 2 | |
| sim-to-sim (HPGe) | [0.993, 0.996] | [0.957, 0.967] | [2.74, 2.93] | [1.36, 1.40] |
| sim-to-real (LaBr 3 ) | [0.982, 0.988] | [0.852, 0.874] | [2.26, 2.55] | [1.87, 1.93] |
| sim-to-real (NaI(Tl)) | [0.983, 0.989] | [0.780, 0.808] | [1.30, 2.53] | [1.93, 1.99] |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Architecture | Parameter | Search Space | Best run |
| MLP | Num Dense Layers | 2 | |
| Dense1 Hidden Units | 8192 | ||
| Dense2 Hidden Units | 4096 | ||
| CNN | Num Convolutional Layers | 1 | |
| Conv Filters | 16 | ||
| Conv Kernel Size | 5 |
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | Learning rate | 4.60e-5 | 1.80e-3 | 1.32e-3 | |
| Batch size | 256 | 512 | 256 | ||
| Weight decay | 1.54e-5 | 1.96e-6 | 1.21e-4 | ||
| Dropout | 0.230 | 0.394 | 0.224 | ||
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | FE learning rate † | 5.87e-7 | 1.67e-4 | 2.18e-4 | |
| DD learning rate † | 9.29e-6 | 3.87e-6 | 5.03e-6 | ||
| Batch size | 128 | 128 | 64 | ||
| Weight decay | 6.89e-6 | 2.35e-6 | 8.33e-4 | ||
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | Learning rate | 5.80e-6 | 9.72e-4 | 6.84e-4 | |
| Batch size | 256 | 256 | 128 | ||
| Weight decay | 1.92e-5 | 8.15e-5 | 6.24e-6 | ||
| Dropout | 0.0278 | 1.10e-3 | 3.71e-3 | ||
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | FE learning rate † | 1.31e-6 | 1.05e-3 | 3.69e-4 | |
| DD learning rate † | 1.06e-5 | 1.52e-4 | 1.17e-4 | ||
| Batch size | 256 | 512 | 512 | ||
| Weight decay | 1.26e-7 | 2.06e-4 | 1.12e-6 | ||
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | Learning rate | 1.34e-4 | 8.95e-4 | 9.41e-4 | |
| Batch size | 512 | 512 | 128 | ||
| Weight decay | 1.37e-6 | 1.19e-6 | 6.11e-7 | ||
| Dropout | 0.297 | 8.24e-3 | 0.0839 | ||
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | Learning rate | 9.25e-4 | 7.15e-4 | 7.24e-4 | |
| Batch size | 512 | 512 | 256 | ||
| Weight decay | 8.33e-6 | 5.23e-6 | 9.17e-4 | ||
| Dropout | 0.207 | 0.177 | 0.216 | ||
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | Learning rate | 9.06e-4 | 5.36e-4 | 8.24e-4 | |
| Batch size | 512 | 64 | 64 | ||
| Weight decay | 7.45e-7 | 6.91e-6 | 6.83e-6 | ||
| Dropout | 0.0229 | 0.183 | 0.121 | ||
| Best run | |||||
| sim-to-sim | sim-to-real | sim-to-real | |||
| Architecture | Parameter | Search Space | HPGe | LaBr 3 | NaI(Tl) |
| MLP | Learning rate | 5.85e-4 | 3.21e-4 | 6.21e-4 | |
| Batch size | 128 | 128 | 128 | ||
| Weight decay | 2.00e-5 | 1.99e-4 | 5.92e-7 | ||
| Dropout | 0.124 | 0.101 | 0.311 | ||
Synthetic-to-Real Domain Adaptation
Synthetic-to-real domain adaptation transfers models trained on simulated or generated data to real-world settings by reducing distribution shifts.
See all 137 papers