TrustMAE: A Noise-Resilient Defect Classification Framework using Memory-Augmented Auto-Encoders with Trust Regions

AI-generated keywords: TrustMAE

AI-generated Key Points

  • TrustMAE is a framework for product defect classification
  • TrustMAE can accept datasets with unlabeled images, unlike other methods
  • It is robust against noises or defective images in the training dataset
  • Utilizes a memory-augmented auto-encoder with a sparse memory addressing scheme
  • Incorporates a trust-region memory updating scheme to prevent noise in memory slots
  • Can reconstruct defect-free images and identify defective regions using a perceptual distance network
  • Performs competitively on noise-free MVTec datasets and outperforms other baselines at up to 40% noise level
  • Achieves the best AUC score on most classes and overall mean AUC compared to other works on defect detection
  • Lower performance in certain classes due to difficulties in modeling large variations in normal data caused by rotations or randomly appearing prints
  • Does not include a warping module but compares favorably to recent works in defect segmentation using pixel-level AUC evaluation
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Daniel Stanley Tan, Yi-Chun Chen, Trista Pei-Chun Chen, Wei-Chao Chen

Accepted at WACV 2021
License: CC BY-NC-SA 4.0

Abstract: In this paper, we propose a framework called TrustMAE to address the problem of product defect classification. Instead of relying on defective images that are difficult to collect and laborious to label, our framework can accept datasets with unlabeled images. Moreover, unlike most anomaly detection methods, our approach is robust against noises, or defective images, in the training dataset. Our framework uses a memory-augmented auto-encoder with a sparse memory addressing scheme to avoid over-generalizing the auto-encoder, and a novel trust-region memory updating scheme to keep the noises away from the memory slots. The result is a framework that can reconstruct defect-free images and identify the defective regions using a perceptual distance network. When compared against various state-of-the-art baselines, our approach performs competitively under noise-free MVTec datasets. More importantly, it remains effective at a noise level up to 40% while significantly outperforming other baselines.

Submitted to arXiv on 29 Dec. 2020

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2012.14629v1

, , , , . In this paper, the authors propose a framework called TrustMAE to address the problem of product defect classification. The existing summary highlights that TrustMAE is capable of accepting datasets with unlabeled images, unlike other methods that rely on difficult-to-collect and laborious-to-label defective images. Additionally, TrustMAE is robust against noises or defective images in the training dataset, which sets it apart from most anomaly detection methods. The framework utilizes a memory-augmented auto-encoder with a sparse memory addressing scheme to prevent over-generalization. It also incorporates a novel trust-region memory updating scheme to keep noises away from the memory slots. This combination allows TrustMAE to reconstruct defect-free images and identify defective regions using a perceptual distance network. To evaluate its performance, TrustMAE is compared against various state-of-the-art baselines using noise-free MVTec datasets. The results show that TrustMAE performs competitively under these conditions. However, what makes it particularly impressive is its effectiveness at a noise level of up to 40%, significantly outperforming other baselines. The authors discuss some additional details about their work and mention that their model achieves the best AUC score on most classes and overall mean AUC when compared to other works on defect detection. However, they acknowledge lower performance in certain classes such as metal nut, pill, screw, and capsule due to difficulties in modeling large variations in normal data caused by rotations or randomly appearing prints. The authors note that while adding a warping module could improve performance by aligning data, it falls outside the scope of their paper. They also compare their method with prior works in defect segmentation using pixel-level AUC evaluation. By imposing a threshold on reconstruction errors, they obtain segmentation maps and demonstrate competitive performance compared to recent works in defect segmentation. Overall, TrustMAE proves to be an effective framework for product defect classification. Its ability to handle unlabeled datasets, robustness against noisy training data, and competitive performance make it a valuable contribution to the field.
Created on 13 Jul. 2023

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.