Imbalanced segmentation for abnormal cotton fiber based on GAN and multiscale residual U-Net

Abstract

The scale of white foreign fibers in bobbin yarn is small, resulting in multiple types of data imbalance in the dataset. These imbalances include a severe imbalance of foreign fiber pixels compared to background pixels and an imbalance in the size target scale. Consequently, conventional semantic segmentation networks struggle to segment these fibers effectively. First, in tackling the scarcity of white foreign fiber instances within bobbin yarn samples, this research utilizes original foreign fiber images to train the DCGAN and generate adequate training samples. Secondly, a multiscale residual U-Net is constructed to extract foreign fiber features from different scales. The network is encouraged to learn semantic features at each scale and each layer of the decoding stage. This overcomes the problem of scale imbalance in the foreign fiber dataset and enhances the model's capability to extract weak semantic information from small targets. Thirdly, a weighted binary cross-entropy loss function is integrated into the network's training phase to rectify the class imbalance and refine segmentation performance. This function adjusts the weighting of foreign fiber pixel data, thereby addressing the disproportionate distribution between foreign fibers and background pixels within the dataset. Finally, the proposed method is experimentally validated using a dataset of white foreign fibers. The experimental results show that the proposed method achieves better results in the critical evaluation metrics, as evidenced by the accuracy of 97.52 %, the MIoU of 95.26 %, the DICE coefficient of 81.29 %, and the F1 Score of 84.92 %. These statistics demonstrate the method's efficacy in achieving high-precision segmentation of white foreign fibers.

Keywords

• Data imbalance
• Foreign fibers
• Segmentation
• GAN
• U-Net