Abstract
Data drift is a change in the distribution of data during machine learning model training and during operation. This occurs regardless of the type of data and adversely affects model performance. However, this method can only analyze the drift detection results from the difference in the distribution of the output of a two-sample test, and does not analyze the actual degradation of the prediction performance of the machine learning model due to drift. In addition to class probability, we believe that detecting drift for changes in the local region that the model is actually gazing at will improve accuracy. In this study, we propose a drift detection method based on the Attention Branch Network (ABN), which enables visualization of the basis of judgment in image classification. In our method, drift is detected using the class probabilities output by the attention branch and perception branch, which constitute the ABN, and the attention map. The results show that the detection rate can be improved by introducing an attention map to drift detection in addition to class probability. We also observed that the attention map tended to shrink with drift.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Dunn, O.J.: Multiple comparisons among means. J. Am. Stat. Assoc. 56(293), 52–64 (1961)
Fukui, H., Hirakawa, T., Yamashita, T., Fujiyoshi, H.: Attention branch network: Learning of attention mechanism for visual explanation. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) (2019)
Gretton, A., Borgwardt, K.M., Rasch, M.J., Schölkopf, B., Smola, A.: A kernel two-sample test. J. Mach. Learn. Res. 13(25), 723–773 (2012)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) (2016)
Hendrycks, D., Dietterich, T.: Benchmarking neural network robustness to common corruptions and perturbations. In: International Conference on Learning Representations (ICLR) (2019)
Krizhevsky, A., Hinton, G., et al.: Learning multiple layers of features from tiny images. Tech. reort, University of Toronto (2009)
LeCun, Y., Bottou, L., Bengio, Y., Haffner, P.: Gradient-based learning applied to document recognition. Proc. IEEE 86(11), 2278–2324 (1998)
Lipton, Z., Wang, Y.X., Smola, A.: Detecting and correcting for label shift with black box predictors. In: Proceedings of the International Conference on Machine Learning (ICML), pp. 3122–3130 (2018)
Rabanser, S., Günnemann, S., Lipton, Z.: Failing loudly: an empirical study of methods for detecting dataset shift. In: Advances in Neural Information Processing Systems, vol. 32 (2019)
Selvaraju, R.R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., Batra, D.: Grad-cam: visual explanations from deep networks via gradient-based localization. In: Proceedings of the IEEE International Conference on Computer Vision (ICCV) (2017)
Zhang, K., Schölkopf, B., Muandet, K., Wang, Z.: Domain adaptation under target and conditional shift. In: Proceedings of the International Conference on Machine Learning (ICML), pp. 819–827 (2013)
Zhao, S., et al.: A review of single-source deep unsupervised visual domain adaptation. IEEE Trans. Neural Networks Learn. Syst. 33(2), 473–493 (2022)
Zhou, B., Khosla, A., Lapedriza, A., Oliva, A., Torralba, A.: Learning deep features for discriminative localization. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) (2016)
Zügner, D., Akbarnejad, A., Günnemann, S.: Adversarial attacks on neural networks for graph data. In: Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (ICKD) (2018)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Nitta, T., Shi, Y., Hirakawa, T., Yamashita, T., Fujiyoshi, H. (2023). Detecting Data Drift with KS Test Using Attention Map. In: Lu, H., Blumenstein, M., Cho, SB., Liu, CL., Yagi, Y., Kamiya, T. (eds) Pattern Recognition. ACPR 2023. Lecture Notes in Computer Science, vol 14406. Springer, Cham. https://doi.org/10.1007/978-3-031-47634-1_6
Download citation
DOI: https://doi.org/10.1007/978-3-031-47634-1_6
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-47633-4
Online ISBN: 978-3-031-47634-1
eBook Packages: Computer ScienceComputer Science (R0)