C-ViT: An Improved ViT Model for Multi-Label Classification of Bamboo Chopstick Defects

  • 0College of Information and Intelligence, Hunan Agricultural University, Changsha 410128, China.

|

|

Summary

This summary is machine-generated.

This study introduces C-ViT, an improved Vision Transformer (ViT) model for bamboo chopstick defect detection. The model enhances quality inspection accuracy by incorporating a convolutional neural network feature extraction module and a novel Hard Examples Contrastive Loss function.

Area Of Science

  • Computer Vision
  • Machine Learning
  • Quality Control

Background

  • Disposable bamboo chopstick quality impacts consumer experience and safety.
  • Current quality inspection methods require refinement for production demands.
  • Vision Transformer (ViT) models face challenges with extreme aspect ratios, like bamboo chopsticks.

Purpose Of The Study

  • To develop an improved ViT model (C-ViT) for multi-label defect classification of bamboo chopsticks.
  • To introduce a novel Hard Examples Contrastive Loss (HCL) function to enhance learning from difficult examples.
  • To improve the accuracy and efficiency of bamboo chopstick quality inspection.

Main Methods

  • Proposed C-ViT model featuring a Convolutional Neural Network Feature Extraction (CFE) module to replace traditional patch embedding.
  • Developed Hard Examples Contrastive Loss (HCL) function for dynamic hard example selection and contrastive learning.
  • Evaluated models on a self-built bamboo chopstick defect dataset (BCDD) and the public VOC2012 dataset.

Main Results

  • C-ViT improved mean Average Precision (mAP) by 1.2% to 92.8% on the BCDD dataset compared to the ViTS model.
  • The addition of HCL further boosted performance to 94.3% mAP on the BCDD dataset.
  • The HCL function demonstrated effectiveness in multi-label classification tasks on the VOC2012 dataset.

Conclusions

  • The C-ViT model, with its CFE module, is effective for defect detection in objects with extreme aspect ratios.
  • The HCL function significantly enhances multi-label classification performance by focusing on hard examples.
  • The proposed methods offer a promising approach for improving automated quality inspection in manufacturing.

Related Concept Videos

Lumber Defects 01:23

563

Lumber defects, which can affect both the appearance and structural integrity of wood, include a variety of growth and manufacturing flaws. Growth defects such as knots and knotholes occur where branches were once attached to the tree trunk, with knotholes forming when these knots fall out. Other natural defects include decay and insect damage, which compromise the wood's strength and durability.
Shakes are minor fractures that run along or across the wood's annual rings, while wane is...

Force Classification 01:22

2.5K

Forces play a crucial role in the study of physics and engineering. They are essential in describing the motion, behavior, and equilibrium of objects in the physical world. Forces can be classified based on their origin, type, and direction of action.
Contact and non-contact forces are two of the most widely used categories of forces. As the name suggests, contact forces require physical contact between two objects to act upon each other. Examples of contact forces include frictional,...

Classification of Neurotransmitters 01:30

5.4K

Neurotransmitters play a crucial role in the communication between neurons in the autonomic nervous system. Neurons in the autonomic nervous system can be cholinergic or adrenergic depending on the neurotransmitters synthesized. Cholinergic neurons use acetylcholine as their primary neurotransmitter. This includes all the preganglionic fibers of the sympathetic and pre- and postganglionic fibers of the parasympathetic nervous systems. In addition, neurons of the somatic nervous system also use...

Classification of Leukocytes 01:30

6.2K

Leukocytes are classified into two groups based on the presence or absence of cytoplasmic granules. Granular leukocytes, which contain granules, belong to the myeloid lineage and are divided into three subtypes: neutrophils, eosinophils, and basophils. These cells are roughly spherical and characterized by the granules in their cytoplasm.
Neutrophils are the most abundant type of granular leukocytes, comprising 50-70% of all leukocytes. They feature small, evenly distributed granules and a...

Classification of Illness 01:17

8.9K

The meaning of illness is individualized to each person who experiences an alteration in health. In contrast, disease is a medical term indicating a pathological change in the structure and function of the body or mind. It is a condition that has specific symptoms and boundaries.
An illness is a response to a disease in which the person's level of functioning is changed compared with a previous level. The general classification of illness includes acute and chronic.
Acute illness is severe...

Classification of Bones 01:18

10.2K

The bones of the human skeletal system are of varied shapes, sizes, and functions. They can be classified based on their shape and function into four major classes: long bones, short bones, flat bones, and irregular bones. Some classifications include a fifth type, the sesamoid bones, as a separate class, whereas others categorize them under short bones.
Long and Short Bones
The appendicular skeleton, particularly the upper and lower limbs, is primarily made of long and short bones. The...