arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2509.03465cs.CV

面向道路缺陷检测的图像生成器与检测器联合训练

Joint Training of Image Generator and Detector for Road Defect Detection

Mitsubishi Electric Research Laboratories (MERL)(三菱电机研究实验室)

Kuan-Chuan Peng

AI总结:

针对边缘设备部署需求,提出JTGD方法联合训练道路缺陷检测的生成器与检测器,通过双判别器、CLIP基FID损失生成高质量难例,在无集成和TTA时性能优于SOTA且参数量不足基线20%。

摘要:

道路缺陷检测对道路管理部门减少因道路缺陷导致的车辆损坏至关重要。考虑到缺陷检测器通常部署在内存和计算资源有限的边缘设备的实际场景,我们的目标是在不使用基于集成的方法或测试时增强(TTA)的情况下实现道路缺陷检测。为此,我们提出了面向道路缺陷检测的图像生成器与检测器联合训练方法(简称JTGD)。我们为生成模型设计了双判别器,以确保合成的缺陷斑块和整体图像都看起来真实可信。我们提出的基于CLIP的Fréchet Inception Distance损失提升了合成图像的质量。JTGD中的生成模型与检测器联合训练,以促使生成模型为检测器合成更难的样本。由于上述设计带来的质量更高、难度更大的合成图像被用于数据增强,在无集成和TTA的条件下,JTGD在RDD2022道路缺陷检测基准的多个国家数据集上均优于当前最优方法。JTGD的参数量仅不到对比基线的20%,这使其更适合实际部署在边缘设备上。

英文摘要:

Road defect detection is important for road authorities to reduce the vehicle damage caused by road defects. Considering the practical scenarios where the defect detectors are typically deployed on edge devices with limited memory and computational resource, we aim at performing road defect detection without using ensemble-based methods or test-time augmentation (TTA). To this end, we propose to Jointly Train the image Generator and Detector for road defect detection (dubbed as JTGD). We design the dual discriminators for the generative model to enforce both the synthesized defect patches and overall images to look plausible. The synthesized image quality is improved by our proposed CLIP-based Fréchet Inception Distance loss. The generative model in JTGD is trained jointly with the detector to encourage the generative model to synthesize harder examples for the detector. Since harder synthesized images of better quality caused by the aforesaid design are used in the data augmentation, JTGD outperforms the state-of-the-art method in the RDD2022 road defect detection benchmark across various countries under the condition of no ensemble and TTA. JTGD only uses less than 20% of the number of parameters compared with the competing baseline, which makes it more suitable for deployment on edge devices in practice.

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

↑