STUI-NET: Semi-Supervised Transformer for Underwater Information Enhancement

Underwater Image Restoration Technology (UIRT) constitutes a pivotal base for subsequent tasks yet confronts obstacles like data scarcity and image distortion in practical deployments. To optimize the use of scarce annotated and copious unlabeled data, we introduce an avant-garde semi-supervised method for underwater image restoration, named STUI-Net. Concurrently, we develop a novel teacher-student architecture utilizing Transformer technology, named GFT-Net. Initially, GFT-Net employs a tripartite branch network—comprising Gradient, Feature Extraction, and Transformer modules to thoroughly extract features from underwater images. This method then amalgamates multi-source features, encompassing edge, gradient, local, and global data, addressing underwater image deterioration in multifaceted environments. Additionally, we engineer a Supervisor role to rectify potential misdirection by the teacher network in the semi-supervised model, thereby enhancing training stability on unlabeled datasets. Empirical studies affirm the robust generalization competence of STUI-Net in authentic underwater environments.