Residential College | false |
Status | 已發表Published |
Uncertainty-aware prototypical learning for anomaly detection in medical images | |
Huang, Chao1,2; Shi, Yushu2; Zhang, Bob1; Lyu, Ke3,4 | |
2024-07 | |
Source Publication | Neural Networks |
ISSN | 0893-6080 |
Volume | 175Pages:106284 |
Abstract | Anomalous object detection (AOD) in medical images aims to recognize the anomalous lesions, and is crucial for early clinical diagnosis of various cancers. However, it is a difficult task because of two reasons: (1) the diversity of the anomalous lesions and (2) the ambiguity of the boundary between anomalous lesions and their normal surroundings. Unlike existing single-modality AOD models based on deterministic mapping, we constructed a probabilistic and deterministic AOD model. Specifically, we designed an uncertainty-aware prototype learning framework, which considers the diversity and ambiguity of anomalous lesions. A prototypical learning transformer (Pformer) is established to extract and store the prototype features of different anomalous lesions. Moreover, Bayesian neural uncertainty quantizer, a probabilistic model, is designed to model the distributions over the outputs of the model to measure the uncertainty of the model's detection results for each pixel. Essentially, the uncertainty of the model's anomaly detection result for a pixel can reflect the anomalous ambiguity of this pixel. Furthermore, an uncertainty-guided reasoning transformer (Uformer) is devised to employ the anomalous ambiguity, encouraging the proposed model to focus on pixels with high uncertainty. Notably, prototypical representations stored in Pformer are also utilized in anomaly reasoning that enables the model to perceive diversities of the anomalous objects. Extensive experiments on five benchmark datasets demonstrate the superiority of our proposed method. The source code will be available in github.com/umchaohuang/UPformer. |
Keyword | Anomalous Object Detection Medical Image Analysis Prototypical Learning |
DOI | 10.1016/j.neunet.2024.106284 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science ; Neurosciences & Neurology |
WOS Subject | Computer Science, Artificial Intelligence ; Neurosciences |
WOS ID | WOS:001225493700001 |
Publisher | PERGAMON-ELSEVIER SCIENCE LTD, THE BOULEVARD, LANGFORD LANE, KIDLINGTON, OXFORD OX5 1GB, ENGLAND |
Scopus ID | 2-s2.0-85189749984 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Zhang, Bob |
Affiliation | 1.PAMI Research Group, Department of Computer and Information Science, University of Macau, Taipa, 519000, Macao Special Administrative Region of China 2.Shenzhen Campus of Sun Yat-sen University, School of Cyber Science and Technology, Shenzhen, 518107, China 3.School of Engineering Sciences, University of the Chinese Academy of Sciences, Beijing, 100049, China 4.Pengcheng Laboratory, Shenzhen, 518055, China |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Huang, Chao,Shi, Yushu,Zhang, Bob,et al. Uncertainty-aware prototypical learning for anomaly detection in medical images[J]. Neural Networks, 2024, 175, 106284. |
APA | Huang, Chao., Shi, Yushu., Zhang, Bob., & Lyu, Ke (2024). Uncertainty-aware prototypical learning for anomaly detection in medical images. Neural Networks, 175, 106284. |
MLA | Huang, Chao,et al."Uncertainty-aware prototypical learning for anomaly detection in medical images".Neural Networks 175(2024):106284. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment