arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.12959cs.CV
收藏

Puzzle-AE:通过求解Puzzle来检测图像中的新颖性

Puzzle-AE: Novelty Detection in Images through Solving Puzzles

Sharif University of Technology(谢里夫理工大学)

Mohammadreza Salehi, Ainaz Eftekhar, Niousha Sadjadi, Mohammad Hossein Rohban, Hamid R. Rabiee

摘要:

自动编码器作为许多异常检测方法的重要组成部分,对复杂数据集的正常数据缺乏灵活性。U-Net被证明是有效的,但过度拟合的训练数据,如果只使用重建误差类似于其他基于AE的框架训练。作为自监督学习(SSL)方法的一个前提任务,谜题解决已经证明了它在学习语义有意义的特征方面的能力。我们表明,基于此任务训练U网是一种有效的补救措施,可以防止过拟合,并促进像素级特征之外的学习。然而,SSL解决方案在SSL任务中是一个很大的挑战,包括拼图游戏。我们提出对抗性鲁棒训练作为一种有效的自动捷径删除。我们在各种玩具和真实世界的数据集上实现了与最先进的(SOTA)异常检测方法相比具有竞争力或优越的结果。与许多竞争对手不同,该框架稳定,快速,数据高效,并且不需要无原则的早期停止。

英文摘要:

Autoencoder, as an essential part of many anomaly detection methods, is lacking flexibility on normal data in complex datasets. U-Net is proved to be effective for this purpose but overfits on the training data if trained by just using reconstruction error similar to other AE-based frameworks. Puzzle-solving, as a pretext task of self-supervised learning (SSL) methods, has earlier proved its ability in learning semantically meaningful features. We show that training U-Nets based on this task is an effective remedy that prevents overfitting and facilitates learning beyond pixel-level features. Shortcut solutions, however, are a big challenge in SSL tasks, including jigsaw puzzles. We propose adversarial robust training as an effective automatic shortcut removal. We achieve competitive or superior results compared to the State of the Art (SOTA) anomaly detection methods on various toy and real-world datasets. Unlike many competitors, the proposed framework is stable, fast, data-efficient, and does not require unprincipled early stopping.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑