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.

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

A Novel Transfer Learning Method for Robot Bearing Fault Diagnosis Based on Deep Convolutional Residual Wasserstein Adversarial Network.

A Dual-Weighted Adversarial Network for Partial Domain Fault Diagnosis of Machinery

A Weighted Subdomain Adaptation Network for Partial Transfer Fault Diagnosis of Rotating Machinery

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

Local Weight Coupled Network: Multi-Modal Unequal Semi-Supervised Domain Adaptation.

Bearing fault diagnosis based on partial domain adaptation adversarial network

A partial domain adaptation scheme based on weighted adversarial nets with improved CBAM for fault diagnosis of wind turbine gearbox

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

A Balanced Adversarial Domain Adaptation Method for Partial Transfer Intelligent Fault Diagnosis

Partial Transfer Learning of Multidiscriminator Deep Weighted Adversarial Network in Cross-Machine Fault Diagnosis

Partial transfer learning in machinery cross-domain fault diagnostics using class-weighted adversarial networks

Defect Category Prediction Based on Multi-Source Domain Adaptation

Discriminative Partial Domain Adversarial Network

Non-Uniformly Weighted Multisource Domain Adaptation Network For Fault Diagnosis Under Varying Working Conditions

Universal Domain Adaptation in Fault Diagnostics With Hybrid Weighted Deep Adversarial Learning

Cross-Machine Transfer Fault Diagnosis by Ensemble Weighting Subdomain Adaptation Network

Tire Defect Detection by Dual-Domain Adaptation-Based Transfer Learning Strategy

Dual Structural Consistent Partial Domain Adaptation Network for Intelligent Machinery Fault Diagnosis

Subdomain adaptation network with category isolation strategy for tire defect detection

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