用无监督域适应让仿真数据训练的伽马识别模型更好用在真实场景。
Unsupervised domain adaptation for radioisotope identification in gamma spectroscopy
- 用无标签真实数据通过特征对齐改进仿真模型
- 准确率从75.4%提升至90.4%
- 适合做核辐射检测的工程师和研究人员
基于伽马能谱的放射性核素识别,因难以获取和标注大量多样化的实验数据,仍是实际应用中的难题。仿真数据可缓解此问题,但其训练模型在分布外的真实环境中性能显著下降。本研究证明,若目标域存在无标签数据,无监督域适应(UDA)可有效提升仿真训练模型在新环境中的泛化能力。传统监督方法无法利用此类数据,因缺乏同位素标签无法定义分类损失。我们对比多种UDA方法,发现特征对齐策略,特别是最大均值差异(MMD)最小化或域对抗训练,表现最稳定。采用自定义的Transformer神经网络,在实验用的LaBr₃测试集上,经MMD最小化后测试准确率达0.904±0.022,较对齐前的0.754±0.014有明显提升。结果表明,利用UDA可使仿真训练的核素识别器适应真实部署场景。
原文摘要 · Abstract (English)
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 $0.904 \pm 0.022$ on an experimental LaBr$_3$ test set after performing unsupervised feature alignment via MMD minimization, compared to $0.754 \pm 0.014$ before alignment. Overall, our results highlight the potential of using UDA to adapt a radioisotope classifier trained on synthetic data for real-world deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。