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

Time-reversed Flow Matching with Worst Transport in High-dimensional Latent Space for Image Anomaly Detection

About

Likelihood-based deep generative models have been widely investigated for Image Anomaly Detection (IAD), particularly Normalizing Flows, yet their strict architectural invertibility needs often constrain scalability, particularly in large-scale data regimes. Although time-parameterized Flow Matching (FM) serves as a scalable alternative, it remains computationally challenging in IAD due to the prohibitive costs of Jacobian-trace estimation. This paper proposes time-reversed Flow Matching (rFM), which shifts the objective from exact likelihood computation to evaluating target-domain regularity through density proxy estimation. We uncover two fundamental theoretical bottlenecks in this paradigm: first, the reversed vector field exhibits a non-Lipschitz singularity at the initial temporal boundary, precipitating explosive estimation errors. Second, the concentration of measure in high-dimensional Gaussian manifolds induces structured irregularities, giving rise to a Centripetal Potential Field (CPF) that steers trajectories away from Optimal Transport (OT) paths. We identify these observations as the inherent dualities between FM and rFM. To address these issues, we introduce local Worst Transport Flow matching (WT-Flow), which amplifies the observed CPF of rFM to mitigate the initial singularity while circumventing the need for exact distribution transformations via density proxy. Experiments on five datasets demonstrate that WT-Flow achieves state-of-the-art performance among single-scale flow-based methods, and competitive performance against leading multi-scale approaches. Furthermore, the proposed framework enables superior one-step inference, achieving a per-image flow latency of only 6.7 ms. Our code is available on https://github.com/lil-wayne-0319/fmad.

Liangwei Li, Lin Liu, Hanzhe Liang, Juanxiu Liu, Jing Zhang, Ruqian Hao, Xiaohui Du, Yong Liu, Pan Li• 2025

Related benchmarks

TaskDatasetResultRank
Anomaly LocalizationMVTec AD--
513
Anomaly DetectionVisA
AUROC96.52
261
Anomaly DetectionMVTec LOCO
AUROC (Mean)87.7
30
Anomaly LocalizationMVTec3D
P-AUROC98.6
30
Anomaly DetectionMVTec AD
Mean AUROC99.05
26
Anomaly LocalizationVisA
AUROC98.9
23
Anomaly DetectionMPDD
Pixel AUROC94.2
18
Anomaly DetectionMVTec-3d
mAUC85.1
8
Anomaly LocalizationMPDD
mAUC98.3
8
Showing 9 of 9 rows

Other info

Follow for update