Residential College | false |
Status | 已發表Published |
Semi-supervised transfer learning with hierarchical self-regularization | |
Li, Xingjian1; Abuduweili, Abulikemu1; Shi, Humphrey6,7,8,9![]() ![]() | |
2023-07-26 | |
Source Publication | Pattern Recognition
![]() |
ISSN | 0031-3203 |
Volume | 144Pages:109831 |
Abstract | Both semi-supervised learning and transfer learning aim at lowering the annotation burden for training models. However, such two tasks are usually studied separately, i.e. most semi-supervised learning algorithms train models from scratch while transfer learning assumes pre-trained models as the initialization. In this work, we focus on a previously-less-concerned setting that further reduces the annotation efforts through incorporating both semi-supervised and transfer learning, where specifically a pre-trained source model is used as the initialization of semi-supervised learning. As those powerful pre-trained models are ubiquitously available nowadays and can considerably benefit various down-streaming tasks, such a setting is relevant to real-world applications yet challenging to design effective algorithms. Aiming at enabling transfer learning under semi-supervised settings, we propose a hierarchical self-regularization mechanism to exploit unlabeled samples more effectively, where a novel self-regularizer has been introduced to incorporate both individual-level and population-level regularization terms. The former term employs self-distillation to regularize learned deep features for each individual sample, and the latter one enforces self-consistency on feature distributions between labeled and unlabeled samples. Samples involved in both regularizers are weighted by an adaptive strategy, where self-regularization effects of both terms are adaptively controlled by the confidence of every sample. To validate our algorithm, exhaustive experiments have been conducted on diverse datasets such as CIFAR-10 for general object recognition, CUB-200-2011/MIT-indoor-67 for fine-grained classification and MURA for medical image classification. Compared with state-of-the-art semi-supervised learning methods including Pseudo Label, Mean Teacher, MixMatch and FixMatch, our algorithm demonstrates two advantages: first of all, the proposed approach adopts a new point of view to tackle problems caused by inadequate supervision and achieves very competitive results; then, it is complementary to these state-of-the-art methods and thus can be combined with them to get additional improvements. Furthermore, our method can also be applied to fully supervised transfer learning and self-supervised learning. We have published our code at https://github.com/SHI-Labs/Semi-Supervised-Transfer-Learning. |
Keyword | Adaptive Sample Selection Deep Learning Fine-tuning Hierarchical Consistency Semi-supervised Learning Transfer Learning |
DOI | 10.1016/j.patcog.2023.109831 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science ; Engineering |
WOS Subject | Computer Science, Artificial Intelligence ; Engineering, Electrical & Electronic |
WOS ID | WOS:001065559000001 |
Scopus ID | 2-s2.0-85167442327 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Science and Technology |
Corresponding Author | Shi, Humphrey |
Affiliation | 1.Carnegie Mellon University, Pittsburg, United States 2.Baidu Research, Beijing, China 3.University of Macau, China 4.Statistical Science, Tsinghua University, Beijing, China 5.Boston Consulting Group, Inc., Beijing, China 6.Computer Science, University of Oregon, Eugene, United States 7.Electrical and Computer Engineering, UIUC, Champaign and Urbana, UIUC, United States 8.Picsart AI Research, Miami, United States 9.School of Interactive Computing, Georgia Tech, Atlanta, United States |
Recommended Citation GB/T 7714 | Li, Xingjian,Abuduweili, Abulikemu,Shi, Humphrey,et al. Semi-supervised transfer learning with hierarchical self-regularization[J]. Pattern Recognition, 2023, 144, 109831. |
APA | Li, Xingjian., Abuduweili, Abulikemu., Shi, Humphrey., Yang, Pengkun., Dou, Dejing., Xiong, Haoyi., & Xu, Chengzhong (2023). Semi-supervised transfer learning with hierarchical self-regularization. Pattern Recognition, 144, 109831. |
MLA | Li, Xingjian,et al."Semi-supervised transfer learning with hierarchical self-regularization".Pattern Recognition 144(2023):109831. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment