Journal of Jilin University (Information Science Edition) ›› 2026, Vol. 44 ›› Issue (4): 851-862.

Previous Articles     Next Articles

Transformer-Based Image Classification Method of Semi-Supervised Partial Multi-Label

LI Yan1,2, ZHOU Zhilong3, ZHU Biao4   

  1. 1. Faculty of Arts and Sciences, Beijing Normal University, Zhuhai 519085, China; 2. School of Applied Mathematics,Beijing Normal University at Zhuhai, Zhuhai 519087, China; 3. School of Statistics, Beijing Normal University, Beijing 100875, China;4. College of Mathematics and Information Science, Hebei University, Baoding 071002, China
  • Received:2025-08-29 Online:2026-08-06 Published:2026-08-06

Abstract:

In SSPML( Semi-Supervised Partial Multi-Label Learning), complex correlations exist between the feature and label, and within them. Models leveraging label correlations are susceptible to interference from noisy labels in candidate label sets. T-SSPML ( Transformer Encoder-Based Semi-Supervised Partial Multi-Label Learning), a novel semi-supervised partial multi-label image classification algorithm is proposed based on Transformer. A backbone network is constructed using Transformer encoders, taking both image features and label features as joint inputs, and a new loss function is defined. The multi-head self-attention mechanism is employed to capture holistic structural information and label interdependencies. A label masking mechanism is
introduced to enhance the model’s capability in learning label correlation information. Extensive experiments on multiple datasets demonstrate that the T-SSPML algorithm achieves superior classification performance compared to other representative methods in most cases.

Key words:

CLC Number: 

  • TP181