Siddiquee Md Mahfuzur Rahman, Shah Jay, Wu Teresa, Chong Catherine, Schwedt Todd J, Dumkrieger Gina, Nikolova Simona, Li Baoxin
Arizona State University.
ASU-Mayo Center for Innovative Imaging.
IEEE Winter Conf Appl Comput Vis. 2024 Jan;2024:7558-7567. doi: 10.1109/wacv57701.2024.00740. Epub 2024 Apr 9.
Harnessing the power of deep neural networks in the medical imaging domain is challenging due to the difficulties in acquiring large annotated datasets, especially for rare diseases, which involve high costs, time, and effort for annotation. Unsupervised disease detection methods, such as anomaly detection, can significantly reduce human effort in these scenarios. While anomaly detection typically focuses on learning from images of healthy subjects only, real-world situations often present unannotated datasets with a mixture of healthy and diseased subjects. Recent studies have demonstrated that utilizing such unannotated images can improve unsupervised disease and anomaly detection. However, these methods do not utilize knowledge specific to registered neuroimages, resulting in a subpar performance in neurologic disease detection. To address this limitation, we propose Brainomaly, a GAN-based image-to-image translation method specifically designed for neurologic disease detection. Brainomaly not only offers tailored image-to-image translation suitable for neuroimages but also leverages unannotated mixed images to achieve superior neurologic disease detection. Additionally, we address the issue of model selection for inference without annotated samples by proposing a pseudo-AUC metric, further enhancing Brainomaly's detection performance. Extensive experiments and ablation studies demonstrate that Brainomaly outperforms existing state-of-the-art unsupervised disease and anomaly detection methods by significant margins in Alzheimer's disease detection using a publicly available dataset and headache detection using an institutional dataset. The code is available from https://github.com/mahfuzmohammad/Brainomaly.
由于获取大量带注释数据集存在困难,尤其是对于罕见疾病而言,这涉及高昂的注释成本、时间和精力,因此在医学成像领域利用深度神经网络的能力具有挑战性。无监督疾病检测方法,如异常检测,在这些情况下可以显著减少人力。虽然异常检测通常只专注于从健康受试者的图像中学习,但现实世界的情况往往是呈现出包含健康和患病受试者的未注释数据集。最近的研究表明,利用此类未注释图像可以改善无监督疾病和异常检测。然而,这些方法没有利用注册神经图像的特定知识,导致在神经疾病检测中性能不佳。为了解决这一局限性,我们提出了Brainomaly,一种基于生成对抗网络(GAN)的图像到图像转换方法,专门设计用于神经疾病检测。Brainomaly不仅提供适合神经图像的定制图像到图像转换,还利用未注释的混合图像实现卓越的神经疾病检测。此外,我们通过提出一种伪AUC指标来解决无注释样本情况下推理的模型选择问题,进一步提高Brainomaly的检测性能。大量实验和消融研究表明,在使用公开可用数据集进行阿尔茨海默病检测以及使用机构数据集进行头痛检测时,Brainomaly在很大程度上优于现有的无监督疾病和异常检测方法。代码可从https://github.com/mahfuzmohammad/Brainomaly获取。