Deep semi-supervised regression via pseudo-label filtering and calibration

被引:0
|
作者
Jo, Yongwon [1 ]
Kahng, Hyungu [2 ]
Kim, Seoung Bum [1 ]
机构
[1] Korea Univ, Dept Ind & Management Engn, 145 Anam Ro, Seoul 02841, South Korea
[2] Korea Univ, Dept Convergence Business, 2511 Sejong Ro, Sejong Si 30019, South Korea
基金
新加坡国家研究基金会;
关键词
Semi-supervised learning; Regression; Uncertainty; Similarity; Pseudo-label;
D O I
10.1016/j.asoc.2024.111670
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Semi -supervised learning (SSL) is a widely used model training paradigm that effectively utilizes a limited set of labeled data and a substantially larger pool of unlabeled data. Historically, the focus of SSL research has predominantly been on classification tasks, employing methods such as consistency regularization and pseudolabeling. However, the direct application of these methods to regression tasks presents significant challenges, primarily due to the complexities associated with evaluating the reliability of pseudo -labels in a regression context. This paper introduces SimRegMatch, a novel semi -supervised regression (SSR) framework devised to overcome this specific challenge, by combining an uncertainty -based filtering mechanism with a similaritybased pseudo -label calibration approach. The former component is tasked with discerning which unlabeled examples possess pseudo -labels of sufficient reliability, achieved through the estimation of uncertainty levels. The latter component then refines these pseudo -labels by propagating information from labeled to unlabeled examples, thereby enhancing the overall quality of the pseudo -labels. The efficacy of SimRegMatch was rigorously tested through experiments conducted on the publicly available AgeDB dataset, which is centered around age prediction, as well as on a practical regression problem focused on the detection of interior noise levels in automobiles using accelerometer data. When benchmarked against current state-of-the-art methods in semi -supervised regression, SimRegMatch exhibited notable improvements in regression performance. Additionally, a series of ablation studies were carried out to dissect and understand the specific elements of the framework that were instrumental in achieving these performance enhancements. SimRegMatch addresses a pivotal issue in semi -supervised regression - the assessment of regression pseudo -label reliability - and substantially elevates model performance. By combining the strengths of uncertainty estimation and pseudolabel calibration, SimRegMatch emerges as a robust and versatile framework with significant potential for broad applicability in various SSR scenarios. A PyTorch implementation is publicly available at https://github.com/ YongwonJo/SimRegMatch.
引用
收藏
页数:14
相关论文
共 50 条
  • [1] Semi-supervised regression via embedding space mapping and pseudo-label smearing
    Liu, Liyan
    Zhang, Jin
    Qian, Kun
    Min, Fan
    APPLIED INTELLIGENCE, 2024, 54 (20) : 9622 - 9640
  • [2] Naive semi-supervised deep learning using pseudo-label
    Zhun Li
    ByungSoo Ko
    Ho-Jin Choi
    Peer-to-Peer Networking and Applications, 2019, 12 : 1358 - 1368
  • [3] Naive semi-supervised deep learning using pseudo-label
    Li, Zhun
    Ko, ByungSoo
    Choi, Ho-Jin
    PEER-TO-PEER NETWORKING AND APPLICATIONS, 2019, 12 (05) : 1358 - 1368
  • [4] Pseudo-Label Calibration Semi-supervised Multi-Modal Entity Alignment
    Wang, Luyao
    Qi, Pengnian
    Bao, Xigang
    Zhou, Chunlai
    Qin, Biao
    THIRTY-EIGHTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, VOL 38 NO 8, 2024, : 9116 - 9124
  • [5] Evidential Pseudo-Label Ensemble for semi-supervised classification
    Wang, Kai
    Zhang, Changqing
    Geng, Yu
    Ma, Huan
    PATTERN RECOGNITION LETTERS, 2024, 177 : 135 - 141
  • [6] Pseudo-label Alignment for Semi-supervised Instance Segmentation
    Hu, Jie
    Chen, Chen
    Cao, Liujuan
    Zhang, Shengchuan
    Shu, Annan
    Jiang, Guannan
    Ji, Rongrong
    2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, : 16291 - 16301
  • [7] Pseudo-label semi-supervised learning for soybean monitoring
    Menezes, Gabriel Kirsten
    Astolfi, Gilberto
    Martins, Jose Augusto Correa
    Tetila, Everton Castelao
    Oliveira Jr, Adair da Silva
    Goncalves, Diogo Nunes
    Marcato Jr, Jose
    Silva, Jonathan Andrade
    Li, Jonathan
    Goncalves, Wesley Nunes
    Pistori, Hemerson
    SMART AGRICULTURAL TECHNOLOGY, 2023, 4
  • [8] Federated Semi-Supervised Learning for Medical Image Segmentation via Pseudo-Label Denoising
    Qiu, Liang
    Cheng, Jierong
    Gao, Huxin
    Xiong, Wei
    Ren, Hongliang
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2023, 27 (10) : 4672 - 4683
  • [9] Semi-supervised medical imaging segmentation with soft pseudo-label fusion
    Xiaoqiang Li
    Yuanchen Wu
    Songmin Dai
    Applied Intelligence, 2023, 53 : 20753 - 20765
  • [10] Semi-supervised medical imaging segmentation with soft pseudo-label fusion
    Li, Xiaoqiang
    Wu, Yuanchen
    Dai, Songmin
    APPLIED INTELLIGENCE, 2023, 53 (18) : 20753 - 20765