Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

UniMMAD: Unified Multi-Modal and Multi-Class Anomaly Detection via MoE-Driven Feature Decompression

About

Existing anomaly detection (AD) methods often treat the modality and class as independent factors. Although this paradigm has enriched the development of AD research branches and produced many specialized models, it has also led to fragmented solutions and excessive memory overhead. Moreover, reconstruction-based multi-class approaches typically rely on shared decoding paths, which struggle to handle large variations across domains, resulting in distorted normality boundaries, domain interference, and high false alarm rates. To address these limitations, we propose UniMMAD, a unified framework for multi-modal and multi-class anomaly detection. At the core of UniMMAD is a Mixture-of-Experts (MoE)-driven feature decompression mechanism, which enables adaptive and disentangled reconstruction tailored to specific domains. This process is guided by a ``general to specific'' paradigm. In the encoding stage, multi-modal inputs of varying combinations are compressed into compact, general-purpose features. The encoder incorporates a feature compression module to suppress latent anomalies, encourage cross-modal interaction, and avoid shortcut learning. In the decoding stage, the general features are decompressed into modality-specific and class-specific forms via a sparsely-gated cross MoE, which dynamically selects expert pathways based on input modality and class. To further improve efficiency, we design a grouped dynamic filtering mechanism and a MoE-in-MoE structure, reducing parameter usage by 75\% while maintaining sparse activation and fast inference. UniMMAD achieves state-of-the-art performance on 9 anomaly detection datasets, spanning 3 fields, 12 modalities, and 66 classes. The source code will be available at https://github.com/yuanzhao-CVLAB/UniMMAD.

Yuan Zhao, Youwei Pang, Lihe Zhang, Hanqi Liu, Jiaming Zuo, Huchuan Lu, Xiaoqi Zhao• 2025

Related benchmarks

TaskDatasetResultRank
Anomaly DetectionVisA--
101
Anomaly DetectionBraTS
Image-level AUROC95.838
93
Anomaly DetectionMVTec AD
Image AUROC99.4
92
Anomaly DetectionUniMed
AUCP96.335
20
Anomaly DetectionEyecandies
AUCP96.913
12
Anomaly Detection and LocalizationMVTec-AD 15 classes 1.0 (test)
Image-level AUROC99.4
12
Anomaly Detection and LocalizationVisA 12 classes 1.0 (test)
Image-level AUROC95
12
Anomaly Detection and LocalizationMVTec-3D 10 classes 1.0 (test)
Image-level AUROC87
12
Anomaly Detection and LocalizationUni-Medical 1.0 (test)
Image-level AUROC84
12
Anomaly Detection and LocalizationReal-IAD 30 classes 1.0 (test)
Image-level AUROC86
12
Showing 10 of 15 rows

Other info

Follow for update