A Data-Centric Strategy to Improve Performance of Automatic Pavement Defects Detection

Tianjie Zhang, Donglei Wang, Yang Lu

Research output: Contribution to journalArticlepeer-review

6 Scopus citations

Abstract

<p> <p id="x-x-x-sp0090"> An integrated data-centric portfolio is presented in this work to tailor the performance of the deep learning method. The proposed portfolio includes attention mechanisms, feature-enabled image augmentation strategy, and orthogonal test-based parameter fine-tuning. It focuses on improving the quality and diversity of the data without changing the architecture of the model. The results show that by embedding attention modules into the deep learning-based model, the mAP50 is increased by 3.1% compared to the benchmark. A Class-Specific Image Augmentation (CSIA) method is proposed to work as an optimism strategy for quantifying the number of generated images for each distress. It outperforms augmenting images for all distresses equally which has been widely used in many studies. An orthogonal test method is introduced to decrease the training time for parameter fine-tuning. With the proposed data-centric portfolio, mAP50 of the YOLOv5s model is significantly improved from 0.594 to 0.818. </p></p>
Original languageAmerican English
Article number105334
JournalAutomation in Construction
Volume160
DOIs
StatePublished - Apr 2024

Keywords

  • YOLO5
  • attention module
  • image augmentation
  • orthogonal test
  • pavement crack detection

EGS Disciplines

  • Civil Engineering

Fingerprint

Dive into the research topics of 'A Data-Centric Strategy to Improve Performance of Automatic Pavement Defects Detection'. Together they form a unique fingerprint.

Cite this