Kullback-Leibler Divergence-Based Out-of-Distribution Detection With Flow-Based Generative Models

被引:3
作者
Zhang, Yufeng [1 ]
Pan, Jialu [1 ]
Liu, Wanwei [2 ]
Chen, Zhenbang [2 ]
Li, Kenli [1 ]
Wang, Ji [2 ]
Liu, Zhiming [3 ]
Wei, Hongmei [4 ]
机构
[1] Hunan Univ, Coll Comp Sci & Elect Engn, Changsha 410012, Hunan, Peoples R China
[2] Natl Univ Def Technol, Coll Comp, Changsha 410073, Hunan, Peoples R China
[3] Southwest Univ, Ctr Res & Innovat Software Engn, Chongqing 400715, Peoples R China
[4] Natl Res Ctr Parallel Comp Engn & Technol, Wuxi 214071, Jiangsu, Peoples R China
基金
中国国家自然科学基金;
关键词
Data models; Anomaly detection; Gaussian distribution; Standards; Surveys; Robustness; Computational modeling; Out-of-distribution detection; deep learning; flow-based model; Kullback-Leibler divergence;
D O I
10.1109/TKDE.2023.3309853
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recent research has revealed that deep generative models including flow-based models and Variational Autoencoders may assign higher likelihoods to out-of-distribution (OOD) data than in-distribution (ID) data. However, we cannot sample OOD data from the model. This counterintuitive phenomenon has not been satisfactorily explained and brings obstacles to OOD detection with flow-based models. In this article, we prove theorems to investigate the Kullback-Leibler divergence in flow-based model and give two explanations for the above phenomenon. Based on our theoretical analysis, we propose a new method KLODS to leverage KL divergence and local pixel dependence of representations to perform anomaly detection. Experimental results on prevalent benchmarks demonstrate the effectiveness and robustness of our method. For group anomaly detection, our method achieves 98.1% AUROC on average with a small batch size of 5. On the contrary, the baseline typicality test-based method only achieves 64.6% AUROC on average due to its failure on challenging problems. Our method also outperforms the state-of-the-art method by 9.1% AUROC. For point-wise anomaly detection, our method achieves 90.7% AUROC on average and outperforms the baseline by 5.2% AUROC. Besides, our method has the least notable failures and is the most robust one.
引用
收藏
页码:1683 / 1697
页数:15
相关论文
共 73 条
[1]  
Altekrüger F, 2024, Arxiv, DOI arXiv:2303.15845
[2]  
An J., 2015, SPECIAL LECT IE, V2, P1, DOI DOI 10.1007/BF00758335
[3]  
Bishop Christopher M., 2006, Pattern recognition and machine learning
[4]  
BUCKLAND M, 1994, J AM SOC INFORM SCI, V45, P12, DOI 10.1002/(SICI)1097-4571(199401)45:1<12::AID-ASI2>3.0.CO
[5]  
2-L
[6]  
Chalapathy R, 2019, Arxiv, DOI [arXiv:1901.03407, DOI 10.48550/ARXIV.1901.03407]
[7]   Anomaly Detection: A Survey [J].
Chandola, Varun ;
Banerjee, Arindam ;
Kumar, Vipin .
ACM COMPUTING SURVEYS, 2009, 41 (03)
[8]  
Chen Ricky T. Q., 2019, Advances in Neural Information Processing Systems, V32
[9]  
Chen Ricky T. Q., 2018, Advances in Neural Information Processing Systems, V31
[10]  
Choi H, 2019, Arxiv, DOI arXiv:1810.01392