Abstract:In recent years, defect prediction techniques based on deep learning have become a prominent research topic in the field of software engineering. These techniques can identify potential defects without executing the code. However, existing approaches mostly concentrate on determining the presence of defects at the method-level code, lacking the ability to precisely classify specific defect categories. Consequently, this undermines the efficiency of developers in locating and rectifying defects. Furthermore, in practical software development, new projects often lack sufficient defect data to train high-accuracy deep learning models. Models trained on historical data from existing projects frequently struggle to achieve satisfactory generalization performance on new projects. Hence, this paper initially reformulates the traditional binary defect prediction task into a multi-label classification problem, employing defect categories described in the Common Weakness Enumeration (CWE) as fine-grained predictive labels. To enhance the model performance in cross-project scenarios, this paper proposes a multi-source domain adaptation framework that integrates adversarial training and attention mechanisms. Specifically, the proposed framework employs adversarial training to mitigate domain (i.e., software projects) discrepancies, and further utilizes domain-invariant features to capture feature correlations between each source domain and the target domain. Simultaneously, the proposed framework employs a weighted maximum mean discrepancy as an attention mechanism to minimize the representation distance between source and target domain features, facilitating model in learning more domain-independent features. The experiments on 8 real-world open-source projects show that the proposed approach achieves significant performance improvements compared to state-of-the-art baselines.

Prototype-guided domain adaptive one-stage object detector for defect detection

Unifying Defect Prediction, Categorization, and Repair by Multi-Task Deep Learning

AEGLR-Net: Attention Enhanced Global-Local Refined Network for Accurate Detection of Car Body Surface Defects

A real-time anchor-free defect detector with global and local feature enhancement for surface defect detection

An Industrial Defect Detection Network with Fine-Grained Supervision and Adaptive Contrast Enhancement.

Cross-Domain Detection Via Graph-Induced Prototype Alignment

PGA-Net: Pyramid Feature Fusion and Global Context Attention Network for Automated Surface Defect Detection

A One‐Stage Deep Learning Model for Industrial Defect Detection

Defect Category Prediction Based on Multi-Source Domain Adaptation

GPA-3D: Geometry-aware Prototype Alignment for Unsupervised Domain Adaptive 3D Object Detection from Point Clouds

UDA-DP: Unsupervised Domain Adaptation for Software Defect Prediction

Category-level selective dual-adversarial network using significance-augmented unsupervised domain adaptation for surface defect detection

Style Adaptation module: Enhancing detector robustness to inter-manufacturer variability in surface defect detection

Surface defect detection of industrial components based on vision

A novel class-level weighted partial domain adaptation network for defect detection

Multi-level joint distributed alignment-based domain adaptation for cross-scenario strip defect recognition

A hierarchical attention detector for bearing surface defect detection

Learning a Domain Classifier Bank for Unsupervised Adaptive Object Detection

A Robust Defect Detection Method with a Generalization Enhancer and Cross-Modality Aggregator for Cylinder Bores

Multiscale Adversarial and Weighted Gradient Domain Adaptive Network for Data Scarcity Surface Defect Detection