Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

RankUp: Boosting Semi-Supervised Regression with an Auxiliary Ranking Classifier

About

State-of-the-art (SOTA) semi-supervised learning techniques, such as FixMatch and it's variants, have demonstrated impressive performance in classification tasks. However, these methods are not directly applicable to regression tasks. In this paper, we present RankUp, a simple yet effective approach that adapts existing semi-supervised classification techniques to enhance the performance of regression tasks. RankUp achieves this by converting the original regression task into a ranking problem and training it concurrently with the original regression objective. This auxiliary ranking classifier outputs a classification result, thus enabling integration with existing semi-supervised classification methods. Moreover, we introduce regression distribution alignment (RDA), a complementary technique that further enhances RankUp's performance by refining pseudo-labels through distribution alignment. Despite its simplicity, RankUp, with or without RDA, achieves SOTA results in across a range of regression benchmarks, including computer vision, audio, and natural language processing tasks. Our code and log data are open-sourced at https://github.com/pm25/semi-supervised-regression.

Pin-Yen Huang, Szu-Wei Fu, Yu Tsao• 2024

Related benchmarks

TaskDatasetResultRank
Age EstimationUTKFace (test)
MAE5.59
36
Age EstimationAgeDB-DIR v1 (test)
MAE8.52
24
Audio Quality AssessmentBVCC (test)
MAE0.463
9
Image Age EstimationUTKFace 50 labeled samples
MAE9.33
9
Image Age EstimationUTKFace 250 labeled samples
MAE6.57
9
Image Age EstimationUTKFace
MAE (Years)5.51
9
NLP Opinion MiningYelp Review (test)
MAE0.632
9
Showing 7 of 7 rows

Other info

Code

Follow for update