TBDA-Net: A Task-based Bias Domain Adaptation Network under Industrial Small Samples

The small sample problem is an open challenge in industrial data-driven fault diagnosis. Domain adaptation (DA) has been successfully used to enhance small samples by transferring fault samples in other working conditions, but it suffers from low accuracy and poor generalization for modeling. To address this issue, a task-based bias domain adaptation network (TBDA-Net) is proposed. First, an adaptive dimension alignment sub-net is proposed, which overcomes the information loss of the target domain caused by the dimension alignment of different domains, thus the accuracy of modeling is improved. Second, a task-based distribution matching sub-net is proposed, in which the correlation difference of domains is considered and the designed auxiliary branch is embedded in the matching network to protect the sample diversity of multi-source domains, thus the generalization of modeling is improved. Experiment results show the TBDA-Net outperforms the existing methods for modeling under industrial small samples.