Comparison of machine learning approaches for radioisotope identification using NaI(TI) gamma-ray spectrum.

Journal: Applied radiation and isotopes : including data, instrumentation and methods for use in agriculture, industry and medicine
PMID:

Abstract

This research aims at comparing the performance of different machine learning algorithms used for NaI(TI) gamma-ray detector based radioisotope identification. Six machine learning algorithms were implemented, including support vector machine (SVM), k-nearest neighbor (KNN), logistic regression (LR), naive Bayes (NB), decision tree (DT), and multilayer perceptron (MLP). The hyper-parameters of each model were elaborately optimized. The effects of data size, statistical fluctuation, and spectrum drift were considered. Results show that for smaller data size (5 types of radioisotopes and 6000 spectra), the support vector machine and the logistic regression classifier exhibit higher identification accuracy with less training/predicting time. Whereas for larger data size (14 types of radioisotopes corresponding to the standard IEC 62327-2017), the multilayer perceptron showed highest accuracy but required the longest time for model training. The naive Bayes classifier and the decision tree were prone to make mistakes when fluctuations and distortions were added to the spectra. The k-nearest neighbor classifier, though showing high accuracy for most data sets, consumed the longest time while making prediction.

Authors

  • Sheng Qi
    State Key Laboratory of NBC Protection for Civilian, Beijing, China.
  • Wei Zhao
    Key Laboratory of Synthetic and Biological Colloids, Ministry of Education, Jiangnan University, Wuxi 214122, Jiangsu Province, P. R. China. lxy@jiangnan.edu.cn zhuye@jiangnan.edu.cn.
  • Ye Chen
    1 Department of Urology, First Affiliated Hospital of Soochow University, Suzhou, China.
  • Wenzhuo Chen
    Department of Electronics and Information Engineering, North China Institute of Science and Technology, Langfang 065201, China.
  • Jinglun Li
    Basic Research and Development Department of Xi'an FiberHome software and technology CO., LTD, Xi'an City, China.
  • Hui Zhao
    School of Mathematics and Computer Science, Shaanxi University of Technology, Hanzhong, 723000, Shaanxi, China.
  • Wuyun Xiao
    State Key Laboratory of NBC Protection for Civilian, Beijing, China.
  • Xianyun Ai
    State Key Laboratory of NBC Protection for Civilian, Beijing, China.
  • Kun Zhang
    Philosophy Department, Carnegie Mellon University, Pittsburgh, Pennsylvania, United States of America.
  • Shanqiang Wang
    State Key Laboratory of NBC Protection for Civilian, Beijing, China. Electronic address: wsq_tj1000@163.com.