Rethinking Autoencoders for Medical Anomaly Detection from A Theoretical Perspective

被引:3
作者
Cai, Yu [1 ]
Chen, Hao [1 ]
Cheng, Kwang-Ting [1 ]
机构
[1] Hong Kong Univ Sci & Technol, Hong Kong, Peoples R China
来源
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT XI | 2024年 / 15011卷
关键词
Anomaly detection; Autoencoders; Information theory;
D O I
10.1007/978-3-031-72120-5_51
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Medical anomaly detection aims to identify abnormal findings using only normal training data, playing a crucial role in health screening and recognizing rare diseases. Reconstruction-based methods, particularly those utilizing autoencoders (AEs), are dominant in this field. They work under the assumption that AEs trained on only normal data cannot reconstruct unseen abnormal regions well, thereby enabling the anomaly detection based on reconstruction errors. However, this assumption does not always hold due to the mismatch between the reconstruction training objective and the anomaly detection task objective, rendering these methods theoretically unsound. This study focuses on providing a theoretical foundation for AE-based reconstruction methods in anomaly detection. By leveraging information theory, we elucidate the principles of these methods and reveal that the key to improving AE in anomaly detection lies in minimizing the information entropy of latent vectors. Experiments on four datasets with two image modalities validate the effectiveness of our theory. To the best of our knowledge, this is the first effort to theoretically clarify the principles and design philosophy of AE for anomaly detection. The code is available at https://github.com/caiyu6666/AE4AD.
引用
收藏
页码:544 / 554
页数:11
相关论文
共 24 条
[1]   GANomaly: Semi-supervised Anomaly Detection via Adversarial Training [J].
Akcay, Samet ;
Atapour-Abarghouei, Amir ;
Breckon, Toby P. .
COMPUTER VISION - ACCV 2018, PT III, 2019, 11363 :622-637
[2]  
Baid U, 2021, Arxiv, DOI arXiv:2107.02314
[3]   VoxelMorph: A Learning Framework for Deformable Medical Image Registration [J].
Balakrishnan, Guha ;
Zhao, Amy ;
Sabuncu, Mert R. ;
Guttag, John ;
Dalca, Adrian, V .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (08) :1788-1800
[4]   Autoencoders for unsupervised anomaly segmentation in brain MR images: A comparative study [J].
Baur, Christoph ;
Denner, Stefan ;
Wiestler, Benedikt ;
Navab, Nassir ;
Albarqouni, Shadi .
MEDICAL IMAGE ANALYSIS, 2021, 69
[5]   Deep Autoencoding Models for Unsupervised Anomaly Segmentation in Brain MR Images [J].
Baur, Christoph ;
Wiestler, Benedikt ;
Albarqouni, Shadi ;
Navab, Nassir .
BRAINLESION: GLIOMA, MULTIPLE SCLEROSIS, STROKE AND TRAUMATIC BRAIN INJURIES, BRAINLES 2018, PT I, 2019, 11383 :161-169
[6]   What Do AEs Learn? Challenging Common Assumptions in Unsupervised Anomaly Detection [J].
Bercea, Cosmin I. ;
Rueckert, Daniel ;
Schnabel, Julia A. .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT V, 2023, 14224 :304-314
[7]   Dual-distribution discrepancy with self-supervised refinement for anomaly detection in medical images [J].
Cai, Yu ;
Chen, Hao ;
Yang, Xin ;
Zhou, Yu ;
Cheng, Kwang-Ting .
MEDICAL IMAGE ANALYSIS, 2023, 86
[8]   Dual-Distribution Discrepancy for Anomaly Detection in Chest X-Rays [J].
Cai, Yu ;
Chen, Hao ;
Yang, Xin ;
Zhou, Yu ;
Cheng, Kwang-Ting .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT III, 2022, 13433 :584-593
[9]  
Chen XR, 2018, Arxiv, DOI arXiv:1806.04972
[10]   Unsupervised lesion detection via image restoration with a normative prior [J].
Chen, Xiaoran ;
You, Suhang ;
Tezcan, Kerem Can ;
Konukoglu, Ender .
MEDICAL IMAGE ANALYSIS, 2020, 64