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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Age Estimation | UTKFace (test) | MAE5.59 | 36 | |
| Age Estimation | AgeDB-DIR v1 (test) | MAE8.52 | 24 | |
| Audio Quality Assessment | BVCC (test) | MAE0.463 | 9 | |
| Image Age Estimation | UTKFace 50 labeled samples | MAE9.33 | 9 | |
| Image Age Estimation | UTKFace 250 labeled samples | MAE6.57 | 9 | |
| Image Age Estimation | UTKFace | MAE (Years)5.51 | 9 | |
| NLP Opinion Mining | Yelp Review (test) | MAE0.632 | 9 |